145 lines
5.0 KiB
Python
145 lines
5.0 KiB
Python
import urllib
|
|
import pytest
|
|
from flask import url_for
|
|
|
|
from atst.eda_client import MockEDAClient
|
|
|
|
from tests.mocks import MOCK_REQUEST, MOCK_USER
|
|
from tests.factories import PENumberFactory, RequestFactory
|
|
|
|
|
|
class TestPENumberInForm:
|
|
|
|
required_data = {
|
|
"pe_id": "123",
|
|
"task_order_number": MockEDAClient.MOCK_CONTRACT_NUMBER,
|
|
"fname_co": "Contracting",
|
|
"lname_co": "Officer",
|
|
"email_co": "jane@mail.mil",
|
|
"office_co": "WHS",
|
|
"fname_cor": "Officer",
|
|
"lname_cor": "Representative",
|
|
"email_cor": "jane@mail.mil",
|
|
"office_cor": "WHS",
|
|
"uii_ids": "1234",
|
|
"treasury_code": "00123456",
|
|
"ba_code": "02A",
|
|
}
|
|
|
|
def _set_monkeypatches(self, monkeypatch):
|
|
monkeypatch.setattr(
|
|
"atst.forms.financial.FinancialForm.validate", lambda s: True
|
|
)
|
|
monkeypatch.setattr(
|
|
"atst.domain.auth.get_current_user", lambda *args: MOCK_USER
|
|
)
|
|
|
|
def submit_data(self, client, data, extended=False):
|
|
request = RequestFactory.create(body=MOCK_REQUEST.body)
|
|
url_kwargs = {"request_id": request.id}
|
|
if extended:
|
|
url_kwargs["extended"] = True
|
|
response = client.post(
|
|
url_for("requests.financial_verification", **url_kwargs),
|
|
data=data,
|
|
follow_redirects=False,
|
|
)
|
|
return response
|
|
|
|
def test_submit_request_form_with_invalid_pe_id(self, monkeypatch, client):
|
|
self._set_monkeypatches(monkeypatch)
|
|
|
|
response = self.submit_data(client, self.required_data)
|
|
|
|
assert "We couldn't find that PE number" in response.data.decode()
|
|
assert response.status_code == 200
|
|
|
|
def test_submit_request_form_with_unchanged_pe_id(self, monkeypatch, client):
|
|
self._set_monkeypatches(monkeypatch)
|
|
|
|
data = dict(self.required_data)
|
|
data["pe_id"] = MOCK_REQUEST.body["financial_verification"]["pe_id"]
|
|
|
|
response = self.submit_data(client, data)
|
|
|
|
assert response.status_code == 302
|
|
assert "/workspaces" in response.headers.get("Location")
|
|
|
|
def test_submit_request_form_with_new_valid_pe_id(self, monkeypatch, client):
|
|
self._set_monkeypatches(monkeypatch)
|
|
pe = PENumberFactory.create(number="8675309U", description="sample PE number")
|
|
|
|
data = dict(self.required_data)
|
|
data["pe_id"] = pe.number
|
|
|
|
response = self.submit_data(client, data)
|
|
|
|
assert response.status_code == 302
|
|
assert "/workspaces" in response.headers.get("Location")
|
|
|
|
def test_submit_request_form_with_missing_pe_id(self, monkeypatch, client):
|
|
self._set_monkeypatches(monkeypatch)
|
|
|
|
data = dict(self.required_data)
|
|
data["pe_id"] = ""
|
|
|
|
response = self.submit_data(client, data)
|
|
|
|
assert "There were some errors" in response.data.decode()
|
|
assert response.status_code == 200
|
|
|
|
def test_submit_financial_form_with_invalid_task_order(
|
|
self, monkeypatch, user_session, client
|
|
):
|
|
user_session()
|
|
|
|
data = dict(self.required_data)
|
|
data["pe_id"] = MOCK_REQUEST.body["financial_verification"]["pe_id"]
|
|
data["task_order_number"] = "1234"
|
|
|
|
response = self.submit_data(client, data)
|
|
|
|
assert "enter TO information manually" in response.data.decode()
|
|
|
|
def test_submit_financial_form_with_valid_task_order(
|
|
self, monkeypatch, user_session, client
|
|
):
|
|
monkeypatch.setattr("atst.domain.requests.Requests.get", lambda i: MOCK_REQUEST)
|
|
user_session()
|
|
|
|
data = dict(self.required_data)
|
|
data["pe_id"] = MOCK_REQUEST.body["financial_verification"]["pe_id"]
|
|
data["task_order_number"] = MockEDAClient.MOCK_CONTRACT_NUMBER
|
|
|
|
response = self.submit_data(client, data)
|
|
|
|
assert "enter TO information manually" not in response.data.decode()
|
|
|
|
def test_submit_extended_financial_form(
|
|
self, monkeypatch, user_session, client, extended_financial_verification_data
|
|
):
|
|
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["task_order_number"] = "1234567"
|
|
|
|
response = self.submit_data(client, data, extended=True)
|
|
|
|
assert response.status_code == 302
|
|
assert "/projects/new" in response.headers.get("Location")
|
|
|
|
def test_submit_invalid_extended_financial_form(
|
|
self, monkeypatch, user_session, client, extended_financial_verification_data
|
|
):
|
|
monkeypatch.setattr("atst.forms.financial.validate_pe_id", lambda *args: True)
|
|
user_session()
|
|
data = {**self.required_data, **extended_financial_verification_data}
|
|
data["task_order_number"] = "1234567"
|
|
del (data["clin_0001"])
|
|
|
|
response = self.submit_data(client, data, extended=True)
|
|
|
|
assert response.status_code == 200
|