TaskOrders gets EDA client from current Flask app

This commit is contained in:
dandds
2018-08-21 14:02:01 -04:00
parent a50b855389
commit 1a9606f2b8
5 changed files with 21 additions and 20 deletions

View File

@@ -14,8 +14,9 @@ def test_can_get_task_order():
assert to.id == to.id
def test_can_get_task_order_from_eda():
to = TaskOrders.get(MockEDAClient.MOCK_CONTRACT_NUMBER, client=MockEDAClient())
def test_can_get_task_order_from_eda(monkeypatch):
monkeypatch.setattr("atst.domain.task_orders.TaskOrders._client", lambda: MockEDAClient())
to = TaskOrders.get(MockEDAClient.MOCK_CONTRACT_NUMBER)
assert to.number == MockEDAClient.MOCK_CONTRACT_NUMBER
@@ -25,6 +26,7 @@ def test_nonexistent_task_order_raises_without_client():
TaskOrders.get("some fake number")
def test_nonexistent_task_order_raises_with_client():
def test_nonexistent_task_order_raises_with_client(monkeypatch):
monkeypatch.setattr("atst.domain.task_orders.TaskOrders._client", lambda: MockEDAClient())
with pytest.raises(NotFoundError):
TaskOrders.get("some other fake numer", client=MockEDAClient())
TaskOrders.get("some other fake numer")

View File

@@ -69,8 +69,9 @@ def test_ba_code_validation(input_, expected):
assert is_valid == expected
def test_task_order_number_validation():
form_invalid = FinancialForm(data={"task_order_number": "1234"}, eda_client=MockEDAClient())
def test_task_order_number_validation(monkeypatch):
monkeypatch.setattr("atst.domain.task_orders.TaskOrders._client", lambda: MockEDAClient())
form_invalid = FinancialForm(data={"task_order_number": "1234"})
form_invalid.validate()
assert "task_order_number" in form_invalid.errors