Added new factories for Caf app.

2 files changed, 17 insertions, 0 deletions

diff --git a/ctrack/caf/tests/__init__.py b/ctrack/caf/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/ctrack/caf/tests/__init__.py
diff --git a/ctrack/caf/tests/factories.py b/ctrack/caf/tests/factories.py
new file mode 100644
index 0000000..ec02bce
--- /dev/null
+++ b/ctrack/caf/tests/factories.py
@@ -0,0 +1,17 @@
+import factory
+
+from ctrack.caf.models import CAF, EssentialService
+from ctrack.organisations.tests.factories import PersonFactory
+
+
+class EssentialServiceFactory(factory.DjangoModelFactory):
+    """Factory for Essential Services."""
+    model = EssentialService
+
+
+class CAFFactory(factory.DjangoModelFactory):
+    """Factory for CAFs."""
+    model = CAF
+    owner = factory.SubFactory(PersonFactory)
+#   triage_ranking = factory.SubFactory(TriageRankingFactory)
+# TODO - we want an abstract Ranking