diff --git a/tests/domain/test_requests.py b/tests/domain/test_requests.py index afcb49c1..7d4e4f3e 100644 --- a/tests/domain/test_requests.py +++ b/tests/domain/test_requests.py @@ -134,6 +134,7 @@ def test_update_financial_verification_without_task_order( extended_financial_verification_data["clin_0001"] ) assert request.task_order.source == TaskOrderSource.MANUAL + assert request.task_order.pdf def test_update_financial_verification_with_task_order(): diff --git a/tests/domain/test_task_orders.py b/tests/domain/test_task_orders.py index f1203938..2e86a2c0 100644 --- a/tests/domain/test_task_orders.py +++ b/tests/domain/test_task_orders.py @@ -36,3 +36,9 @@ def test_nonexistent_task_order_raises_with_client(monkeypatch): ) with pytest.raises(NotFoundError): 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 diff --git a/tests/routes/test_financial_verification.py b/tests/routes/test_financial_verification.py index 6bd235b1..631b6743 100644 --- a/tests/routes/test_financial_verification.py +++ b/tests/routes/test_financial_verification.py @@ -119,10 +119,11 @@ class TestPENumberInForm: def test_submit_extended_financial_form( 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() data = {**self.required_data, **extended_financial_verification_data} - data["pe_id"] = MOCK_REQUEST.body["financial_verification"]["pe_id"] data["task_order_number"] = "1234567" response = self.submit_data(client, data, extended=True)