29 lines
823 B
Python
29 lines
823 B
Python
import pytest
|
|
|
|
from atst.domain.exceptions import NotFoundError
|
|
from atst.domain.legacy_task_orders import LegacyTaskOrders
|
|
from atst.eda_client import MockEDAClient
|
|
|
|
from tests.factories import LegacyTaskOrderFactory
|
|
|
|
|
|
def test_can_get_task_order():
|
|
new_to = LegacyTaskOrderFactory.create(number="0101969F")
|
|
to = LegacyTaskOrders.get(new_to.number)
|
|
|
|
assert to.id == to.id
|
|
|
|
|
|
def test_nonexistent_task_order_raises_without_client():
|
|
with pytest.raises(NotFoundError):
|
|
LegacyTaskOrders.get("some fake number")
|
|
|
|
|
|
def test_nonexistent_task_order_raises_with_client(monkeypatch):
|
|
monkeypatch.setattr(
|
|
"atst.domain.legacy_task_orders.LegacyTaskOrders._client",
|
|
lambda: MockEDAClient(),
|
|
)
|
|
with pytest.raises(NotFoundError):
|
|
LegacyTaskOrders.get("some other fake numer")
|