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