def test_replace_resource_from_source(self):
filters = {"database": {"scenario_filter": {13: False}}}
connection = Connection("source", "bottom", "destination", "top", filters)
original = database_resource("source", "sqlite:///db.sqlite", label="database")
connection.receive_resources_from_source([original])
self.assertEqual(connection.database_resources, {original})
modified = database_resource("source", "sqlite:///db2.sqlite", label="new database")
connection.replace_resource_from_source(original, modified)
self.assertEqual(connection.database_resources, {modified})
self.assertEqual(connection.resource_filters, {"new database": {"scenario_filter": {13: False}}})
def test_fetch_tools(self):
connection = Connection("source", "bottom", "destination", "top")
url = "sqlite:///" + os.path.join(self._temp_dir.name, "db.sqlite")
db_map = DiffDatabaseMapping(url, create=True)
import_tools(db_map, ("tool",))
db_map.commit_session("Add test data.")
db_map.connection.close()
resources = [database_resource("source", url)]
connection.receive_resources_from_source(resources)
self.assertFalse(connection.has_filters())
connection.fetch_database_items()
self.assertTrue(connection.has_filters())
self.assertEqual(connection.resource_filters, {resources[0].label: {"tool_filter": {1: False}}})