atst/tests/models/test_task_order.py
2019-01-15 10:54:39 -05:00

33 lines
897 B
Python

from atst.models.task_order import TaskOrder, Status
from tests.factories import random_future_date, random_past_date
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 not to.is_submitted
to = TaskOrder(number="42")
assert to.is_submitted