tommy/dfdewey
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

postgresql datastore tests

Browse source
This commit is contained in:
Jason Solomon 2020-11-17 12:48:00 +11:00
parent f109dd7ad8
commit 89b1a91659
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
dfdewey/datastore/postgresql_test.py
View file

@ -31,7 +31,7 @@ class PostgresqlTest(unittest.TestCase):
""" """
with mock.patch('psycopg2.connect') as _: with mock.patch('psycopg2.connect') as _:
db = PostgresqlDataStore(autocommit=True) db = PostgresqlDataStore(autocommit=True)
return db return db
@mock.patch('psycopg2.extras.execute_values') @mock.patch('psycopg2.extras.execute_values')
def test_bulk_insert(self, _): def test_bulk_insert(self, _):
@ -70,7 +70,8 @@ class PostgresqlTest(unittest.TestCase):
def test_switch_database(self): def test_switch_database(self):
"""Test switch database method.""" """Test switch database method."""
db = self._get_datastore() db = self._get_datastore()
db.switch_database(db_name='dfdewey', autocommit=True) with mock.patch('psycopg2.connect') as _:
db.switch_database(db_name='dfdewey', autocommit=True)
def test_table_exists(self): def test_table_exists(self):
"""Test table exists method.""" """Test table exists method."""