Remove legacy task order model
This commit is contained in:
@@ -1,28 +0,0 @@
|
||||
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")
|
@@ -10,7 +10,6 @@ from atst.forms import data
|
||||
from atst.models.attachment import Attachment
|
||||
from atst.models.environment import Environment
|
||||
from atst.models.application import Application
|
||||
from atst.models.legacy_task_order import LegacyTaskOrder, Source, FundingType
|
||||
from atst.models.task_order import TaskOrder
|
||||
from atst.models.user import User
|
||||
from atst.models.role import Role
|
||||
@@ -100,25 +99,6 @@ class UserFactory(Base):
|
||||
return cls.create(atat_role=role, **kwargs)
|
||||
|
||||
|
||||
class LegacyTaskOrderFactory(Base):
|
||||
class Meta:
|
||||
model = LegacyTaskOrder
|
||||
|
||||
source = Source.MANUAL
|
||||
funding_type = FundingType.PROC
|
||||
funding_type_other = None
|
||||
number = factory.LazyFunction(
|
||||
lambda: "".join(random.choices(string.ascii_uppercase + string.digits, k=13))
|
||||
)
|
||||
expiration_date = factory.LazyFunction(random_future_date)
|
||||
clin_0001 = random.randrange(100, 100_000)
|
||||
clin_0003 = random.randrange(100, 100_000)
|
||||
clin_1001 = random.randrange(100, 100_000)
|
||||
clin_1003 = random.randrange(100, 100_000)
|
||||
clin_2001 = random.randrange(100, 100_000)
|
||||
clin_2003 = random.randrange(100, 100_000)
|
||||
|
||||
|
||||
class PortfolioFactory(Base):
|
||||
class Meta:
|
||||
model = Portfolio
|
||||
|
@@ -1,20 +0,0 @@
|
||||
from tests.factories import LegacyTaskOrderFactory
|
||||
from tests.assert_util import dict_contains
|
||||
|
||||
|
||||
def test_as_dictionary():
|
||||
data = LegacyTaskOrderFactory.dictionary()
|
||||
real_task_order = LegacyTaskOrderFactory.create(**data)
|
||||
assert dict_contains(real_task_order.to_dictionary(), data)
|
||||
|
||||
|
||||
def test_budget():
|
||||
legacy_task_order = LegacyTaskOrderFactory.create(
|
||||
clin_0001=500,
|
||||
clin_0003=200,
|
||||
clin_1001=None,
|
||||
clin_1003=None,
|
||||
clin_2001=None,
|
||||
clin_2003=None,
|
||||
)
|
||||
assert legacy_task_order.budget == 700
|
Reference in New Issue
Block a user