Remove PE number model

This commit is contained in:
Patrick Smith
2019-02-20 18:26:56 -05:00
parent 3ea9562284
commit 75f9b57979
7 changed files with 32 additions and 104 deletions

View File

@@ -1,28 +0,0 @@
import pytest
from atst.domain.exceptions import NotFoundError
from atst.domain.pe_numbers import PENumbers
from tests.factories import PENumberFactory
def test_can_get_pe_number():
new_pen = PENumberFactory.create(
number="0701367F", description="Combat Support - Offensive"
)
pen = PENumbers.get(new_pen.number)
assert pen.number == new_pen.number
def test_nonexistent_pe_number_raises():
with pytest.raises(NotFoundError):
PENumbers.get("some fake number")
def test_create_many():
pen_list = [["123456", "Land Speeder"], ["7891011", "Lightsaber"]]
PENumbers.create_many(pen_list)
assert PENumbers.get(pen_list[0][0])
assert PENumbers.get(pen_list[1][0])