From 33248daa6b23a26a787c7a71a4ec6e6a6c219b86 Mon Sep 17 00:00:00 2001 From: Montana Date: Mon, 10 Jun 2019 14:40:12 -0400 Subject: [PATCH] Add route tests --- atst/routes/task_orders/index.py | 1 - tests/routes/task_orders/test_index.py | 25 ++++++++++++++++++++++++- 2 files changed, 24 insertions(+), 2 deletions(-) diff --git a/atst/routes/task_orders/index.py b/atst/routes/task_orders/index.py index 63e8a5cf..5922aeb7 100644 --- a/atst/routes/task_orders/index.py +++ b/atst/routes/task_orders/index.py @@ -39,7 +39,6 @@ def review_task_order(task_order_id): ) -# TODO write test, verify permission @task_orders_bp.route("/task_orders//submit", methods=["POST"]) @user_can(Permissions.CREATE_TASK_ORDER, "submit task order") def submit_task_order(task_order_id): diff --git a/tests/routes/task_orders/test_index.py b/tests/routes/task_orders/test_index.py index c1b435fa..fbc830bb 100644 --- a/tests/routes/task_orders/test_index.py +++ b/tests/routes/task_orders/test_index.py @@ -4,6 +4,7 @@ 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.utils.localization import translate @@ -22,11 +23,33 @@ from tests.utils import captured_templates def portfolio(): return PortfolioFactory.create() - @pytest.fixture 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 + class TestPortfolioFunding: @pytest.mark.skip(reason="Update later when CLINs are implemented")