Skip to content

Commit 1825b25

Browse files
Add tests of APDB update record handling
The `test_ppdbBigQuery` module was also renamed to `tests_ppdb_bigquery` following snakecase conventions.
1 parent bb21823 commit 1825b25

8 files changed

Lines changed: 1343 additions & 139 deletions

python/lsst/dax/ppdb/tests/_bigquery.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@
3737
)
3838
from lsst.dax.apdb.sql import ApdbSql
3939
from lsst.dax.ppdb import PpdbConfig
40-
from lsst.dax.ppdb.bigquery import ChunkUploader, PpdbBigQuery
40+
from lsst.dax.ppdb.bigquery import PpdbBigQuery
41+
from lsst.dax.ppdb.bigquery.chunk_uploader import ChunkUploader
4142
from lsst.dax.ppdb.tests._ppdb import TEST_SCHEMA_RESOURCE_PATH
4243

4344
try:

tests/test_ppdbBigQuery.py

Lines changed: 0 additions & 138 deletions
This file was deleted.

tests/test_ppdb_bigquery.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
# This file is part of dax_ppdb.
2+
#
3+
# Developed for the LSST Data Management System.
4+
# This product includes software developed by the LSST Project
5+
# (http://www.lsst.org).
6+
# See the COPYRIGHT file at the top-level directory of this distribution
7+
# for details of code ownership.
8+
#
9+
# This program is free software: you can redistribute it and/or modify
10+
# it under the terms of the GNU General Public License as published by
11+
# the Free Software Foundation, either version 3 of the License, or
12+
# (at your option) any later version.
13+
#
14+
# This program is distributed in the hope that it will be useful,
15+
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16+
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17+
# GNU General Public License for more details.
18+
#
19+
# You should have received a copy of the GNU General Public License
20+
# along with this program. If not, see <http://www.gnu.org/licenses/>.
21+
22+
import unittest
23+
24+
from lsst.dax.ppdb.tests import PpdbTest
25+
from lsst.dax.ppdb.tests._bigquery import PostgresMixin, SqliteMixin
26+
27+
try:
28+
import testing.postgresql
29+
except ImportError:
30+
testing = None
31+
32+
33+
class SqliteTestCase(SqliteMixin, PpdbTest, unittest.TestCase):
34+
"""A test case for the PpdbBigQuery class using a SQLite backend."""
35+
36+
37+
@unittest.skipUnless(testing is not None, "testing.postgresql module not found")
38+
class PostgresTestCase(PostgresMixin, PpdbTest, unittest.TestCase):
39+
"""A test case for the PpdbBigQuery class using a Postgres backend."""

0 commit comments

Comments
 (0)