TaskOrders domain class can query EDA api via client

This commit is contained in:
dandds
2018-08-20 16:09:08 -04:00
parent 735b9b5179
commit 42325d5a28
3 changed files with 38 additions and 4 deletions

View File

@@ -2,6 +2,7 @@ import pytest
from atst.domain.exceptions import NotFoundError
from atst.domain.task_orders import TaskOrders
from atst.eda_client import MockEDAClient
from tests.factories import TaskOrderFactory
@@ -13,6 +14,17 @@ def test_can_get_task_order():
assert to.id == to.id
def test_nonexistent_task_order_raises():
def test_can_get_task_order_from_eda():
to = TaskOrders.get(MockEDAClient.MOCK_CONTRACT_NUMBER, client=MockEDAClient())
assert to.number == MockEDAClient.MOCK_CONTRACT_NUMBER
def test_nonexistent_task_order_raises_without_client():
with pytest.raises(NotFoundError):
TaskOrders.get("some fake number")
def test_nonexistent_task_order_raises_with_client():
with pytest.raises(NotFoundError):
TaskOrders.get("some other fake numer", client=MockEDAClient())