change task orders domain class name

This commit is contained in:
dandds
2018-12-12 16:48:18 -05:00
parent e509c34cdd
commit 15d0db01b6
4 changed files with 20 additions and 17 deletions

View File

@@ -1,7 +1,7 @@
import pytest
from atst.domain.exceptions import NotFoundError
from atst.domain.task_orders import TaskOrders
from atst.domain.legacy_task_orders import LegacyTaskOrders
from atst.eda_client import MockEDAClient
from tests.factories import LegacyTaskOrderFactory
@@ -9,19 +9,20 @@ from tests.factories import LegacyTaskOrderFactory
def test_can_get_task_order():
new_to = LegacyTaskOrderFactory.create(number="0101969F")
to = TaskOrders.get(new_to.number)
to = LegacyTaskOrders.get(new_to.number)
assert to.id == to.id
def test_nonexistent_task_order_raises_without_client():
with pytest.raises(NotFoundError):
TaskOrders.get("some fake number")
LegacyTaskOrders.get("some fake number")
def test_nonexistent_task_order_raises_with_client(monkeypatch):
monkeypatch.setattr(
"atst.domain.task_orders.TaskOrders._client", lambda: MockEDAClient()
"atst.domain.legacy_task_orders.LegacyTaskOrders._client",
lambda: MockEDAClient(),
)
with pytest.raises(NotFoundError):
TaskOrders.get("some other fake numer")
LegacyTaskOrders.get("some other fake numer")