Task change assignee

conftest.py

@@ -0,0 +1,82 @@
+import pytest
+from users.models import User, Team, Role
+from tasks.models import Task, Priority, Status
+
+
+@pytest.fixture
+def teams():
+    """
+    Adds teams to DB
+    """
+    team1 = Team.objects.create(name="Team1",
+                                description="This is a test team")
+    team2 = Team.objects.create(name="Team2",
+                                description="This is a test team")
+    team3 = Team.objects.create(name="Team3",
+                                description="This is a test team")
+    return team1, team2, team3
+
+
+@pytest.fixture
+def users(teams):
+    """
+    Adds users to DB
+    """
+    team1, team2, team3 = teams
+    employees = []
+    managers = []
+    users_counter = 0
+    for i in range(3):
+        for team in (team1, team2):
+            employee = User.create_user(username=f"User{users_counter}",
+                                        email=f"User{users_counter}@redhat.com",
+                                        password="ssSSAD231!@",
+                                        first_name=f"User{users_counter}",
+                                        last_name=f"User{users_counter}",
+                                        role=Role.EMPLOYEE,
+                                        team=team)
+            employees += [employee]
+            users_counter += 1
+    for i, team in enumerate((team1, team2, team3)):
+        manager = User.create_user(username=f"Manager{i}",
+                                   email=f"Manager{i}@redhat.com",
+                                   password="ssSSAD231!@",
+                                   first_name=f"Manager{i}",
+                                   last_name=f"Manager{i}",
+                                   role=Role.MANAGER,
+                                   team=team)
+        managers += [manager]
+    return teams, tuple(employees), tuple(managers)
+
+
+@pytest.fixture
+def tasks(users):
+    """
+    Adds tasks to DB
+    """
+    teams, employees, managers = users
+    tasks = []
+    for k in range(3):
+        for i in range(3):
+            for j in range(2):
+                task = Task.objects.create(title=f"Task{j}",
+                                           assignee=employees[k + i],
+                                           created_by=managers[k],
+                                           priority=Priority.LOW,
+                                           status=Status.BACKLOG,
+                                           description="Test task")
+                tasks += [task]
+        return teams, employees, managers, tuple(tasks)
+
+
+@pytest.fixture
+def test_db(tasks):
+    """
+    Returns test DB components as tuples.
+    The test DB contains: 3 teams, 3 users per team, 2 tasks per user, 1 manager per team.
+    """
+    teams = tasks[0]
+    employees = tasks[1]
+    managers = tasks[2]
+    task = tasks[3]
+    return teams, employees, managers, task

tasks/models.py

@@ -53,6 +53,9 @@ def filter_by_symbol(cls, priority_filter):
         print(priority_filter)
         return cls.objects.filter(priority=priority_filter)
 
+    def change_assignee(self, new_assignee):
+        self.assignee = new_assignee
+
 
 class Comment(models.Model):
     appUser = models.ForeignKey(

tasks/tests/test_change_assignee.py

@@ -0,0 +1,12 @@
+import pytest
+from tasks.models import Task
+
+
+@pytest.mark.django_db
+class TestChangeTaskAssignee:
+    def test_change_task_assignee(self, test_db):
+        _, employees, _, _ = test_db
+        task = Task.objects.filter(assignee=employees[0]).first()
+        assert task is not None
+        task.change_assignee(employees[1])
+        assert task.assignee == employees[1]