125 lines
5.1 KiB
Python
125 lines
5.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Business Applications
|
|
# Copyright (c) 2013-TODAY OpenERP S.A. <http://www.openerp.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
from datetime import datetime
|
|
|
|
from openerp.tests import common
|
|
|
|
|
|
class TestResourceCommon(common.TransactionCase):
|
|
|
|
def setUp(self):
|
|
super(TestResourceCommon, self).setUp()
|
|
cr, uid = self.cr, self.uid
|
|
self.context = context = dict(tz='UTC')
|
|
|
|
self.registry('res.users').write(cr, uid, uid, {'tz': 'UTC'}, context=context)
|
|
|
|
# Usefull models
|
|
self.resource_resource = self.registry('resource.resource')
|
|
self.resource_calendar = self.registry('resource.calendar')
|
|
self.resource_attendance = self.registry('resource.calendar.attendance')
|
|
self.resource_leaves = self.registry('resource.calendar.leaves')
|
|
|
|
# Some demo data
|
|
self.date1 = datetime.strptime('2013-02-12 09:08:07', '%Y-%m-%d %H:%M:%S') # weekday() returns 1, isoweekday() returns 2
|
|
self.date2 = datetime.strptime('2013-02-15 10:11:12', '%Y-%m-%d %H:%M:%S') # weekday() returns 4, isoweekday() returns 5
|
|
# Leave1: 19/02/2013, from 9 to 12, is a day 1
|
|
self.leave1_start = datetime.strptime('2013-02-19 09:00:00', '%Y-%m-%d %H:%M:%S')
|
|
self.leave1_end = datetime.strptime('2013-02-19 12:00:00', '%Y-%m-%d %H:%M:%S')
|
|
# Leave2: 22/02/2013, from 9 to 15, is a day 4
|
|
self.leave2_start = datetime.strptime('2013-02-22 09:00:00', '%Y-%m-%d %H:%M:%S')
|
|
self.leave2_end = datetime.strptime('2013-02-22 15:00:00', '%Y-%m-%d %H:%M:%S')
|
|
# Leave3: 25/02/2013 (day0) -> 01/03/2013 (day4)
|
|
self.leave3_start = datetime.strptime('2013-02-25 13:00:00', '%Y-%m-%d %H:%M:%S')
|
|
self.leave3_end = datetime.strptime('2013-03-01 11:30:00', '%Y-%m-%d %H:%M:%S')
|
|
|
|
# Resource data
|
|
# Calendar working days: 1 (8-16 -> 8hours), 4 (8-13, 16-23 -> 12hours)
|
|
self.calendar_id = self.resource_calendar.create(
|
|
cr, uid, {
|
|
'name': 'TestCalendar',
|
|
}, context=context
|
|
)
|
|
self.att1_id = self.resource_attendance.create(
|
|
cr, uid, {
|
|
'name': 'Att1',
|
|
'dayofweek': '1',
|
|
'hour_from': 8,
|
|
'hour_to': 16,
|
|
'calendar_id': self.calendar_id,
|
|
}, context=context
|
|
)
|
|
self.att2_id = self.resource_attendance.create(
|
|
cr, uid, {
|
|
'name': 'Att2',
|
|
'dayofweek': '4',
|
|
'hour_from': 8,
|
|
'hour_to': 13,
|
|
'calendar_id': self.calendar_id,
|
|
}, context=context
|
|
)
|
|
self.att3_id = self.resource_attendance.create(
|
|
cr, uid, {
|
|
'name': 'Att3',
|
|
'dayofweek': '4',
|
|
'hour_from': 16,
|
|
'hour_to': 23,
|
|
'calendar_id': self.calendar_id,
|
|
}, context=context
|
|
)
|
|
self.resource1_id = self.resource_resource.create(
|
|
cr, uid, {
|
|
'name': 'TestResource1',
|
|
'resource_type': 'user',
|
|
'time_efficiency': 150.0,
|
|
'calendar_id': self.calendar_id,
|
|
}, context=context
|
|
)
|
|
self.leave1_id = self.resource_leaves.create(
|
|
cr, uid, {
|
|
'name': 'GenericLeave',
|
|
'calendar_id': self.calendar_id,
|
|
'date_from': self.leave1_start,
|
|
'date_to': self.leave1_end,
|
|
}, context=context
|
|
)
|
|
self.leave2_id = self.resource_leaves.create(
|
|
cr, uid, {
|
|
'name': 'ResourceLeave',
|
|
'calendar_id': self.calendar_id,
|
|
'resource_id': self.resource1_id,
|
|
'date_from': self.leave2_start,
|
|
'date_to': self.leave2_end,
|
|
}, context=context
|
|
)
|
|
self.leave3_id = self.resource_leaves.create(
|
|
cr, uid, {
|
|
'name': 'ResourceLeave2',
|
|
'calendar_id': self.calendar_id,
|
|
'resource_id': self.resource1_id,
|
|
'date_from': self.leave3_start,
|
|
'date_to': self.leave3_end,
|
|
}, context=context
|
|
)
|
|
# Some browse data
|
|
self.calendar = self.resource_calendar.browse(cr, uid, self.calendar_id, context=context)
|