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