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