taiga-back/tests/integration/test_neighbors.py

341 lines
14 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/>.
from functools import partial
import pytest
from taiga.projects.userstories.models import UserStory
from taiga.projects.issues.models import Issue
from taiga.base import tags
from taiga.base import neighbors as n
from .. import factories as f
from ..utils import disconnect_signals, reconnect_signals
def setup_module():
disconnect_signals()
def teardown_module():
reconnect_signals()
class TestGetAttribute:
def test_no_attribute(self, object):
object.full_name = "name"
with pytest.raises(AttributeError):
n.get_attribute(object, "name")
with pytest.raises(AttributeError):
n.get_attribute(object, "full_name__last_name")
def test_one_level(self, object):
object.name = "name"
assert n.get_attribute(object, "name") == object.name
def test_two_levels(self, object):
object.name = object
object.name.full_name = "first name"
assert n.get_attribute(object, "name__full_name") == object.name.full_name
def test_three_levels(self, object):
object.info = object
object.info.name = object
object.info.name.full_name = "first name"
assert n.get_attribute(object, "info__name__full_name") == object.info.name.full_name
def test_transform_field_into_lookup():
transform = partial(n.transform_field_into_lookup, value="chuck", operator="__lt",
operator_if_desc="__gt")
assert transform(name="name") == {"name__lt": "chuck"}
assert transform(name="-name") == {"name__gt": "chuck"}
def test_disjunction_filters():
filters = [{"age__lt": 21, "name__eq": "chuck"}]
result_str = str(n.disjunction_filters(filters))
assert result_str.startswith("(OR: ")
assert "('age__lt', 21)" in result_str
assert "('name__eq', 'chuck')" in result_str
@pytest.mark.django_db
class TestUserStories:
def test_no_filters(self):
project = f.ProjectFactory.create()
us1 = f.UserStoryFactory.create(project=project)
us2 = f.UserStoryFactory.create(project=project)
us3 = f.UserStoryFactory.create(project=project)
neighbors = n.get_neighbors(us2)
assert neighbors.left == us1
assert neighbors.right == us3
def test_filtered_by_tags(self):
tag_names = ["test"]
project = f.ProjectFactory.create()
f.UserStoryFactory.create(project=project)
us1 = f.UserStoryFactory.create(project=project, tags=tag_names)
us2 = f.UserStoryFactory.create(project=project, tags=tag_names)
test_user_stories = tags.filter(UserStory.objects.get_queryset(), contains=tag_names)
neighbors = n.get_neighbors(us1, results_set=test_user_stories)
assert neighbors.left is None
assert neighbors.right == us2
def test_filtered_by_milestone(self):
project = f.ProjectFactory.create()
milestone = f.MilestoneFactory.create(project=project)
f.UserStoryFactory.create(project=project)
us1 = f.UserStoryFactory.create(project=project, milestone=milestone)
us2 = f.UserStoryFactory.create(project=project, milestone=milestone)
milestone_user_stories = UserStory.objects.filter(milestone=milestone)
neighbors = n.get_neighbors(us1, results_set=milestone_user_stories)
assert neighbors.left is None
assert neighbors.right == us2
@pytest.mark.django_db
class TestIssues:
def test_no_filters(self):
project = f.ProjectFactory.create()
issue1 = f.IssueFactory.create(project=project)
issue2 = f.IssueFactory.create(project=project)
issue3 = f.IssueFactory.create(project=project)
neighbors = n.get_neighbors(issue2)
assert neighbors.left == issue3
assert neighbors.right == issue1
def test_empty_related_queryset(self):
project = f.ProjectFactory.create()
issue1 = f.IssueFactory.create(project=project)
issue2 = f.IssueFactory.create(project=project)
issue3 = f.IssueFactory.create(project=project)
neighbors = n.get_neighbors(issue2, Issue.objects.none())
assert neighbors.left == issue3
assert neighbors.right == issue1
def test_ordering_by_severity(self):
project = f.ProjectFactory.create()
severity1 = f.SeverityFactory.create(project=project, order=1)
severity2 = f.SeverityFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, severity=severity2)
issue2 = f.IssueFactory.create(project=project, severity=severity1)
issue3 = f.IssueFactory.create(project=project, severity=severity1)
issues = Issue.objects.filter(project=project).order_by("severity")
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
issue3_neighbors = n.get_neighbors(issue3, results_set=issues)
assert issue3_neighbors.left is None
assert issue3_neighbors.right == issue2
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right == issue1
def test_ordering_by_severity_desc(self):
project = f.ProjectFactory.create()
severity1 = f.SeverityFactory.create(project=project, order=1)
severity2 = f.SeverityFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, severity=severity2)
issue2 = f.IssueFactory.create(project=project, severity=severity1)
issue3 = f.IssueFactory.create(project=project, severity=severity1)
issues = Issue.objects.filter(project=project).order_by("-severity")
issue1_neighbors = n.get_neighbors(issue1, results_set=issues)
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
assert issue1_neighbors.left is None
assert issue1_neighbors.right == issue3
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right is None
def test_ordering_by_status(self):
project = f.ProjectFactory.create()
status1 = f.IssueStatusFactory.create(project=project, order=1)
status2 = f.IssueStatusFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, status=status2)
issue2 = f.IssueFactory.create(project=project, status=status1)
issue3 = f.IssueFactory.create(project=project, status=status1)
issues = Issue.objects.filter(project=project).order_by("status")
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
issue3_neighbors = n.get_neighbors(issue3, results_set=issues)
assert issue3_neighbors.left is None
assert issue3_neighbors.right == issue2
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right == issue1
def test_ordering_by_status_desc(self):
project = f.ProjectFactory.create()
status1 = f.IssueStatusFactory.create(project=project, order=1)
status2 = f.IssueStatusFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, status=status2)
issue2 = f.IssueFactory.create(project=project, status=status1)
issue3 = f.IssueFactory.create(project=project, status=status1)
issues = Issue.objects.filter(project=project).order_by("-status")
issue1_neighbors = n.get_neighbors(issue1, results_set=issues)
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
assert issue1_neighbors.left is None
assert issue1_neighbors.right == issue3
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right is None
def test_ordering_by_priority(self):
project = f.ProjectFactory.create()
priority1 = f.PriorityFactory.create(project=project, order=1)
priority2 = f.PriorityFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, priority=priority2)
issue2 = f.IssueFactory.create(project=project, priority=priority1)
issue3 = f.IssueFactory.create(project=project, priority=priority1)
issues = Issue.objects.filter(project=project).order_by("priority")
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
issue3_neighbors = n.get_neighbors(issue3, results_set=issues)
assert issue3_neighbors.left is None
assert issue3_neighbors.right == issue2
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right == issue1
def test_ordering_by_priority_desc(self):
project = f.ProjectFactory.create()
priority1 = f.PriorityFactory.create(project=project, order=1)
priority2 = f.PriorityFactory.create(project=project, order=2)
issue1 = f.IssueFactory.create(project=project, priority=priority2)
issue2 = f.IssueFactory.create(project=project, priority=priority1)
issue3 = f.IssueFactory.create(project=project, priority=priority1)
issues = Issue.objects.filter(project=project).order_by("-priority")
issue1_neighbors = n.get_neighbors(issue1, results_set=issues)
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
assert issue1_neighbors.left is None
assert issue1_neighbors.right == issue3
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right is None
def test_ordering_by_owner(self):
project = f.ProjectFactory.create()
owner1 = f.UserFactory.create(full_name="Chuck Norris")
owner2 = f.UserFactory.create(full_name="George Of The Jungle")
issue1 = f.IssueFactory.create(project=project, owner=owner2)
issue2 = f.IssueFactory.create(project=project, owner=owner1)
issue3 = f.IssueFactory.create(project=project, owner=owner1)
issues = Issue.objects.filter(project=project).order_by("owner__full_name")
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
issue3_neighbors = n.get_neighbors(issue3, results_set=issues)
assert issue3_neighbors.left is None
assert issue3_neighbors.right == issue2
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right == issue1
def test_ordering_by_owner_desc(self):
project = f.ProjectFactory.create()
owner1 = f.UserFactory.create(full_name="Chuck Norris")
owner2 = f.UserFactory.create(full_name="George Of The Jungle")
issue1 = f.IssueFactory.create(project=project, owner=owner2)
issue2 = f.IssueFactory.create(project=project, owner=owner1)
issue3 = f.IssueFactory.create(project=project, owner=owner1)
issues = Issue.objects.filter(project=project).order_by("-owner__full_name")
issue1_neighbors = n.get_neighbors(issue1, results_set=issues)
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
assert issue1_neighbors.left is None
assert issue1_neighbors.right == issue3
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right is None
def test_ordering_by_assigned_to(self):
project = f.ProjectFactory.create()
assigned_to1 = f.UserFactory.create(full_name="Chuck Norris")
assigned_to2 = f.UserFactory.create(full_name="George Of The Jungle")
issue1 = f.IssueFactory.create(project=project, assigned_to=assigned_to2)
issue2 = f.IssueFactory.create(project=project, assigned_to=assigned_to1)
issue3 = f.IssueFactory.create(project=project, assigned_to=assigned_to1)
issues = Issue.objects.filter(project=project).order_by("assigned_to__full_name")
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
issue3_neighbors = n.get_neighbors(issue3, results_set=issues)
assert issue3_neighbors.left is None
assert issue3_neighbors.right == issue2
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right == issue1
def test_ordering_by_assigned_to_desc(self):
project = f.ProjectFactory.create()
assigned_to1 = f.UserFactory.create(full_name="Chuck Norris")
assigned_to2 = f.UserFactory.create(full_name="George Of The Jungle")
issue1 = f.IssueFactory.create(project=project, assigned_to=assigned_to2)
issue2 = f.IssueFactory.create(project=project, assigned_to=assigned_to1)
issue3 = f.IssueFactory.create(project=project, assigned_to=assigned_to1)
issues = Issue.objects.filter(project=project).order_by("-assigned_to__full_name")
issue1_neighbors = n.get_neighbors(issue1, results_set=issues)
issue2_neighbors = n.get_neighbors(issue2, results_set=issues)
assert issue1_neighbors.left is None
assert issue1_neighbors.right == issue3
assert issue2_neighbors.left == issue3
assert issue2_neighbors.right is None