Add additional factories for tests.

remotes/origin/enhancement/email-actions
Andrey Antukh 2014-06-05 12:49:51 +02:00
parent 61c20354bf
commit 39bb7bcfcd
1 changed files with 24 additions and 0 deletions

View File

@ -133,6 +133,23 @@ class IssueFactory(Factory):
milestone = factory.SubFactory("tests.factories.MilestoneFactory")
class TaskFactory(Factory):
FACTORY_FOR = get_model("tasks", "Task")
subject = factory.Sequence(lambda n: "Task {}".format(n))
owner = factory.SubFactory("tests.factories.UserFactory")
project = factory.SubFactory("tests.factories.ProjectFactory")
status = factory.SubFactory("tests.factories.TaskStatusFactory")
milestone = factory.SubFactory("tests.factories.MilestoneFactory")
class WikiPageFactory(Factory):
FACTORY_FOR = get_model("wiki", "WikiPage")
project = factory.SubFactory("tests.factories.ProjectFactory")
owner = factory.SubFactory("tests.factories.UserFactory")
class IssueStatusFactory(Factory):
FACTORY_FOR = get_model("projects", "IssueStatus")
@ -140,6 +157,13 @@ class IssueStatusFactory(Factory):
project = factory.SubFactory("tests.factories.ProjectFactory")
class TaskStatusFactory(Factory):
FACTORY_FOR = get_model("projects", "TaskStatus")
name = factory.Sequence(lambda n: "Issue Status {}".format(n))
project = factory.SubFactory("tests.factories.ProjectFactory")
class SeverityFactory(Factory):
FACTORY_FOR = get_model("projects", "Severity")