Ensure draft version exists on dandiset created from zarr factory

dandi · Dec 17, 2024 · eb6fb85 · eb6fb85
1 parent 38a90fb
commit eb6fb85
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/dandiapi/zarr/tests/factories.py b/dandiapi/zarr/tests/factories.py
@@ -14,6 +14,15 @@ class Meta:
     name = factory.Faker('catch_phrase')
     dandiset = factory.SubFactory(DandisetFactory)
 
+    @factory.post_generation
+    def ensure_draft_version(obj: ZarrArchive, *args, **kwargs):  # type: ignore  # noqa: N805, PGH003
+        from dandiapi.api.tests.factories import DraftVersionFactory
+
+        if obj.dandiset.versions.filter(version='draft').exists():
+            return
+
+        DraftVersionFactory(dandiset=obj.dandiset)
+
 
 class EmbargoedZarrArchiveFactory(ZarrArchiveFactory):
     embargoed = True