diff options
Diffstat (limited to '')
| -rw-r--r-- | ctrack/caf/tests/factories.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/ctrack/caf/tests/factories.py b/ctrack/caf/tests/factories.py index ce337f2..53953fe 100644 --- a/ctrack/caf/tests/factories.py +++ b/ctrack/caf/tests/factories.py @@ -3,7 +3,7 @@ import random import factory from factory import Faker -from ctrack.caf.models import EssentialService, Grading, DocumentFile, FileStore, CAFFileStore +from ctrack.caf.models import EssentialService, Grading, DocumentFile, FileStore from ctrack.organisations.tests.factories import OrganisationFactory @@ -23,21 +23,20 @@ class GradingFactory(factory.DjangoModelFactory): model = Grading -# TODO: test these two factories -class CAFFileStoreFactory(factory.DjangoModelFactory): +class FileStoreFactory(factory.DjangoModelFactory): descriptor = "File Store X" virtual_location = Faker("street_name") physical_location = random.choice(["Cupboard A", "Tin Box", "The Vault"]) physical_location_organisation = factory.SubFactory(OrganisationFactory) class Meta: - model = CAFFileStore + model = FileStore class DocumentFileFactory(factory.DjangoModelFactory): name = Faker("file_name", extension="xlsx") type = random.choice([1, 2, 3, 4]) - file_store_location = factory.SubFactory(CAFFileStoreFactory) + file_store_location = factory.SubFactory(FileStoreFactory) class Meta: model = DocumentFile |