import factory
|
from core.utils.common import load_func
|
from django.conf import settings
|
from io_storages.base_models import ImportStorage, ProjectStorageMixin
|
from io_storages.models import (
|
AzureBlobExportStorage,
|
AzureBlobImportStorage,
|
GCSImportStorage,
|
RedisImportStorage,
|
S3ImportStorage,
|
)
|
|
|
class StorageFactory(factory.django.DjangoModelFactory):
|
title = factory.Faker('bs')
|
description = factory.Faker('paragraph')
|
|
class Meta:
|
model = ImportStorage
|
abstract = True
|
|
|
class ImportStorageFactory(StorageFactory):
|
class Meta:
|
model = ImportStorage
|
abstract = True
|
|
|
class ProjectStorageMixinFactory(factory.django.DjangoModelFactory):
|
project = factory.SubFactory(load_func(settings.PROJECT_FACTORY))
|
|
class Meta:
|
model = ProjectStorageMixin
|
abstract = True
|
|
|
class AzureBlobStorageMixinFactory(factory.django.DjangoModelFactory):
|
account_name = factory.Faker('word')
|
account_key = factory.Faker('word')
|
|
class Meta:
|
abstract = True
|
|
|
class AzureBlobImportStorageBaseFactory(AzureBlobStorageMixinFactory, ImportStorageFactory):
|
class Meta:
|
model = AzureBlobImportStorage
|
abstract = True
|
|
|
class AzureBlobImportStorageFactory(AzureBlobImportStorageBaseFactory, ProjectStorageMixinFactory):
|
class Meta:
|
model = AzureBlobImportStorage
|
|
|
class S3StorageMixinFactory(factory.django.DjangoModelFactory):
|
class Meta:
|
abstract = True
|
|
|
class S3ImportStorageBaseFactory(S3StorageMixinFactory, ImportStorageFactory):
|
class Meta:
|
model = S3ImportStorage
|
abstract = True
|
|
|
class S3ImportStorageFactory(S3ImportStorageBaseFactory, ProjectStorageMixinFactory):
|
class Meta:
|
model = S3ImportStorage
|
|
|
class GCSStorageMixinFactory(factory.django.DjangoModelFactory):
|
class Meta:
|
abstract = True
|
|
|
class GCSImportStorageBaseFactory(GCSStorageMixinFactory, ImportStorageFactory):
|
class Meta:
|
model = GCSImportStorage
|
abstract = True
|
|
|
class GCSImportStorageFactory(GCSImportStorageBaseFactory, ProjectStorageMixinFactory):
|
class Meta:
|
model = GCSImportStorage
|
|
|
class RedisStorageMixinFactory(factory.django.DjangoModelFactory):
|
class Meta:
|
abstract = True
|
|
|
class RedisImportStorageBaseFactory(RedisStorageMixinFactory, ImportStorageFactory):
|
class Meta:
|
model = RedisImportStorage
|
abstract = True
|
|
|
class RedisImportStorageFactory(RedisImportStorageBaseFactory, ProjectStorageMixinFactory):
|
class Meta:
|
model = RedisImportStorage
|
|
|
class AzureBlobExportStorageFactory(factory.django.DjangoModelFactory):
|
class Meta:
|
model = AzureBlobExportStorage
|
