Migrate zarr tests

dandiapi/api/tests/factories.py

@@ -19,7 +19,6 @@
     UserMetadata,
     Version,
 )
-from dandiapi.zarr.models import ZarrArchive, ZarrUploadFile
 
 
 class UserMetadataFactory(factory.django.DjangoModelFactory):
@@ -234,38 +233,3 @@ def etag(self):
 class EmbargoedUploadFactory(UploadFactory):
     class Meta:
         model = EmbargoedUpload
-
-
-class ZarrArchiveFactory(factory.django.DjangoModelFactory):
-    class Meta:
-        model = ZarrArchive
-
-    zarr_id = factory.Faker('uuid4')
-    name = factory.Faker('catch_phrase')
-    dandiset = factory.SubFactory(DandisetFactory)
-
-
-class ZarrUploadFileFactory(factory.django.DjangoModelFactory):
-    class Meta:
-        model = ZarrUploadFile
-
-    zarr_archive = factory.SubFactory(ZarrArchiveFactory)
-
-    @factory.lazy_attribute
-    def path(self):
-        # No / prefix
-        return faker.Faker().file_path(extension='nwb')[1:]
-
-    @factory.lazy_attribute
-    def blob(self):
-        return django_files.File(
-            django_files.base.ContentFile(faker.Faker().binary(length=100)).file,
-            self.zarr_archive.s3_path(self.path),
-        )
-
-    @factory.lazy_attribute
-    def etag(self):
-        h = hashlib.md5()
-        h.update(self.blob.read())
-        self.blob.seek(0)
-        return h.hexdigest()

dandiapi/api/tests/conftest.py → dandiapi/conftest.py

@@ -7,8 +7,7 @@
 from storages.backends.s3boto3 import S3Boto3Storage
 
 from dandiapi.api.storage import create_s3_storage
-
-from .factories import (
+from dandiapi.api.tests.factories import (
     AssetBlobFactory,
     DandisetFactory,
     DraftAssetFactory,
@@ -20,9 +19,8 @@
     SocialAccountFactory,
     UploadFactory,
     UserFactory,
-    ZarrArchiveFactory,
-    ZarrUploadFileFactory,
 )
+from dandiapi.zarr.tests.factories import ZarrArchiveFactory, ZarrUploadFileFactory
 
 register(PublishedAssetFactory, _name='published_asset')
 register(DraftAssetFactory, _name='draft_asset')
@@ -38,6 +36,8 @@
 register(UserFactory)
 register(SocialAccountFactory)
 register(UploadFactory)
+
+# zarr app
 register(ZarrArchiveFactory)
 register(ZarrUploadFileFactory)
 

dandiapi/zarr/tests/factories.py

@@ -0,0 +1,43 @@
+import hashlib
+
+from django.core import files as django_files
+import factory
+import faker
+
+from dandiapi.api.tests.factories import DandisetFactory
+from dandiapi.zarr.models import ZarrArchive, ZarrUploadFile
+
+
+class ZarrArchiveFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = ZarrArchive
+
+    zarr_id = factory.Faker('uuid4')
+    name = factory.Faker('catch_phrase')
+    dandiset = factory.SubFactory(DandisetFactory)
+
+
+class ZarrUploadFileFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = ZarrUploadFile
+
+    zarr_archive = factory.SubFactory(ZarrArchiveFactory)
+
+    @factory.lazy_attribute
+    def path(self):
+        # No / prefix
+        return faker.Faker().file_path(extension='nwb')[1:]
+
+    @factory.lazy_attribute
+    def blob(self):
+        return django_files.File(
+            django_files.base.ContentFile(faker.Faker().binary(length=100)).file,
+            self.zarr_archive.s3_path(self.path),
+        )
+
+    @factory.lazy_attribute
+    def etag(self):
+        h = hashlib.md5()
+        h.update(self.blob.read())
+        self.blob.seek(0)
+        return h.hexdigest()