Add additional factories for tests.
parent
61c20354bf
commit
39bb7bcfcd
|
@ -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")
|
||||
|
||||
|
|
Loading…
Reference in New Issue