From 784284200fdb24031e5ac0679bb9e756b8cbeb81 Mon Sep 17 00:00:00 2001
From: mreficent <miquel.raich@eficent.com>
Date: Wed, 28 Nov 2018 14:24:37 +0100
Subject: [PATCH] [ADD] tests

---
 mail_activity_team/tests/__init__.py          |   1 +
 .../tests/test_mail_activity_team.py          | 118 ++++++++++++++++++
 2 files changed, 119 insertions(+)
 create mode 100644 mail_activity_team/tests/__init__.py
 create mode 100644 mail_activity_team/tests/test_mail_activity_team.py

diff --git a/mail_activity_team/tests/__init__.py b/mail_activity_team/tests/__init__.py
new file mode 100644
index 00000000..f4a7f595
--- /dev/null
+++ b/mail_activity_team/tests/__init__.py
@@ -0,0 +1 @@
+from . import test_mail_activity_team
diff --git a/mail_activity_team/tests/test_mail_activity_team.py b/mail_activity_team/tests/test_mail_activity_team.py
new file mode 100644
index 00000000..3f7928cb
--- /dev/null
+++ b/mail_activity_team/tests/test_mail_activity_team.py
@@ -0,0 +1,118 @@
+# Copyright 2018 Eficent Business and IT Consulting Services, S.L.
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+from odoo.tests.common import TransactionCase
+from odoo.exceptions import ValidationError
+
+
+class TestMailActivityTeam(TransactionCase):
+
+    def setUp(self):
+        super(TestMailActivityTeam, self).setUp()
+
+        self.env["mail.activity.team"].search([]).unlink()
+
+        self.employee = self.env['res.users'].create({
+            'company_id': self.env.ref("base.main_company").id,
+            'name': "Employee",
+            'login': "csu",
+            'email': "crmuser@yourcompany.com",
+            'groups_id': [(6, 0, [
+                self.env.ref('base.group_user').id,
+                self.env.ref('base.group_partner_manager').id])]
+        })
+
+        self.employee2 = self.env['res.users'].create({
+            'company_id': self.env.ref("base.main_company").id,
+            'name': "Employee 2",
+            'login': "csu2",
+            'email': "crmuser2@yourcompany.com",
+            'groups_id': [(6, 0, [self.env.ref('base.group_user').id])]
+        })
+
+        self.partner_ir_model = self.env['ir.model']._get('res.partner')
+
+        activity_type_model = self.env['mail.activity.type']
+        self.activity1 = activity_type_model.create({
+            'name': 'Initial Contact',
+            'days': 5,
+            'summary': 'ACT 1 : Presentation, barbecue, ... ',
+            'res_model_id': self.partner_ir_model.id,
+        })
+        self.activity2 = activity_type_model.create({
+            'name': 'Call for Demo',
+            'days': 6,
+            'summary': 'ACT 2 : I want to show you my ERP !',
+            'res_model_id': self.partner_ir_model.id,
+        })
+
+        self.partner_client = self.env.ref("base.res_partner_1")
+
+        self.act1 = self.env['mail.activity'].sudo(self.employee).create({
+            'activity_type_id': self.activity1.id,
+            'note': 'Partner activity 1.',
+            'res_id': self.partner_client.id,
+            'res_model_id': self.partner_ir_model.id,
+            'user_id': self.employee.id,
+        })
+
+        self.team1 = self.env['mail.activity.team'].sudo().create({
+            'name': 'Team 1',
+            'res_model_ids': [(6, 0, [self.partner_ir_model.id])],
+            'member_ids': [(6, 0, [self.employee.id])],
+        })
+
+        self.team2 = self.env['mail.activity.team'].sudo().create({
+            'name': 'Team 2',
+            'res_model_ids': [(6, 0, [self.partner_ir_model.id])],
+            'member_ids': [(6, 0, [self.employee.id, self.employee2.id])],
+        })
+
+        self.act2 = self.env['mail.activity'].sudo(self.employee).create({
+            'activity_type_id': self.activity2.id,
+            'note': 'Partner activity 2.',
+            'res_id': self.partner_client.id,
+            'res_model_id': self.partner_ir_model.id,
+            'user_id': self.employee.id,
+        })
+
+    def test_missing_activities(self):
+        self.assertFalse(
+            self.act1.team_id, 'Error: Activity 1 should not have a team.')
+        self.assertEqual(self.team1.count_missing_activities, 1)
+        self.team1.assign_team_to_unassigned_activities()
+        self.team1._compute_missing_activities()
+        self.assertEqual(self.team1.count_missing_activities, 0)
+        self.assertEqual(self.act1.team_id, self.team1)
+
+    def test_activity_onchanges(self):
+        self.assertEqual(
+            self.act2.team_id, self.team1,
+            'Error: Activity 2 should have Team 1.')
+        with self.env.do_in_onchange():
+            self.act2.team_id = False
+            self.act2._onchange_team_id()
+            self.assertEqual(self.act2.user_id, self.employee)
+            self.act2.team_id = self.team2
+            self.act2._onchange_team_id()
+            self.assertEqual(self.act2.user_id, self.employee)
+            self.act2.user_id = self.employee2
+            self.act2._onchange_user_id()
+            self.assertEqual(self.act2.team_id, self.team2)
+            self.act2.team_id = self.team1
+            self.act2._onchange_team_id()
+            self.assertFalse(self.act2.user_id)
+            with self.assertRaises(ValidationError):
+                self.act2.write({
+                    'user_id': self.employee2.id,
+                    'team_id': self.team1.id,
+                })
+
+    def test_team_onchanges(self):
+        self.assertFalse(
+            self.team2.user_id,
+            'Error: Team 2 should not have a Team Leader yet.')
+        with self.env.do_in_onchange():
+            self.team2.user_id = self.employee
+            self.team2.member_ids = [(3, self.employee.id)]
+            self.team2._onchange_member_ids()
+            self.assertFalse(self.team2.user_id)