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

@@ -8,27 +8,36 @@ from atst.models.user import User
from atst.models.role import Role
class RequestFactory(factory.Factory):
class RequestFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = Request
id = factory.Sequence(lambda x: uuid4())
class PENumberFactory(factory.Factory):
class PENumberFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = PENumber
class TaskOrderFactory(factory.Factory):
class TaskOrderFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = TaskOrder
class RoleFactory(factory.Factory):
class RoleFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = Role
permissions = []
class UserFactory(factory.Factory):
class UserFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = User