def get_table_names(
cls,
database: "Database",
inspector: Inspector,
schema: Optional[str],
) -> List[str]:
return BaseEngineSpec.get_table_names(
database=database,
inspector=inspector,
schema=schema,
)
def test_get_table_names(self):
inspector = mock.Mock()
inspector.get_table_names = mock.Mock(
return_value=["schema.table", "table_2"])
inspector.get_foreign_table_names = mock.Mock(return_value=["table_3"])
""" Make sure base engine spec removes schema name from table name
ie. when try_remove_schema_from_table_name == True. """
base_result_expected = ["table", "table_2"]
base_result = BaseEngineSpec.get_table_names(database=mock.ANY,
schema="schema",
inspector=inspector)
self.assertListEqual(base_result_expected, base_result)
def test_get_table_names(self):
inspector = mock.Mock()
inspector.get_table_names = mock.Mock(
return_value=['schema.table', 'table_2'])
inspector.get_foreign_table_names = mock.Mock(return_value=['table_3'])
""" Make sure base engine spec removes schema name from table name
ie. when try_remove_schema_from_table_name == True. """
base_result_expected = ['table', 'table_2']
base_result = BaseEngineSpec.get_table_names(schema='schema',
inspector=inspector)
self.assertListEqual(base_result_expected, base_result)
""" Make sure postgres doesn't try to remove schema name from table name
ie. when try_remove_schema_from_table_name == False. """
pg_result_expected = ['schema.table', 'table_2', 'table_3']
pg_result = PostgresEngineSpec.get_table_names(schema='schema',
inspector=inspector)
self.assertListEqual(pg_result_expected, pg_result)
