Merge pull request #880 from dod-ccpo/to-review-modal

TO Review Submit Modal

tests/routes/task_orders/test_index.py

@@ -1,20 +1,16 @@
+from datetime import date
 from flask import url_for
 import pytest
 from datetime import timedelta, date
 
 from atst.domain.permission_sets import PermissionSets
 from atst.domain.task_orders import TaskOrders
+from atst.models import *
 from atst.models.portfolio_role import Status as PortfolioStatus
+from atst.models.task_order import Status as TaskOrderStatus
 from atst.utils.localization import translate
 
-from tests.factories import (
-    PortfolioFactory,
-    PortfolioRoleFactory,
-    TaskOrderFactory,
-    UserFactory,
-    random_future_date,
-    random_past_date,
-)
+from tests.factories import *
 from tests.utils import captured_templates
 
 
@@ -28,6 +24,49 @@ def user():
     return UserFactory.create()
 
 
+@pytest.fixture
+def task_order():
+    user = UserFactory.create()
+    portfolio = PortfolioFactory.create(owner=user)
+    attachment = Attachment(filename="sample_attachment", object_name="sample")
+
+    return TaskOrderFactory.create(creator=user, portfolio=portfolio)
+
+
+def test_review_task_order(client, user_session, task_order):
+    user_session(task_order.portfolio.owner)
+    response = client.get(
+        url_for("task_orders.review_task_order", task_order_id=task_order.id)
+    )
+    assert response.status_code == 200
+
+
+def test_submit_task_order(client, user_session, task_order):
+    user_session(task_order.portfolio.owner)
+    response = client.post(
+        url_for("task_orders.submit_task_order", task_order_id=task_order.id)
+    )
+    assert response.status_code == 302
+
+    active_start_date = date.today() - timedelta(days=1)
+    active_task_order = TaskOrderFactory(portfolio=task_order.portfolio)
+    CLINFactory(task_order=active_task_order, start_date=active_start_date)
+    assert active_task_order.status == TaskOrderStatus.UNSIGNED
+    response = client.post(
+        url_for("task_orders.submit_task_order", task_order_id=active_task_order.id)
+    )
+    assert active_task_order.status == TaskOrderStatus.ACTIVE
+
+    upcoming_start_date = date.today() + timedelta(days=1)
+    upcoming_task_order = TaskOrderFactory(portfolio=task_order.portfolio)
+    CLINFactory(task_order=upcoming_task_order, start_date=upcoming_start_date)
+    assert upcoming_task_order.status == TaskOrderStatus.UNSIGNED
+    response = client.post(
+        url_for("task_orders.submit_task_order", task_order_id=upcoming_task_order.id)
+    )
+    assert upcoming_task_order.status == TaskOrderStatus.UPCOMING
+
+
 class TestPortfolioFunding:
     @pytest.mark.skip(reason="Update later when CLINs are implemented")
     def test_funded_portfolio(self, app, user_session, portfolio):