implement database sessions for test factories
This commit is contained in:
@@ -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
|
||||
|
||||
|
Reference in New Issue
Block a user