Initial commit: Test packages

odoo-bringout-oca-ocb-test_resource/test_resource/tests/__init__.py

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import test_resource
+from . import test_performance

odoo-bringout-oca-ocb-test_resource/test_resource/tests/common.py

@@ -0,0 +1,104 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo.tests.common import TransactionCase
+
+
+class TestResourceCommon(TransactionCase):
+
+    @classmethod
+    def _define_calendar(cls, name, attendances, tz):
+        return cls.env['resource.calendar'].create({
+            'name': name,
+            'tz': tz,
+            'attendance_ids': [
+                (0, 0, {
+                    'name': '%s_%d' % (name, index),
+                    'hour_from': att[0],
+                    'hour_to': att[1],
+                    'dayofweek': str(att[2]),
+                })
+                for index, att in enumerate(attendances)
+            ],
+        })
+
+    @classmethod
+    def _define_calendar_2_weeks(cls, name, attendances, tz):
+        return cls.env['resource.calendar'].create({
+            'name': name,
+            'tz': tz,
+            'two_weeks_calendar': True,
+            'attendance_ids': [
+                (0, 0, {
+                    'name': '%s_%d' % (name, index),
+                    'hour_from': att[0],
+                    'hour_to': att[1],
+                    'dayofweek': str(att[2]),
+                    'week_type': att[3],
+                    'display_type': att[4],
+                    'sequence': att[5],
+                })
+                for index, att in enumerate(attendances)
+            ],
+        })
+
+    @classmethod
+    def setUpClass(cls):
+        super(TestResourceCommon, cls).setUpClass()
+        cls.env.company.resource_calendar_id.tz = "Europe/Brussels"
+
+        # UTC+1 winter, UTC+2 summer
+        cls.calendar_jean = cls._define_calendar('40 Hours', [(8, 16, i) for i in range(5)], 'Europe/Brussels')
+        # UTC+6
+        cls.calendar_patel = cls._define_calendar('38 Hours', sum([((9, 12, i), (13, 17, i)) for i in range(5)], ()), 'Etc/GMT-6')
+        # UTC-8 winter, UTC-7 summer
+        cls.calendar_john = cls._define_calendar('8+12 Hours', [(8, 16, 1), (8, 13, 4), (16, 23, 4)], 'America/Los_Angeles')
+        # UTC+1 winter, UTC+2 summer
+        cls.calendar_jules = cls._define_calendar_2_weeks('Week 1: 30 Hours - Week 2: 16 Hours', [
+            (0, 0, 0, '0', 'line_section', 0), (8, 16, 0, '0', False, 1), (9, 17, 1, '0', False, 2),
+            (0, 0, 0, '1', 'line_section', 10), (8, 16, 0, '1', False, 11), (7, 15, 2, '1', False, 12),
+            (8, 16, 3, '1', False, 13), (10, 16, 4, '1', False, 14)], 'Europe/Brussels')
+
+        cls.calendar_paul = cls._define_calendar('Morning and evening shifts', sum([((2, 7, i), (10, 16, i)) for i in range(5)], ()), 'America/Noronha')
+
+        # Employee is linked to a resource.resource via resource.mixin
+        cls.jean = cls.env['resource.test'].create({
+            'name': 'Jean',
+            'resource_calendar_id': cls.calendar_jean.id,
+        })
+        cls.patel = cls.env['resource.test'].create({
+            'name': 'Patel',
+            'resource_calendar_id': cls.calendar_patel.id,
+        })
+        cls.john = cls.env['resource.test'].create({
+            'name': 'John',
+            'resource_calendar_id': cls.calendar_john.id,
+        })
+        cls.jules = cls.env['resource.test'].create({
+            'name': 'Jules',
+            'resource_calendar_id': cls.calendar_jules.id,
+        })
+
+        cls.paul = cls.env['resource.test'].create({
+            'name': 'Paul',
+            'resource_calendar_id': cls.calendar_paul.id,
+        })
+
+        cls.two_weeks_resource = cls._define_calendar_2_weeks(
+            'Two weeks resource',
+            [
+                (0, 0, 0, '0', 'line_section', 0),
+                (8, 16, 0, '0', False, 1),
+                (8, 16, 1, '0', False, 2),
+                (8, 16, 2, '0', False, 3),
+                (8, 16, 3, '0', False, 4),
+                (8, 16, 4, '0', False, 5),
+                (0, 0, 0, '1', 'line_section', 10),
+                (8, 16, 0, '1', False, 11),
+                (8, 16, 1, '1', False, 12),
+                (8, 16, 2, '1', False, 13),
+                (8, 16, 3, '1', False, 14),
+                (8, 16, 4, '1', False, 15)
+            ],
+            'Europe/Brussels'
+        )

odoo-bringout-oca-ocb-test_resource/test_resource/tests/test_performance.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+import logging
+import time
+import pytz
+
+from datetime import datetime
+from dateutil.relativedelta import relativedelta
+
+from odoo.tests import TransactionCase, warmup
+
+_logger = logging.getLogger(__name__)
+
+class TestResourcePerformance(TransactionCase):
+
+    @warmup
+    def test_performance_attendance_intervals_batch(self):
+        # Tests the performance of _attendance_intervals_batch with a batch of 100 resources
+        calendar = self.env['resource.calendar'].create({
+            'name': 'Calendar',
+        })
+        resources = self.env['resource.test'].create([
+            {
+                'name': 'Resource ' + str(i),
+                'resource_calendar_id': calendar.id,
+            }
+            for i in range(100)
+        ])
+        with self.assertQueryCount(__system__=1):
+            # Generate attendances for a whole year
+            start = pytz.utc.localize(datetime.now() + relativedelta(month=1, day=1))
+            stop = pytz.utc.localize(datetime.now() + relativedelta(month=12, day=31))
+            start_time = time.time()
+            calendar._attendance_intervals_batch(start, stop, resources=resources)
+            _logger.info('Attendance Intervals Batch (100): --- %s seconds ---', time.time() - start_time)
+            # Before
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 2.0667169094085693 seconds ---
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 2.0868310928344727 seconds ---
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 1.9209258556365967 seconds ---
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 1.9474620819091797 seconds ---
+            # After
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 0.4092371463775635 seconds ---
+            #INFO master test_performance: Attendance Intervals Batch (100): --- 0.3598649501800537 seconds ---