flectra/addons/resource/tests/common.py

# -*- coding: utf-8 -*-
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.

from flectra.fields import Datetime
from flectra.tests.common import TransactionCase


class TestResourceCommon(TransactionCase):

    def setUp(self):
        super(TestResourceCommon, self).setUp()
        self.env.user.tz = 'UTC'

        # Some demo data
        self.date1 = Datetime.from_string('2013-02-12 09:08:07')  # weekday() returns 1, isoweekday() returns 2
        self.date2 = Datetime.from_string('2013-02-15 10:11:12')  # weekday() returns 4, isoweekday() returns 5

        # Resource data
        # Calendar working days: 1 (8-16 -> 8hours), 4 (8-13, 16-23 -> 12hours)
        self.calendar = self.env['resource.calendar'].create({
            'name': 'TestCalendar',
            'attendance_ids': [(5, 0, 0)]
        })
        self.att_1 = self.env['resource.calendar.attendance'].create({
            'name': 'Att1',
            'calendar_id': self.calendar.id,
            'dayofweek': '1',
            'hour_from': 8,
            'hour_to': 16
        })
        self.att_2 = self.env['resource.calendar.attendance'].create({
            'name': 'Att2',
            'calendar_id': self.calendar.id,
            'dayofweek': '4',
            'hour_from': 8,
            'hour_to': 13
        })
        self.att_3 = self.env['resource.calendar.attendance'].create({
            'name': 'Att3',
            'calendar_id': self.calendar.id,
            'dayofweek': '4',
            'hour_from': 16,
            'hour_to': 23
        })

        self.resource1_id = self.env['resource.resource'].create(
            {
                'name': 'TestResource1',
                'resource_type': 'user',
                'time_efficiency': 150.0,
                'calendar_id': self.calendar.id,
            }
        ).id

        # Leave1: 19/02/2013, from 9 to 12, is a day 1
        self.leave1 = self.env['resource.calendar.leaves'].create({
            'name': 'GenericLeave',
            'calendar_id': self.calendar.id,
            'date_from': Datetime.from_string('2013-02-19 09:00:00'),
            'date_to': Datetime.from_string('2013-02-19 12:00:00')})
        # Leave2: 22/02/2013, from 9 to 15, is a day 4
        self.leave2 = self.env['resource.calendar.leaves'].create({
            'name': 'ResourceLeave',
            'calendar_id': self.calendar.id,
            'resource_id': self.resource1_id,
            'date_from': Datetime.from_string('2013-02-22 09:00:00'),
            'date_to': Datetime.from_string('2013-02-22 15:00:00')})
        # Leave3: 25/02/2013 (day0) -> 01/03/2013 (day4)
        self.leave3 = self.env['resource.calendar.leaves'].create({
            'name': 'ResourceLeave2',
            'calendar_id': self.calendar.id,
            'resource_id': self.resource1_id,
            'date_from': Datetime.from_string('2013-02-25 13:00:00'),
            'date_to': Datetime.from_string('2013-03-01 11:30:00')})