tweaks - added factories for CAFFileStore and DocumentFile

1 files changed, 20 insertions, 8 deletions

diff --git a/ctrack/caf/tests/factories.py b/ctrack/caf/tests/factories.py
index bae0d9c..ce337f2 100644
--- a/ctrack/caf/tests/factories.py
+++ b/ctrack/caf/tests/factories.py
@@ -1,9 +1,10 @@
-import factory
+import random
 
+import factory
 from factory import Faker
 
-from ctrack.caf.models import CAF, EssentialService, Grading
-from ctrack.organisations.tests.factories import PersonFactory
+from ctrack.caf.models import EssentialService, Grading, DocumentFile, FileStore, CAFFileStore
+from ctrack.organisations.tests.factories import OrganisationFactory
 
 
 class EssentialServiceFactory(factory.DjangoModelFactory):
@@ -22,10 +23,21 @@ class GradingFactory(factory.DjangoModelFactory):
         model = Grading
 
 
-class CAFFactory(factory.DjangoModelFactory):
-    """Factory for CAFs."""
-    owner = factory.SubFactory(PersonFactory)
-#   triage_ranking = factory.SubFactory(TriageRankingFactory)
+# TODO: test these two factories
+class CAFFileStoreFactory(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
+
+
+class DocumentFileFactory(factory.DjangoModelFactory):
+    name = Faker("file_name", extension="xlsx")
+    type = random.choice([1, 2, 3, 4])
+    file_store_location = factory.SubFactory(CAFFileStoreFactory)
 
     class Meta:
-        model = CAF
+        model = DocumentFile