def test_connect(self):
mock_postgresql = Mock()
with patch.dict('sys.modules', psycopg2=mock_postgresql):
mock_postgresql.connect.return_value = 'OK'
postgresql = PostgreSQLDatabase('test_host', 1234, 'test_database',
'test_user', 'password')
result = postgresql.connect()
self.assertEqual('OK', result)
mock_postgresql.connect.assert_called_once_with(
host='test_host', port=1234, dbname='test_database',
user='******', password='******',
)
def test_get_column_definitions(self, mock_fetch):
PostgreSQLDatabase().get_column_definitions('test_schema', 'test_table')
mock_fetch.assert_called_once_with(
'SELECT DISTINCT "column_name","data_type" ' \
'FROM "INFORMATION_SCHEMA"."columns" ' \
'WHERE "table_schema"=\'test_schema\' AND "table_name"=\'test_table\' ' \
'ORDER BY "column_name"',
connection=None
)
def test_get_column_definitions(self, mock_fetch):
PostgreSQLDatabase().get_column_definitions('test_schema',
'test_table')
mock_fetch.assert_called_once_with(
'SELECT DISTINCT "column_name","data_type" '
'FROM "information_schema"."columns" '
'WHERE "table_schema"=%(schema)s AND "table_name"=%(table)s '
'ORDER BY "column_name"',
connection=None,
parameters={
'schema': 'test_schema',
'table': 'test_table'
},
)
def setUpClass(cls):
super(TestPostgreSQL, cls).setUpClass()
cls.database = PostgreSQLDatabase()
def setUpClass(cls):
cls.database = PostgreSQLDatabase()