Add unit tests

2019-06-05 09:40:33 -04:00
parent e4cbe892fc
commit f462c3bd5e
2 changed files with 47 additions and 2 deletions
--- a/tests/factories.py
+++ b/tests/factories.py
@@ -269,6 +269,18 @@ class TaskOrderFactory(Base):
    number = factory.LazyFunction(random_task_order_number)


+class CLINFactory(Base):
+    class Meta:
+        model = CLIN
+
+    task_order = factory.SubFactory(TaskOrderFactory)
+    number = factory.LazyFunction(random_task_order_number)
+    start_date = datetime.date.today()
+    end_date = factory.LazyFunction(random_future_date)
+    obligated_amount = random.randint(100, 999999)
+    jedi_clin_type = random.choice([e.value for e in clin.JEDICLINType])
+
+
 class NotificationRecipientFactory(Base):
    class Meta:
        model = NotificationRecipient