From df8a7fb73c67612284e4c20ec23c4abdb1a68ffa Mon Sep 17 00:00:00 2001 From: Anler Hp Date: Wed, 18 Jun 2014 14:13:20 +0200 Subject: [PATCH] create_membership factory function --- tests/factories.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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)