diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py index 9d55ed02..fc4535b8 100644 --- a/atst/routes/task_orders/new.py +++ b/atst/routes/task_orders/new.py @@ -134,7 +134,9 @@ def cancel_edit(task_order_id): task_order = TaskOrders.update(task_order_id, **form.data) portfolio_id = task_order.portfolio_id else: - task_order = TaskOrders.create(g.current_user, portfolio_id, **form.data) + task_order = TaskOrders.create( + g.current_user, portfolio_id, **form.data + ) return redirect(url_for("task_orders.portfolio_funding", portfolio_id=portfolio_id)) diff --git a/tests/routes/task_orders/test_new.py b/tests/routes/task_orders/test_new.py index 1ed515bd..90e34209 100644 --- a/tests/routes/task_orders/test_new.py +++ b/tests/routes/task_orders/test_new.py @@ -6,12 +6,7 @@ from atst.domain.task_orders import TaskOrders from atst.models.task_order import Status as TaskOrderStatus from atst.models import TaskOrder -from tests.factories import ( - CLINFactory, - PortfolioFactory, - TaskOrderFactory, - UserFactory, -) +from tests.factories import CLINFactory, PortfolioFactory, TaskOrderFactory, UserFactory @pytest.fixture @@ -325,7 +320,10 @@ def test_task_orders_edit_redirects_to_latest_incomplete_step( def test_can_cancel_edit_and_save_task_order(client, user_session, task_order, session): user_session(task_order.portfolio.owner) - response = client.post(url_for("task_orders.cancel_edit", task_order_id=task_order.id, save=True), data={"number": "0123456789012"}) + response = client.post( + url_for("task_orders.cancel_edit", task_order_id=task_order.id, save=True), + data={"number": "0123456789012"}, + ) assert response.status_code == 302 updated_task_order = session.query(TaskOrder).get(task_order.id)