Pass in task_order through test arguments

This commit is contained in:
George Drummond 2019-02-18 09:58:42 -05:00
parent 7d4e14dff2
commit 13999d7c60
No known key found for this signature in database
GPG Key ID: 296DD6077123BF17

View File

@ -102,9 +102,8 @@ def test_create_new_task_order_for_portfolio(client, user_session):
assert created_task_order.portfolio == portfolio assert created_task_order.portfolio == portfolio
def test_task_order_form_shows_errors(client, user_session): def test_task_order_form_shows_errors(client, user_session, task_order):
to = task_order() creator = task_order.creator
creator = to.creator
user_session(creator) user_session(creator)
task_order_data = TaskOrderFactory.dictionary() task_order_data = TaskOrderFactory.dictionary()
@ -113,7 +112,7 @@ def test_task_order_form_shows_errors(client, user_session):
funding_data.update({"clin_01": "one milllllion dollars"}) funding_data.update({"clin_01": "one milllllion dollars"})
response = client.post( response = client.post(
url_for("task_orders.update", screen=2, task_order_id=to.id), url_for("task_orders.update", screen=2, task_order_id=task_order.id),
data=funding_data, data=funding_data,
follow_redirects=False, follow_redirects=False,
) )
@ -123,9 +122,8 @@ def test_task_order_form_shows_errors(client, user_session):
assert "Not a valid decimal" in body assert "Not a valid decimal" in body
def test_task_order_validates_email_address(client, user_session): def test_task_order_validates_email_address(client, user_session, task_order):
to = task_order() creator = task_order.creator
creator = to.creator
user_session(creator) user_session(creator)
task_order_data = TaskOrderFactory.dictionary() task_order_data = TaskOrderFactory.dictionary()
@ -133,7 +131,7 @@ def test_task_order_validates_email_address(client, user_session):
oversight_data.update({"ko_email": "not an email"}) oversight_data.update({"ko_email": "not an email"})
response = client.post( response = client.post(
url_for("task_orders.update", screen=3, task_order_id=to.id), url_for("task_orders.update", screen=3, task_order_id=task_order.id),
data=oversight_data, data=oversight_data,
follow_redirects=False, follow_redirects=False,
) )
@ -143,21 +141,23 @@ def test_task_order_validates_email_address(client, user_session):
assert "Invalid email" in body assert "Invalid email" in body
def test_review_screen_when_all_sections_complete(client, user_session): def test_review_screen_when_all_sections_complete(client, user_session, task_order):
to = task_order() user_session(task_order.creator)
user_session(to.creator) response = client.get(
response = client.get(url_for("task_orders.new", screen=4, task_order_id=to.id)) url_for("task_orders.new", screen=4, task_order_id=task_order.id)
)
body = response.data.decode() body = response.data.decode()
assert translate("task_orders.form.draft_alert_title") not in body assert translate("task_orders.form.draft_alert_title") not in body
assert response.status_code == 200 assert response.status_code == 200
def test_review_screen_when_not_all_sections_complete(client, user_session): def test_review_screen_when_not_all_sections_complete(client, user_session, task_order):
to = task_order() TaskOrders.update(task_order.creator, task_order, clin_01=None)
TaskOrders.update(to.creator, to, clin_01=None) user_session(task_order.creator)
user_session(to.creator) response = client.get(
response = client.get(url_for("task_orders.new", screen=4, task_order_id=to.id)) url_for("task_orders.new", screen=4, task_order_id=task_order.id)
)
body = response.data.decode() body = response.data.decode()
assert translate("task_orders.form.draft_alert_title") in body assert translate("task_orders.form.draft_alert_title") in body