implement database sessions for test factories
This commit is contained in:
@@ -6,20 +6,8 @@ from atst.domain.pe_numbers import PENumbers
|
||||
from tests.factories import PENumberFactory
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def new_pe_number(session):
|
||||
def make_pe_number(**kwargs):
|
||||
pen = PENumberFactory.create(**kwargs)
|
||||
session.add(pen)
|
||||
session.commit()
|
||||
|
||||
return pen
|
||||
|
||||
return make_pe_number
|
||||
|
||||
|
||||
def test_can_get_pe_number(new_pe_number):
|
||||
new_pen = new_pe_number(number="0701367F", description="Combat Support - Offensive")
|
||||
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
|
||||
|
@@ -9,11 +9,7 @@ from tests.factories import RequestFactory
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def new_request(session):
|
||||
created_request = RequestFactory.create()
|
||||
session.add(created_request)
|
||||
session.commit()
|
||||
|
||||
return created_request
|
||||
return RequestFactory.create()
|
||||
|
||||
|
||||
def test_can_get_request(new_request):
|
||||
|
@@ -6,20 +6,8 @@ from atst.domain.task_orders import TaskOrders
|
||||
from tests.factories import TaskOrderFactory
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def new_task_order(session):
|
||||
def make_task_order(**kwargs):
|
||||
to = TaskOrderFactory.create(**kwargs)
|
||||
session.add(to)
|
||||
session.commit()
|
||||
|
||||
return to
|
||||
|
||||
return make_task_order
|
||||
|
||||
|
||||
def test_can_get_task_order(new_task_order):
|
||||
new_to = new_task_order(number="0101969F")
|
||||
def test_can_get_task_order():
|
||||
new_to = TaskOrderFactory.create(number="0101969F")
|
||||
to = TaskOrders.get(new_to.number)
|
||||
|
||||
assert to.id == to.id
|
||||
|
Reference in New Issue
Block a user