1 files changed, 80 insertions, 0 deletions
diff --git a/helpdesk_mgmt/tests/test_helpdesk_ticket_team.py b/helpdesk_mgmt/tests/test_helpdesk_ticket_team.py
new file mode 100644
index 0000000..ea8c54b
--- /dev/null
+++ b/helpdesk_mgmt/tests/test_helpdesk_ticket_team.py
@@ -0,0 +1,80 @@
+from odoo.tests import common
+
+
+class TestHelpdeskTicketTeam(common.SavepointCase):
+    @classmethod
+    def setUpClass(cls):
+        super(TestHelpdeskTicketTeam, cls).setUpClass()
+        helpdesk_ticket = cls.env["helpdesk.ticket"]
+        helpdesk_ticket_team = cls.env["helpdesk.ticket.team"]
+        mail_alias = cls.env["mail.alias"]
+        cls.user_demo = cls.env.ref("base.user_demo")
+        cls.stage_closed = cls.env.ref("helpdesk_mgmt.helpdesk_ticket_stage_done")
+        cls.mail_alias_id = mail_alias.create(
+            {
+                "alias_name": "Test Mail Alias",
+                "alias_model_id": cls.env["ir.model"]
+                .search([("model", "=", "helpdesk.ticket")])
+                .id,
+            }
+        )
+        cls.team_id = helpdesk_ticket_team.create(
+            {"name": "Team 1", "alias_id": cls.mail_alias_id.id}
+        )
+        cls.helpdesk_ticket_1 = helpdesk_ticket.create(
+            {
+                "name": "Ticket 1",
+                "description": "Description",
+                "team_id": cls.team_id.id,
+                "priority": "3",
+            }
+        )
+        cls.helpdesk_ticket_2 = helpdesk_ticket.create(
+            {
+                "name": "Ticket 2",
+                "description": "Description",
+                "team_id": cls.team_id.id,
+                "user_id": cls.user_demo.id,
+                "priority": "1",
+            }
+        )
+
+    def test_helpdesk_ticket_todo(self):
+        self.assertEqual(
+            self.team_id.todo_ticket_count,
+            2,
+            "Helpdesk Ticket: Helpdesk ticket team should " "have two tickets to do.",
+        )
+        self.assertEqual(
+            self.team_id.todo_ticket_count_unassigned,
+            1,
+            "Helpdesk Ticket: Helpdesk ticket team should "
+            "have one tickets unassigned.",
+        )
+        self.assertEqual(
+            self.team_id.todo_ticket_count_high_priority,
+            1,
+            "Helpdesk Ticket: Helpdesk ticket team should "
+            "have two tickets with high priority.",
+        )
+        self.assertEqual(
+            self.team_id.todo_ticket_count_unattended,
+            2,
+            "Helpdesk Ticket: Helpdesk ticket team should "
+            "have two tickets unattended.",
+        )
+
+        self.helpdesk_ticket_1.write({"stage_id": self.stage_closed.id})
+
+        self.assertEqual(
+            self.team_id.todo_ticket_count_unattended,
+            1,
+            "Helpdesk Ticket: Helpdesk ticket team should "
+            "have one ticket unattended.",
+        )
+
+        self.assertEqual(
+            self.team_id.todo_ticket_count,
+            1,
+            "Helpdesk Ticket: Helpdesk ticket team should " "have one ticket to do.",
+        )