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

Add routes for first step of TO builder

Browse Source
This commit is contained in:
leigh-mil
2019-07-15 16:42:15 -04:00
parent c8d8b7cce2
commit 45390835ed
8 changed files with 147 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -33,36 +33,23 @@ def user():
return UserFactory.create()
def test_task_orders_edit(client, user_session, portfolio):
def test_task_orders_add_to_pdf(client, user_session, portfolio):
user_session(portfolio.owner)
response = client.get(url_for("task_orders.edit", portfolio_id=portfolio.id))
response = client.get(url_for("task_orders.add_to_pdf", portfolio_id=portfolio.id))
assert response.status_code == 200
def test_task_orders_update(client, user_session, portfolio):
def test_task_orders_upload_to_pdf(
client, user_session, portfolio, pdf_upload, session
):
user_session(portfolio.owner)
form_data = {
"number": "0123456789",
"pdf": pdf_upload,
"clins-0-jedi_clin_type": "JEDI_CLIN_1",
"clins-0-clin_number": "12312",
"clins-0-start_date": "01/01/2020",
"clins-0-end_date": "01/01/2021",
"clins-0-obligated_amount": "5000",
"clins-0-loas-0": "123123123123",
"clins-0-loas-1": "345345234",
"clins-1-jedi_clin_type": "JEDI_CLIN_1",
"clins-1-number": "12312",
"clins-1-start_date": "01/01/2020",
"clins-1-end_date": "01/01/2021",
"clins-1-obligated_amount": "5000",
"clins-1-loas-0": "78979087",
}
form_data = {"pdf": pdf_upload}
response = client.post(
url_for("task_orders.update", portfolio_id=portfolio.id), data=form_data
url_for("task_orders.upload_to_pdf", portfolio_id=portfolio.id), data=form_data
)
assert response.status_code == 302
task_order = session.query(TaskOrder).filter_by(number=data["number"]).one()
task_order = portfolio.task_orders[0]
assert task_order.pdf.filename == pdf_upload.filename
@@ -84,12 +71,27 @@ def test_task_orders_save_incomplete(client, user_session, portfolio):
assert response.location == expected_url
def test_task_orders_edit_existing_to(client, user_session, task_order):
def test_task_orders_add_to_pdf_existing_to(client, user_session, task_order):
user_session(task_order.creator)
response = client.get(url_for("task_orders.edit", task_order_id=task_order.id))
response = client.get(
url_for("task_orders.add_to_pdf", task_order_id=task_order.id)
)
assert response.status_code == 200
def test_task_orders_upload_to_pdf_existing_to(
client, user_session, task_order, pdf_upload
):
user_session(task_order.creator)
form_data = {"pdf": pdf_upload}
response = client.post(
url_for("task_orders.upload_to_pdf", task_order_id=task_order.id),
data=form_data,
)
assert response.status_code == 302
assert task_order.pdf.filename == pdf_upload.filename
def test_task_orders_update_existing_to(client, user_session, task_order):
user_session(task_order.creator)
form_data = {
@@ -108,6 +110,7 @@ def test_task_orders_update_existing_to(client, user_session, task_order):
assert task_order.number == "0123456789"
@pytest.mark.skip(reason="Reevaluate how form handles invalid data")
def test_task_orders_update_invalid_data(client, user_session, portfolio):
user_session(portfolio.owner)
num_task_orders = len(portfolio.task_orders)