diff --git a/tests/factories.py b/tests/factories.py index 90625b90..05ef5c37 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -270,3 +270,18 @@ def create_task(**kwargs): defaults.update(kwargs) return TaskFactory.create(**defaults) + + +def create_membership(**kwargs): + "Create a membership along with its dependencies" + project = kwargs.pop("project", ProjectFactory()) + project.points.add(PointsFactory.create(project=project, value=None)) + + defaults = { + "project": project, + "user": project.owner, + "role": RoleFactory.create(project=project) + } + defaults.update(kwargs) + + return MembershipFactory.create(**defaults)