Update dataset factories in test_models

MedVIC-Lab · Oct 8, 2024 · 6b02ec8 · 6b02ec8
1 parent 853088c
commit 6b02ec8
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/swcc/tests/test_models.py b/swcc/tests/test_models.py
@@ -90,6 +90,7 @@ def test_dataset_force_create_overwrite(session):
         license=old_dataset.license,
         description=old_dataset.description,
         acknowledgement=old_dataset.acknowledgement,
+        file_source='./tests/test_data/project_demo.swproj',
     ).force_create()
     new_dataset.assert_remote()
 
@@ -109,6 +110,7 @@ def test_dataset_force_create_no_existing(session):
         license='license',
         description='description',
         acknowledgement='acknowledgement',
+        file_source='./tests/test_data/project_demo.swproj',
     ).force_create()
     dataset.assert_remote()
 
@@ -130,6 +132,7 @@ def test_dataset_force_create_backup(session, version):
         license=old_dataset.license,
         description=old_dataset.description,
         acknowledgement=old_dataset.acknowledgement,
+        file_source='./tests/test_data/project_demo.swproj',
     ).force_create(backup=True)
     new_dataset.assert_remote()
 
@@ -150,6 +153,7 @@ def test_dataset_force_create_backup_no_version_suffix(session):
         license=old_dataset.license,
         description=old_dataset.description,
         acknowledgement=old_dataset.acknowledgement,
+        file_source='./tests/test_data/project_demo.swproj',
     ).force_create(backup=True)
     new_dataset.assert_remote()
 
@@ -161,16 +165,26 @@ def test_dataset_force_create_backup_no_version_suffix(session):
 
 def test_dataset_force_create_backup_multiple_conflicts(session):
     old_dataset = factories.DatasetFactory(name='dataset').create()
-    factories.DatasetFactory(name='dataset-v1').create()
-    factories.DatasetFactory(name='dataset-v2').create()
-    factories.DatasetFactory(name='dataset-v3').create()
+    factories.DatasetFactory(
+        name='dataset-v1',
+        file_source='./tests/test_data/project_demo.swproj'
+    ).create()
+    factories.DatasetFactory(
+        name='dataset-v2',
+        file_source='./tests/test_data/project_demo.swproj'
+    ).create()
+    factories.DatasetFactory(
+        name='dataset-v3',
+        file_source='./tests/test_data/project_demo.swproj'
+    ).create()
     assert models.Dataset.from_id(old_dataset.id)
 
     new_dataset = models.Dataset(
         name='dataset',
         license=old_dataset.license,
         description=old_dataset.description,
         acknowledgement=old_dataset.acknowledgement,
+        file_source='./tests/test_data/project_demo.swproj',
     ).force_create(backup=True)
     new_dataset.assert_remote()