From c9451e0f93b889f520f6ecf6272b191803014ce5 Mon Sep 17 00:00:00 2001 From: leigh-mil Date: Tue, 30 Jul 2019 15:26:07 -0400 Subject: [PATCH] Update tests to use proper route Pass portfolio and creator to TOFactory so user has perms to edit TO --- tests/routes/task_orders/test_index.py | 5 +---- tests/routes/task_orders/test_new.py | 6 ++++-- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/routes/task_orders/test_index.py b/tests/routes/task_orders/test_index.py index 33a75cd8..c800ce25 100644 --- a/tests/routes/task_orders/test_index.py +++ b/tests/routes/task_orders/test_index.py @@ -51,7 +51,4 @@ def test_review_task_order_draft(client, user_session, task_order): url_for("task_orders.review_task_order", task_order_id=task_order.id) ) assert response.status_code == 302 - assert ( - url_for("task_orders.form_step_one_add_pdf", task_order_id=task_order.id) - in response.location - ) + assert url_for("task_orders.edit", task_order_id=task_order.id) in response.location diff --git a/tests/routes/task_orders/test_new.py b/tests/routes/task_orders/test_new.py index feef2844..b88bbf7b 100644 --- a/tests/routes/task_orders/test_new.py +++ b/tests/routes/task_orders/test_new.py @@ -307,7 +307,7 @@ def test_task_orders_submit_task_order(client, user_session, task_order): @pytest.mark.parametrize( "to_factory_args,expected_step", [ - ({"pdf": None, "number": "", "clins": []}, "step_1"), + ({"_pdf": None, "number": "", "clins": []}, "step_1"), ({"number": "", "clins": []}, "step_2"), ({"number": "1234567890123", "clins": []}, "step_3"), ({"number": "1234567890123", "create_clins": [1]}, "step_4"), @@ -316,7 +316,9 @@ def test_task_orders_submit_task_order(client, user_session, task_order): def test_task_orders_edit_redirects_to_latest_incomplete_step( client, user_session, portfolio, to_factory_args, expected_step ): - task_order = TaskOrderFactory.create(**to_factory_args) + task_order = TaskOrderFactory.create( + portfolio=portfolio, creator=portfolio.owner, **to_factory_args + ) user_session(portfolio.owner) response = client.get(url_for("task_orders.edit", task_order_id=task_order.id))