pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
richard-dds 2019-07-30 09:53:50 -04:00
parent 59545aaf0e
commit 88853e352c
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/routes/task_orders/new.py
View File

@ -134,7 +134,9 @@ def cancel_edit(task_order_id):
task_order = TaskOrders.update(task_order_id, **form.data) task_order = TaskOrders.update(task_order_id, **form.data)
portfolio_id = task_order.portfolio_id portfolio_id = task_order.portfolio_id
else: 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)) return redirect(url_for("task_orders.portfolio_funding", portfolio_id=portfolio_id))

12
tests/routes/task_orders/test_new.py
View File

@ -6,12 +6,7 @@ from atst.domain.task_orders import TaskOrders
from atst.models.task_order import Status as TaskOrderStatus from atst.models.task_order import Status as TaskOrderStatus
from atst.models import TaskOrder from atst.models import TaskOrder
from tests.factories import ( from tests.factories import CLINFactory, PortfolioFactory, TaskOrderFactory, UserFactory
CLINFactory,
PortfolioFactory,
TaskOrderFactory,
UserFactory,
)
@pytest.fixture @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): def test_can_cancel_edit_and_save_task_order(client, user_session, task_order, session):
user_session(task_order.portfolio.owner) 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 assert response.status_code == 302
updated_task_order = session.query(TaskOrder).get(task_order.id) updated_task_order = session.query(TaskOrder).get(task_order.id)