131 lines
3.2 KiB
Python
131 lines
3.2 KiB
Python
import pytest
|
|
|
|
from atst.eda_client import MockEDAClient
|
|
from atst.routes.requests.financial_verification import UpdateFinancialVerification
|
|
|
|
from tests.mocks import MOCK_REQUEST, MOCK_USER, MOCK_VALID_PE_ID
|
|
from tests.factories import (
|
|
PENumberFactory,
|
|
RequestFactory,
|
|
UserFactory,
|
|
RequestStatusEventFactory,
|
|
RequestReviewFactory,
|
|
)
|
|
from atst.forms.exceptions import FormValidationError
|
|
from atst.domain.requests.financial_verification import (
|
|
PENumberValidator,
|
|
TaskOrderNumberValidator,
|
|
)
|
|
|
|
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",
|
|
}
|
|
|
|
|
|
class MockPEValidator(object):
|
|
def validate(self, request, field):
|
|
return True
|
|
|
|
|
|
class MockTaskOrderValidator(object):
|
|
def validate(self, field):
|
|
return True
|
|
|
|
|
|
def test_update():
|
|
request = RequestFactory.create()
|
|
user = UserFactory.create()
|
|
data = {**required_data, "pe_id": MOCK_VALID_PE_ID}
|
|
|
|
response_context = UpdateFinancialVerification(
|
|
MockPEValidator(),
|
|
MockTaskOrderValidator(),
|
|
user,
|
|
request,
|
|
data,
|
|
is_extended=False,
|
|
).execute()
|
|
|
|
assert response_context.get("workspace")
|
|
|
|
|
|
def test_re_enter_pe_number():
|
|
request = RequestFactory.create()
|
|
user = UserFactory.create()
|
|
data = {**required_data, "pe_id": "0101228M"}
|
|
update_fv = UpdateFinancialVerification(
|
|
PENumberValidator(),
|
|
MockTaskOrderValidator(),
|
|
user,
|
|
request,
|
|
data,
|
|
is_extended=False,
|
|
)
|
|
|
|
with pytest.raises(FormValidationError):
|
|
update_fv.execute()
|
|
response_context = update_fv.execute()
|
|
|
|
assert response_context.get("status", "submitted")
|
|
|
|
|
|
def test_invalid_task_order_number():
|
|
request = RequestFactory.create()
|
|
user = UserFactory.create()
|
|
data = {**required_data, "task_order_number": "DCA10096D0051"}
|
|
update_fv = UpdateFinancialVerification(
|
|
MockPEValidator(),
|
|
TaskOrderNumberValidator(),
|
|
user,
|
|
request,
|
|
data,
|
|
is_extended=False,
|
|
)
|
|
|
|
with pytest.raises(FormValidationError):
|
|
update_fv.execute()
|
|
|
|
|
|
def test_extended_fv_data(extended_financial_verification_data):
|
|
request = RequestFactory.create()
|
|
user = UserFactory.create()
|
|
data = {**required_data, **extended_financial_verification_data}
|
|
update_fv = UpdateFinancialVerification(
|
|
MockPEValidator(),
|
|
TaskOrderNumberValidator(),
|
|
user,
|
|
request,
|
|
data,
|
|
is_extended=True,
|
|
)
|
|
|
|
assert update_fv.execute()
|
|
|
|
|
|
def test_missing_extended_fv_data():
|
|
request = RequestFactory.create()
|
|
user = UserFactory.create()
|
|
update_fv = UpdateFinancialVerification(
|
|
MockPEValidator(),
|
|
TaskOrderNumberValidator(),
|
|
user,
|
|
request,
|
|
required_data,
|
|
is_extended=True,
|
|
)
|
|
|
|
with pytest.raises(FormValidationError):
|
|
update_fv.execute()
|