Fix execution tests

Re spine-tools/Spine-Database-API#316
spine-tools · Jan 15, 2024 · cf4da56 · cf4da56
1 parent 1864d37
commit cf4da56
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/execution_tests/export_entities_with_entity_alternatives/execution_test.py b/execution_tests/export_entities_with_entity_alternatives/execution_test.py
@@ -20,7 +20,7 @@ def setUp(self):
             self._database_path.unlink()
         url = "sqlite:///" + str(self._database_path)
         with DatabaseMapping(url, create=True) as db_map:
-            self._assert_item_added(db_map.add_entity_class_item(name="Object"))
+            self._assert_item_added(db_map.add_entity_class_item(name="Object", active_by_default=True))
             self._assert_item_added(db_map.add_entity_item(entity_class_name="Object", name="none_none"))
             self._assert_item_added(db_map.add_entity_item(entity_class_name="Object", name="true_none"))
             self._assert_item_added(db_map.add_entity_item(entity_class_name="Object", name="none_true"))

diff --git a/execution_tests/modify_connection_filter_by_script/execution_test.py b/execution_tests/modify_connection_filter_by_script/execution_test.py
@@ -8,6 +8,7 @@
     DatabaseMapping,
     import_alternatives,
     import_entities,
+    import_entity_alternatives,
     import_entity_classes,
     import_parameter_definitions,
     import_parameter_values,
@@ -30,10 +31,11 @@ def setUp(self):
             self._database_path.unlink()
         url = "sqlite:///" + str(self._database_path)
         with DatabaseMapping(url, create=True) as db_map:
-            import_entity_classes(db_map, ("object_class",))
+            import_entity_classes(db_map, (("object_class",),))
             import_entities(db_map, (("object_class", "object"),))
             import_parameter_definitions(db_map, (("object_class", "parameter"),))
             import_alternatives(db_map, ("alternative",))
+            import_entity_alternatives(db_map, (("object_class", "object", "alternative", True),))
             import_parameter_values(
                 db_map,
                 (