forked from Yaltik/golem
146 lines
6.1 KiB
Python
146 lines
6.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
|
|
# Copyright 2018 Fabien Bourgeois <fabien@yaltik.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/>.
|
|
|
|
""" GOLEM Resource Pack testing """
|
|
|
|
import logging
|
|
from odoo.tests.common import TransactionCase
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class TestGolemResourcePack(TransactionCase):
|
|
""" GOLEM Resource Reservation testing """
|
|
|
|
def setUp(self, *args, **kwargs):
|
|
""" Bootstrap Resource Reservation """
|
|
super(TestGolemResourcePack, self).setUp(*args, **kwargs)
|
|
|
|
self.resource = self.env['golem.resource'].create({
|
|
'name': 'Resource',
|
|
'availability_start': '2018-01-01',
|
|
'availability_stop': '2020-01-01',
|
|
'availability_24_7': True
|
|
})
|
|
self.resource_val = self.env['golem.resource'].create({
|
|
'name': 'Resource to validate',
|
|
'availability_start': '2018-01-01',
|
|
'availability_stop': '2020-01-01',
|
|
'validation_required': True,
|
|
'availability_24_7': True
|
|
})
|
|
|
|
self.partner = self.env['res.partner'].create({'firstname': 'John',
|
|
'lastname': 'DOE',
|
|
'is_company': False})
|
|
|
|
reservation_obj = self.env['golem.resource.reservation']
|
|
self.reservation_1 = reservation_obj.create({
|
|
'resource_id': self.resource.id,
|
|
'date_start': '2018-02-05 11:00:00',
|
|
'date_stop': '2018-02-05 12:00:00',
|
|
'partner_id': self.partner.id
|
|
})
|
|
self.reservation_2 = reservation_obj.create({
|
|
'resource_id': self.resource.id,
|
|
'date_start': '2018-02-06 11:00:00',
|
|
'date_stop': '2018-02-06 12:00:00',
|
|
'partner_id': self.partner.id
|
|
})
|
|
self.reservation_3 = reservation_obj.create({
|
|
'resource_id': self.resource_val.id,
|
|
'date_start': '2018-02-05 11:00:00', # is monday
|
|
'date_stop': '2018-02-05 12:00:00',
|
|
'partner_id': self.partner.id
|
|
})
|
|
self.pack_obj = self.env['golem.resource.pack']
|
|
self.pack_data = {
|
|
'name': 'Pack Test',
|
|
'partner_id': self.partner.id
|
|
}
|
|
|
|
|
|
def test_pack_basic(self):
|
|
""" Test pack bases """
|
|
pack = self.pack_obj.create(self.pack_data)
|
|
self.assertFalse(pack.reservation_ids)
|
|
self.assertFalse(self.reservation_1.pack_id)
|
|
pack.reservation_ids = [(4, self.reservation_1.id, 0),
|
|
(4, self.reservation_2.id, 0)]
|
|
self.assertEqual(pack.reservation_ids[0].state, "draft")
|
|
self.assertEqual(self.reservation_1.pack_id.id, pack.id)
|
|
self.assertEqual(pack.state, "draft")
|
|
self.assertEqual(pack.reservation_count, 2)
|
|
#confirm pack ==> all validated
|
|
pack.state_confirm()#no validation required
|
|
self.assertEqual(pack.reservation_ids[0].state, "validated")
|
|
self.assertEqual(pack.state, "validated")
|
|
#pack draft ==> all draft
|
|
pack.state_draft()
|
|
self.assertEqual(pack.reservation_ids[0].state, "draft")
|
|
self.assertEqual(pack.state, "draft")
|
|
# confirm pack and draft reservation ==> pack draft
|
|
pack.state_confirm()
|
|
pack.reservation_ids[0].state_draft()
|
|
self.assertEqual(pack.reservation_ids[0].state, "draft")
|
|
self.assertEqual(pack.state, "draft")
|
|
#confirm reservation ==> pack validated
|
|
pack.reservation_ids[0].state_confirm()
|
|
self.assertEqual(pack.reservation_ids[0].state, "validated")
|
|
self.assertEqual(pack.state, "validated")
|
|
pack.reservation_ids = [(5, 0, 0)]
|
|
self.assertFalse(pack.reservation_ids)
|
|
self.assertEqual(pack.reservation_count, 0)
|
|
|
|
|
|
def test_pack_validation(self):
|
|
""" Test pack validation """
|
|
pack = self.pack_obj.create(self.pack_data)
|
|
self.assertFalse(pack.reservation_ids)
|
|
self.assertFalse(self.reservation_1.pack_id)
|
|
pack.reservation_ids = [(4, self.reservation_1.id, 0),
|
|
(4, self.reservation_3.id, 0)]
|
|
self.assertEqual(pack.reservation_ids[0].state, "draft")
|
|
self.assertEqual(self.reservation_1.pack_id.id, pack.id)
|
|
self.assertEqual(pack.state, "draft")
|
|
pack.state_confirm()#validation required
|
|
self.assertEqual(pack.state, "confirmed")
|
|
pack.state_validated()
|
|
self.assertEqual(pack.state, "validated")
|
|
self.assertEqual(pack.reservation_ids[0].state, "validated")
|
|
self.assertEqual(pack.reservation_ids[1].state, "validated")
|
|
|
|
def test_pack_rejection(self):
|
|
""" test pack rejection """
|
|
pack = self.pack_obj.create(self.pack_data)
|
|
self.assertFalse(pack.reservation_ids)
|
|
self.assertFalse(self.reservation_1.pack_id)
|
|
pack.reservation_ids = [(4, self.reservation_1.id, 0),
|
|
(4, self.reservation_3.id, 0)]
|
|
self.assertEqual(pack.reservation_ids[0].state, "draft")
|
|
self.assertEqual(self.reservation_1.pack_id.id, pack.id)
|
|
self.assertEqual(pack.state, "draft")
|
|
pack.state_confirm()#validation required
|
|
rej_wizard = self.env['golem.pack.rejection.wizard'].create({
|
|
'pack_id': pack.id,
|
|
'reason' : 'reason1'
|
|
})
|
|
rej_wizard.reject()
|
|
self.assertEqual(pack.state, "rejected")
|
|
self.assertEqual(self.reservation_3.state, "rejected")
|
|
self.assertEqual(pack.rejection_reason, 'reason1')
|