Test that the pdf is attached

This commit is contained in:
Montana 2018-08-27 12:15:05 -04:00
parent cfe9b7589e
commit 6df897191c
3 changed files with 10 additions and 2 deletions

View File

@ -134,6 +134,7 @@ def test_update_financial_verification_without_task_order(
extended_financial_verification_data["clin_0001"] extended_financial_verification_data["clin_0001"]
) )
assert request.task_order.source == TaskOrderSource.MANUAL assert request.task_order.source == TaskOrderSource.MANUAL
assert request.task_order.pdf
def test_update_financial_verification_with_task_order(): def test_update_financial_verification_with_task_order():

View File

@ -36,3 +36,9 @@ def test_nonexistent_task_order_raises_with_client(monkeypatch):
) )
with pytest.raises(NotFoundError): with pytest.raises(NotFoundError):
TaskOrders.get("some other fake numer") TaskOrders.get("some other fake numer")
def test_create_attachment(extended_financial_verification_data):
task_order_data = extended_financial_verification_data.copy()
task_order_data["pdf"] = task_order_data.pop("task_order")
task_order = TaskOrders.get_or_create_task_order("abc123", task_order_data)
assert task_order.pdf

View File

@ -119,10 +119,11 @@ class TestPENumberInForm:
def test_submit_extended_financial_form( def test_submit_extended_financial_form(
self, monkeypatch, user_session, client, extended_financial_verification_data self, monkeypatch, user_session, client, extended_financial_verification_data
): ):
monkeypatch.setattr("atst.domain.requests.Requests.get", lambda i: MOCK_REQUEST) request = RequestFactory.create()
monkeypatch.setattr("atst.domain.requests.Requests.get", lambda i: request)
monkeypatch.setattr("atst.forms.financial.validate_pe_id", lambda *args: True)
user_session() user_session()
data = {**self.required_data, **extended_financial_verification_data} data = {**self.required_data, **extended_financial_verification_data}
data["pe_id"] = MOCK_REQUEST.body["financial_verification"]["pe_id"]
data["task_order_number"] = "1234567" data["task_order_number"] = "1234567"
response = self.submit_data(client, data, extended=True) response = self.submit_data(client, data, extended=True)