Add active & expired task order statuses

tests/factories.py

@@ -1,3 +1,4 @@
+import operator
 import random
 import string
 import factory
@@ -41,14 +42,22 @@ def random_phone_number():
     return "".join(random.choices(string.digits, k=10))
 
 
+def random_past_date(year_min=1, year_max=5):
+    return _random_date(year_min, year_max, operator.sub)
+
+
 def random_future_date(year_min=1, year_max=5):
+    return _random_date(year_min, year_max, operator.add)
+
+
+def _random_date(year_min, year_max, operation):
     if year_min == year_max:
         inc = year_min
     else:
         inc = random.randrange(year_min, year_max)
 
     return datetime.date(
-        datetime.date.today().year + inc,
+        operation(datetime.date.today().year, inc),
         random.randrange(1, 12),
         random.randrange(1, 28),
     )

tests/models/test_task_order.py

@@ -1,9 +1,37 @@
 from atst.models.task_order import TaskOrder, Status
 
+from tests.factories import random_future_date, random_past_date
 
-def test_default_status():
+
+class TestTaskOrderStatus:
+
+    def test_pending_status(self):
+        to = TaskOrder()
+        assert to.status == Status.PENDING
+
+        to = TaskOrder(number='42', start_date=random_future_date())
+        assert to.status == Status.PENDING
+
+    def test_active_status(self):
+        to = TaskOrder(
+            number='42',
+            start_date=random_past_date(),
+            end_date=random_future_date(),
+        )
+        assert to.status == Status.ACTIVE
+
+    def test_expired_status(self):
+        to = TaskOrder(
+            number='42',
+            start_date=random_past_date(),
+            end_date=random_past_date(),
+        )
+        assert to.status == Status.EXPIRED
+
+
+def test_is_submitted():
     to = TaskOrder()
-    assert to.status == Status.PENDING
+    assert not to.is_submitted
 
-    with_args = TaskOrder(number="42")
-    assert to.status == Status.PENDING
+    to = TaskOrder(number='42')
+    assert to.is_submitted