initial commit 2

1 files changed, 91 insertions, 0 deletions

diff --git a/addons/hr_contract/tests/test_contract.py b/addons/hr_contract/tests/test_contract.py
new file mode 100644
index 00000000..7f4b9ed8
--- /dev/null
+++ b/addons/hr_contract/tests/test_contract.py
@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from datetime import datetime, date
+from odoo.exceptions import ValidationError
+from odoo.addons.hr_contract.tests.common import TestContractCommon
+
+
+class TestHrContracts(TestContractCommon):
+
+    @classmethod
+    def setUpClass(cls):
+        super(TestHrContracts, cls).setUpClass()
+        cls.contracts = cls.env['hr.contract'].with_context(tracking_disable=True)
+
+    def create_contract(self, state, kanban_state, start, end=None):
+        return self.env['hr.contract'].create({
+            'name': 'Contract',
+            'employee_id': self.employee.id,
+            'state': state,
+            'kanban_state': kanban_state,
+            'wage': 1,
+            'date_start': start,
+            'date_end': end,
+        })
+
+    def test_incoming_overlapping_contract(self):
+        start = datetime.strptime('2015-11-01', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-11-30', '%Y-%m-%d').date()
+        self.create_contract('open', 'normal', start, end)
+
+        # Incoming contract
+        with self.assertRaises(ValidationError, msg="It should not create two contract in state open or incoming"):
+            start = datetime.strptime('2015-11-15', '%Y-%m-%d').date()
+            end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+            self.create_contract('draft', 'done', start, end)
+
+    def test_pending_overlapping_contract(self):
+        start = datetime.strptime('2015-11-01', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-11-30', '%Y-%m-%d').date()
+        self.create_contract('open', 'normal', start, end)
+
+        # Pending contract
+        with self.assertRaises(ValidationError, msg="It should not create two contract in state open or pending"):
+            start = datetime.strptime('2015-11-15', '%Y-%m-%d').date()
+            end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+            self.create_contract('open', 'blocked', start, end)
+
+        # Draft contract -> should not raise
+        start = datetime.strptime('2015-11-15', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+        self.create_contract('draft', 'normal', start, end)
+
+    def test_draft_overlapping_contract(self):
+        start = datetime.strptime('2015-11-01', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-11-30', '%Y-%m-%d').date()
+        self.create_contract('open', 'normal', start, end)
+
+        # Draft contract -> should not raise even if overlapping
+        start = datetime.strptime('2015-11-15', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+        self.create_contract('draft', 'normal', start, end)
+
+    def test_overlapping_contract_no_end(self):
+
+        # No end date
+        self.create_contract('open', 'normal', datetime.strptime('2015-11-01', '%Y-%m-%d').date())
+
+        with self.assertRaises(ValidationError):
+            start = datetime.strptime('2015-11-15', '%Y-%m-%d').date()
+            end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+            self.create_contract('draft', 'done', start, end)
+
+    def test_overlapping_contract_no_end2(self):
+
+        start = datetime.strptime('2015-11-01', '%Y-%m-%d').date()
+        end = datetime.strptime('2015-12-30', '%Y-%m-%d').date()
+        self.create_contract('open', 'normal', start, end)
+
+        with self.assertRaises(ValidationError):
+            # No end
+            self.create_contract('draft', 'done', datetime.strptime('2015-01-01', '%Y-%m-%d').date())
+
+    def test_set_employee_contract_create(self):
+        contract = self.create_contract('open', 'normal', date(2018, 1, 1), date(2018, 1, 2))
+        self.assertEqual(self.employee.contract_id, contract)
+
+    def test_set_employee_contract_write(self):
+        contract = self.create_contract('draft', 'normal', date(2018, 1, 1), date(2018, 1, 2))
+        contract.state = 'open'
+        self.assertEqual(self.employee.contract_id, contract)