519 lines
17 KiB
Python
519 lines
17 KiB
Python
# Copyright (C) 2014 Andrey Antukh <niwi@niwi.be>
|
|
# Copyright (C) 2014 Jesús Espino <jespinog@gmail.com>
|
|
# Copyright (C) 2014 David Barragán <bameda@dbarragan.com>
|
|
# Copyright (C) 2014 Anler Hernández <hello@anler.me>
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import uuid
|
|
import threading
|
|
from datetime import date, timedelta
|
|
|
|
from django.conf import settings
|
|
|
|
import factory
|
|
|
|
|
|
class Factory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
strategy = factory.CREATE_STRATEGY
|
|
model = None
|
|
abstract = True
|
|
|
|
_SEQUENCE = 1
|
|
_SEQUENCE_LOCK = threading.Lock()
|
|
|
|
@classmethod
|
|
def _setup_next_sequence(cls):
|
|
with cls._SEQUENCE_LOCK:
|
|
cls._SEQUENCE += 1
|
|
return cls._SEQUENCE
|
|
|
|
|
|
class ProjectTemplateFactory(Factory):
|
|
class Meta:
|
|
strategy = factory.CREATE_STRATEGY
|
|
model = "projects.ProjectTemplate"
|
|
django_get_or_create = ("slug",)
|
|
|
|
name = "Template name"
|
|
slug = settings.DEFAULT_PROJECT_TEMPLATE
|
|
description = factory.Sequence(lambda n: "Description {}".format(n))
|
|
|
|
us_statuses = []
|
|
points = []
|
|
task_statuses = []
|
|
issue_statuses = []
|
|
issue_types = []
|
|
priorities = []
|
|
severities = []
|
|
roles = []
|
|
default_owner_role = "tester"
|
|
|
|
|
|
class ProjectFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Project"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Project {}".format(n))
|
|
slug = factory.Sequence(lambda n: "project-{}-slug".format(n))
|
|
description = "Project description"
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
creation_template = factory.SubFactory("tests.factories.ProjectTemplateFactory")
|
|
|
|
|
|
class RoleFactory(Factory):
|
|
class Meta:
|
|
model = "users.Role"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Role {}".format(n))
|
|
slug = factory.Sequence(lambda n: "test-role-{}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class PointsFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Points"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Points {}".format(n))
|
|
value = 2
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class RolePointsFactory(Factory):
|
|
class Meta:
|
|
model = "userstories.RolePoints"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
user_story = factory.SubFactory("tests.factories.UserStoryFactory")
|
|
role = factory.SubFactory("tests.factories.RoleFactory")
|
|
points = factory.SubFactory("tests.factories.PointsFactory")
|
|
|
|
|
|
class UserStoryAttachmentFactory(Factory):
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
content_object = factory.SubFactory("tests.factories.UserStoryFactory")
|
|
attached_file = factory.django.FileField(data=b"File contents")
|
|
|
|
class Meta:
|
|
model = "attachments.Attachment"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
|
|
class TaskAttachmentFactory(Factory):
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
content_object = factory.SubFactory("tests.factories.TaskFactory")
|
|
attached_file = factory.django.FileField(data=b"File contents")
|
|
|
|
class Meta:
|
|
model = "attachments.Attachment"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
|
|
class IssueAttachmentFactory(Factory):
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
content_object = factory.SubFactory("tests.factories.IssueFactory")
|
|
attached_file = factory.django.FileField(data=b"File contents")
|
|
|
|
class Meta:
|
|
model = "attachments.Attachment"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
|
|
class WikiAttachmentFactory(Factory):
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
content_object = factory.SubFactory("tests.factories.WikiFactory")
|
|
attached_file = factory.django.FileField(data=b"File contents")
|
|
|
|
class Meta:
|
|
model = "attachments.Attachment"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
|
|
class RolePointsFactory(Factory):
|
|
class Meta:
|
|
model = "userstories.RolePoints"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
user_story = factory.SubFactory("tests.factories.UserStoryFactory")
|
|
role = factory.SubFactory("tests.factories.RoleFactory")
|
|
points = factory.SubFactory("tests.factories.PointsFactory")
|
|
|
|
|
|
class UserFactory(Factory):
|
|
class Meta:
|
|
model = "users.User"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
username = factory.Sequence(lambda n: "user{}".format(n))
|
|
email = factory.LazyAttribute(lambda obj: '%s@email.com' % obj.username)
|
|
password = factory.PostGeneration(lambda obj, *args, **kwargs: obj.set_password(obj.username))
|
|
|
|
|
|
class MembershipFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Membership"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
token = factory.LazyAttribute(lambda obj: str(uuid.uuid1()))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
role = factory.SubFactory("tests.factories.RoleFactory")
|
|
user = factory.SubFactory("tests.factories.UserFactory")
|
|
|
|
|
|
class InvitationFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Membership"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
token = factory.LazyAttribute(lambda obj: str(uuid.uuid1()))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
role = factory.SubFactory("tests.factories.RoleFactory")
|
|
email = factory.Sequence(lambda n: "user{}@email.com".format(n))
|
|
|
|
|
|
class StorageEntryFactory(Factory):
|
|
class Meta:
|
|
model = "userstorage.StorageEntry"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
key = factory.Sequence(lambda n: "key-{}".format(n))
|
|
value = factory.Sequence(lambda n: "value {}".format(n))
|
|
|
|
|
|
class UserStoryFactory(Factory):
|
|
class Meta:
|
|
model = "userstories.UserStory"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
ref = factory.Sequence(lambda n: n)
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
subject = factory.Sequence(lambda n: "User Story {}".format(n))
|
|
description = factory.Sequence(lambda n: "User Story {} description".format(n))
|
|
status = factory.SubFactory("tests.factories.UserStoryStatusFactory")
|
|
|
|
|
|
class UserStoryStatusFactory(Factory):
|
|
class Meta:
|
|
model = "projects.UserStoryStatus"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "User Story status {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
|
|
class TaskStatusFactory(Factory):
|
|
class Meta:
|
|
model = "projects.TaskStatus"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Task status {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class MilestoneFactory(Factory):
|
|
class Meta:
|
|
model = "milestones.Milestone"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Milestone {}".format(n))
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
estimated_start = factory.LazyAttribute(lambda o: date.today())
|
|
estimated_finish = factory.LazyAttribute(lambda o: o.estimated_start + timedelta(days=7))
|
|
|
|
|
|
class IssueFactory(Factory):
|
|
class Meta:
|
|
model = "issues.Issue"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
subject = factory.Sequence(lambda n: "Issue {}".format(n))
|
|
description = factory.Sequence(lambda n: "Issue {} description".format(n))
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
status = factory.SubFactory("tests.factories.IssueStatusFactory")
|
|
severity = factory.SubFactory("tests.factories.SeverityFactory")
|
|
priority = factory.SubFactory("tests.factories.PriorityFactory")
|
|
type = factory.SubFactory("tests.factories.IssueTypeFactory")
|
|
milestone = factory.SubFactory("tests.factories.MilestoneFactory")
|
|
|
|
|
|
class TaskFactory(Factory):
|
|
class Meta:
|
|
model = "tasks.Task"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
ref = factory.Sequence(lambda n: n)
|
|
subject = factory.Sequence(lambda n: "Task {}".format(n))
|
|
description = factory.Sequence(lambda n: "Task {} description".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")
|
|
user_story = factory.SubFactory("tests.factories.UserStoryFactory")
|
|
|
|
|
|
class WikiPageFactory(Factory):
|
|
class Meta:
|
|
model = "wiki.WikiPage"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
slug = factory.Sequence(lambda n: "wiki-page-{}".format(n))
|
|
content = factory.Sequence(lambda n: "Wiki Page {} content".format(n))
|
|
|
|
|
|
class WikiLinkFactory(Factory):
|
|
class Meta:
|
|
model = "wiki.WikiLink"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
title = factory.Sequence(lambda n: "Wiki Link {} title".format(n))
|
|
href = factory.Sequence(lambda n: "link-{}".format(n))
|
|
order = factory.Sequence(lambda n: n)
|
|
|
|
|
|
class IssueStatusFactory(Factory):
|
|
class Meta:
|
|
model = "projects.IssueStatus"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Issue Status {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class UserStoryStatusFactory(Factory):
|
|
class Meta:
|
|
model = "projects.UserStoryStatus"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "User Story Status {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class SeverityFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Severity"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Severity {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class PriorityFactory(Factory):
|
|
class Meta:
|
|
model = "projects.Priority"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Priority {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
class IssueTypeFactory(Factory):
|
|
class Meta:
|
|
model = "projects.IssueType"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
name = factory.Sequence(lambda n: "Issue Type {}".format(n))
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
|
|
|
|
# class FanFactory(Factory):
|
|
# project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
# user = factory.SubFactory("tests.factories.UserFactory")
|
|
|
|
|
|
# class StarsFactory(Factory):
|
|
# project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
# count = 0
|
|
|
|
|
|
class VoteFactory(Factory):
|
|
class Meta:
|
|
model = "votes.Vote"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
content_type = factory.SubFactory("tests.factories.ContentTypeFactory")
|
|
object_id = factory.Sequence(lambda n: n)
|
|
user = factory.SubFactory("tests.factories.UserFactory")
|
|
|
|
|
|
class VotesFactory(Factory):
|
|
class Meta:
|
|
model = "votes.Votes"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
content_type = factory.SubFactory("tests.factories.ContentTypeFactory")
|
|
object_id = factory.Sequence(lambda n: n)
|
|
|
|
|
|
class ContentTypeFactory(Factory):
|
|
class Meta:
|
|
model = "contenttypes.ContentType"
|
|
strategy = factory.CREATE_STRATEGY
|
|
django_get_or_create = ("app_label", "model")
|
|
|
|
app_label = factory.LazyAttribute(lambda obj: "issues")
|
|
model = factory.LazyAttribute(lambda obj: "Issue")
|
|
|
|
|
|
class AttachmentFactory(Factory):
|
|
class Meta:
|
|
model = "attachments.Attachment"
|
|
strategy = factory.CREATE_STRATEGY
|
|
|
|
owner = factory.SubFactory("tests.factories.UserFactory")
|
|
project = factory.SubFactory("tests.factories.ProjectFactory")
|
|
content_type = factory.SubFactory("tests.factories.ContentTypeFactory")
|
|
object_id = factory.Sequence(lambda n: n)
|
|
attached_file = factory.django.FileField(data=b"File contents")
|
|
|
|
|
|
def create_issue(**kwargs):
|
|
"Create an issue and along with its dependencies."
|
|
owner = kwargs.pop("owner", None)
|
|
if owner is None:
|
|
owner = UserFactory.create()
|
|
|
|
project = kwargs.pop("project", None)
|
|
if project is None:
|
|
project = ProjectFactory.create(owner=owner)
|
|
|
|
defaults = {
|
|
"project": project,
|
|
"owner": owner,
|
|
"status": IssueStatusFactory.create(project=project),
|
|
"milestone": MilestoneFactory.create(project=project),
|
|
"priority": PriorityFactory.create(project=project),
|
|
"severity": SeverityFactory.create(project=project),
|
|
"type": IssueTypeFactory.create(project=project),
|
|
}
|
|
defaults.update(kwargs)
|
|
|
|
return IssueFactory.create(**defaults)
|
|
|
|
|
|
def create_task(**kwargs):
|
|
"Create a task and along with its dependencies."
|
|
owner = kwargs.pop("owner", None)
|
|
if not owner:
|
|
owner = UserFactory.create()
|
|
|
|
project = kwargs.pop("project", None)
|
|
if project is None:
|
|
project = ProjectFactory.create(owner=owner)
|
|
|
|
defaults = {
|
|
"project": project,
|
|
"owner": owner,
|
|
"status": TaskStatusFactory.create(project=project),
|
|
"milestone": MilestoneFactory.create(project=project),
|
|
"user_story": UserStoryFactory.create(project=project, owner=owner),
|
|
}
|
|
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)
|
|
|
|
|
|
def create_invitation(**kwargs):
|
|
"Create an invitation along with its dependencies"
|
|
project = kwargs.pop("project", ProjectFactory())
|
|
project.points.add(PointsFactory.create(project=project, value=None))
|
|
|
|
defaults = {
|
|
"project": project,
|
|
"role": RoleFactory.create(project=project),
|
|
"email": "invited-user@email.com",
|
|
"token": "tokenvalue",
|
|
"invited_by_id": project.owner.id
|
|
}
|
|
defaults.update(kwargs)
|
|
|
|
return MembershipFactory.create(**defaults)
|
|
|
|
|
|
def create_userstory(**kwargs):
|
|
"Create an user story along with its dependencies"
|
|
|
|
owner = kwargs.pop("owner", None)
|
|
if not owner:
|
|
owner = UserFactory.create()
|
|
|
|
project = kwargs.pop("project", None)
|
|
if project is None:
|
|
project = ProjectFactory.create(owner=owner)
|
|
|
|
defaults = {
|
|
"project": project,
|
|
"owner": owner,
|
|
"milestone": MilestoneFactory.create(project=project, owner=owner)
|
|
}
|
|
defaults.update(kwargs)
|
|
|
|
return UserStoryFactory(**defaults)
|
|
|
|
|
|
def create_project(**kwargs):
|
|
"Create a project along with its dependencies"
|
|
defaults = {}
|
|
defaults.update(kwargs)
|
|
|
|
ProjectTemplateFactory.create(slug=settings.DEFAULT_PROJECT_TEMPLATE)
|
|
|
|
project = ProjectFactory.create(**defaults)
|
|
project.default_issue_status = IssueStatusFactory.create(project=project)
|
|
project.default_severity = SeverityFactory.create(project=project)
|
|
project.default_priority = PriorityFactory.create(project=project)
|
|
project.default_issue_type = IssueTypeFactory.create(project=project)
|
|
project.default_us_status = UserStoryStatusFactory.create(project=project)
|
|
project.default_task_status = TaskStatusFactory.create(project=project)
|
|
|
|
project.save()
|
|
|
|
return project
|
|
|
|
|
|
def create_user(**kwargs):
|
|
"Create an user along with her dependencies"
|
|
ProjectTemplateFactory.create(slug=settings.DEFAULT_PROJECT_TEMPLATE)
|
|
RoleFactory.create()
|
|
return UserFactory.create(**kwargs)
|