implement database sessions for test factories

This commit is contained in:
dandds
2018-08-07 15:16:28 -04:00
parent e59b1f09c4
commit 033578fa21
7 changed files with 39 additions and 49 deletions

View File

@@ -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

View File

@@ -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):

View File

@@ -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