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()