mirror of
https://github.com/OCA/pms.git
synced 2025-01-29 00:17:45 +02:00
[WIP] Preparing tests
This commit is contained in:
Pablo
@@ -23,19 +23,16 @@
|
||||
from datetime import timedelta
|
||||
from odoo import api, fields
|
||||
from odoo.tests import common
|
||||
from openerp.tools import (
|
||||
from odoo.tools import (
|
||||
DEFAULT_SERVER_DATE_FORMAT,
|
||||
DEFAULT_SERVER_DATETIME_FORMAT)
|
||||
from odoo.addons.mail.tests.common import TestMail
|
||||
from odoo.addons.hotel import date_utils
|
||||
import pytz
|
||||
import logging
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# TestMail crea recursos utiles para nuestros test...
|
||||
# por ejemplo, usuarios con distintos tipos de nivel, etc...
|
||||
class TestHotel(TestMail):
|
||||
class TestHotel(common.SavepointCase):
|
||||
|
||||
@classmethod
|
||||
def _init_mock_hotel(cls):
|
||||
@@ -67,13 +64,13 @@ class TestHotel(TestMail):
|
||||
"Hotel Calendar can't create a new reservation!")
|
||||
|
||||
# Create Reservation Lines + Update Reservation Price
|
||||
days_diff = date_utils.date_diff(checkin, checkout, hours=False)
|
||||
res = reservation.sudo(creator).prepare_reservation_lines(
|
||||
checkin.strftime(DEFAULT_SERVER_DATETIME_FORMAT), days_diff)
|
||||
reservation.sudo(creator).write({
|
||||
'reservation_lines': res['commands'],
|
||||
'price_unit': res['total_price'],
|
||||
})
|
||||
# days_diff = date_utils.date_diff(checkin, checkout, hours=False)
|
||||
# res = reservation.sudo(creator).prepare_reservation_lines(
|
||||
# checkin.strftime(DEFAULT_SERVER_DATETIME_FORMAT), days_diff)
|
||||
# reservation.sudo(creator).write({
|
||||
# 'reservation_lines': res['commands'],
|
||||
# 'price_unit': res['total_price'],
|
||||
# })
|
||||
|
||||
return reservation
|
||||
|
||||
@@ -83,169 +80,16 @@ class TestHotel(TestMail):
|
||||
|
||||
cls._init_mock_hotel()
|
||||
|
||||
# Restriction Plan
|
||||
cls.restriction_1 = cls.env['hotel.room.type.restriction'].create({
|
||||
'name': 'Restriction Test #1',
|
||||
'active': True
|
||||
})
|
||||
|
||||
# Pricelist
|
||||
cls.pricelist_1 = cls.env['product.pricelist'].create({
|
||||
'name': 'Pricelist Test #1',
|
||||
})
|
||||
|
||||
# Minimal Hotel Configuration
|
||||
cls.tz_hotel = 'Europe/Madrid'
|
||||
cls.default_pricelist_id = cls.pricelist_1.id
|
||||
cls.default_restriction_id = cls.restriction_1.id
|
||||
cls.env['ir.values'].sudo().set_default('res.config.settings',
|
||||
'tz_hotel', cls.tz_hotel)
|
||||
cls.env['ir.values'].sudo().set_default('res.config.settings',
|
||||
'default_pricelist_id',
|
||||
cls.default_pricelist_id)
|
||||
cls.env['ir.values'].sudo().set_default('res.config.settings',
|
||||
'default_restriction_id',
|
||||
cls.default_restriction_id)
|
||||
|
||||
# User Groups
|
||||
user_group_hotel_manager = cls.env.ref('hotel.group_hotel_manager')
|
||||
user_group_hotel_user = cls.env.ref('hotel.group_hotel_user')
|
||||
user_group_employee = cls.env.ref('base.group_user')
|
||||
user_group_public = cls.env.ref('base.group_public')
|
||||
user_group_account_inv = cls.env.ref('account.group_account_invoice')
|
||||
user_group_sale_manager = cls.env.ref('sales_team.group_sale_manager')
|
||||
user_group_base_partner_manager = cls.env.ref(
|
||||
'base.group_partner_manager')
|
||||
|
||||
# Create Test Users
|
||||
Users = cls.env['res.users'].with_context({
|
||||
'no_reset_password': True,
|
||||
'mail_create_nosubscribe': True
|
||||
})
|
||||
cls.user_hotel_manager = Users.create({
|
||||
'name': 'Jeff Hotel Manager',
|
||||
'login': 'hoteljeff',
|
||||
'email': 'mynameisjeff@example.com',
|
||||
'signature': '--\nJeff',
|
||||
'notify_email': 'always',
|
||||
'groups_id': [(6, 0, [user_group_hotel_manager.id,
|
||||
user_group_employee.id,
|
||||
user_group_account_inv.id,
|
||||
user_group_sale_manager.id,
|
||||
user_group_base_partner_manager.id])]
|
||||
})
|
||||
cls.user_hotel_user = Users.create({
|
||||
'name': 'Juancho Hotel User',
|
||||
'login': 'juancho',
|
||||
'email': 'juancho@example.com',
|
||||
'signature': '--\nJuancho',
|
||||
'notify_email': 'always',
|
||||
'groups_id': [(6, 0, [user_group_hotel_user.id,
|
||||
user_group_public.id])]
|
||||
})
|
||||
|
||||
# Create Tests Records
|
||||
RoomTypes = cls.env['hotel.room.type']
|
||||
cls.hotel_room_type_simple = RoomTypes.create({
|
||||
'name': 'Simple',
|
||||
'code_type': 'TSMP',
|
||||
})
|
||||
cls.hotel_room_type_double = RoomTypes.create({
|
||||
'name': 'Double',
|
||||
'code_type': 'TDBL',
|
||||
})
|
||||
cls.room_type_0 = cls.env.ref('hotel_room_type_0')
|
||||
cls.room_type_1 = cls.env.ref('hotel_room_type_1')
|
||||
cls.room_type_2 = cls.env.ref('hotel_room_type_2')
|
||||
cls.room_type_3 = cls.env.ref('hotel_room_type_3')
|
||||
|
||||
VRooms = cls.env['hotel.virtual.room']
|
||||
cls.hotel_room_type_budget = VRooms.create({
|
||||
'name': 'Budget Room',
|
||||
'virtual_code': '001',
|
||||
'list_price': 50,
|
||||
})
|
||||
cls.hotel_room_type_special = VRooms.create({
|
||||
'name': 'Special Room',
|
||||
'virtual_code': '002',
|
||||
'list_price': 150,
|
||||
})
|
||||
|
||||
Rooms = cls.env['hotel.room']
|
||||
cls.hotel_room_simple_100 = Rooms.create({
|
||||
'name': '100',
|
||||
'sale_price_type': 'room_type',
|
||||
'price_room_type': cls.hotel_room_type_budget.id,
|
||||
'categ_id': cls.hotel_room_type_simple.cat_id.id,
|
||||
'capacity': 1,
|
||||
})
|
||||
cls.hotel_room_simple_101 = Rooms.create({
|
||||
'name': '101',
|
||||
'sale_price_type': 'room_type',
|
||||
'price_room_type': cls.hotel_room_type_budget.id,
|
||||
'categ_id': cls.hotel_room_type_simple.cat_id.id,
|
||||
'capacity': 1,
|
||||
'sequence': 1,
|
||||
})
|
||||
cls.hotel_room_double_200 = Rooms.create({
|
||||
'name': '200',
|
||||
'sale_price_type': 'room_type',
|
||||
'price_room_type': cls.hotel_room_type_special.id,
|
||||
'categ_id': cls.hotel_room_type_double.cat_id.id,
|
||||
'capacity': 2,
|
||||
})
|
||||
|
||||
cls.hotel_room_type_budget.write({
|
||||
'room_ids': [(6, False, [cls.hotel_room_simple_100.id,
|
||||
cls.hotel_room_simple_101.id])],
|
||||
})
|
||||
cls.hotel_room_type_special.write({
|
||||
'room_ids': [(6, False, [cls.hotel_room_double_200.id])],
|
||||
})
|
||||
|
||||
# Create a week of fresh data
|
||||
now_utc_dt = date_utils.now()
|
||||
cls.avails_tmp = {
|
||||
cls.hotel_room_type_budget.id: (1, 2, 2, 1, 1, 2, 2),
|
||||
cls.hotel_room_type_special.id: (1, 1, 1, 1, 1, 1, 1),
|
||||
}
|
||||
cls.prices_tmp = {
|
||||
cls.hotel_room_type_budget.id: (10.0, 80.0, 80.0, 95.0, 90.0, 80.0,
|
||||
20.0),
|
||||
cls.hotel_room_type_special.id: (5.0, 15.0, 15.0, 35.0, 35.0, 10.0,
|
||||
10.0),
|
||||
}
|
||||
cls.restrictions_min_stay_tmp = {
|
||||
cls.hotel_room_type_budget.id: (0, 1, 2, 1, 1, 0, 0),
|
||||
cls.hotel_room_type_special.id: (3, 1, 0, 2, 0, 1, 4),
|
||||
}
|
||||
budget_product_id = cls.hotel_room_type_budget.product_id
|
||||
special_product_id = cls.hotel_room_type_special.product_id
|
||||
product_tmpl_ids = {
|
||||
cls.hotel_room_type_budget.id: budget_product_id.product_tmpl_id.id,
|
||||
cls.hotel_room_type_special.id: special_product_id.product_tmpl_id.id,
|
||||
}
|
||||
room_type_avail_obj = cls.env['hotel.room.type.availability']
|
||||
room_type_rest_item_obj = cls.env['hotel.room.type.restriction.item']
|
||||
pricelist_item_obj = cls.env['product.pricelist.item']
|
||||
for k_vr, v_vr in cls.avails_tmp.iteritems():
|
||||
for i in range(0, len(v_vr)):
|
||||
ndate = now_utc_dt + timedelta(days=i)
|
||||
room_type_avail_obj.create({
|
||||
'room_type_id': k_vr,
|
||||
'avail': v_vr[i],
|
||||
'date': ndate.strftime(DEFAULT_SERVER_DATE_FORMAT)
|
||||
})
|
||||
room_type_rest_item_obj.create({
|
||||
'room_type_id': k_vr,
|
||||
'restriction_id': cls.default_restriction_id,
|
||||
'date_start': ndate.strftime(DEFAULT_SERVER_DATE_FORMAT),
|
||||
'date_end': ndate.strftime(DEFAULT_SERVER_DATE_FORMAT),
|
||||
'applied_on': '0_room_type',
|
||||
'min_stay': cls.restrictions_min_stay_tmp[k_vr][i],
|
||||
})
|
||||
pricelist_item_obj.create({
|
||||
'pricelist_id': cls.default_pricelist_id,
|
||||
'date_start': ndate.strftime(DEFAULT_SERVER_DATE_FORMAT),
|
||||
'date_end': ndate.strftime(DEFAULT_SERVER_DATE_FORMAT),
|
||||
'compute_price': 'fixed',
|
||||
'applied_on': '1_product',
|
||||
'product_tmpl_id': product_tmpl_ids[k_vr],
|
||||
'fixed_price': cls.prices_tmp[k_vr][i],
|
||||
})
|
||||
cls.room_0 = cls.env.ref('hotel_room_0')
|
||||
cls.room_1 = cls.env.ref('hotel_room_1')
|
||||
cls.room_2 = cls.env.ref('hotel_room_2')
|
||||
cls.room_3 = cls.env.ref('hotel_room_3')
|
||||
cls.room_4 = cls.env.ref('hotel_room_4')
|
||||
cls.room_5 = cls.env.ref('hotel_room_5')
|
||||
cls.room_6 = cls.env.ref('hotel_room_6')
|
||||
|
||||
55
hotel/tests/test_hotel_room_model.py
Normal file
55
hotel/tests/test_hotel_room_model.py
Normal file
@@ -0,0 +1,55 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
##############################################################################
|
||||
#
|
||||
# OpenERP, Open Source Management Solution
|
||||
# Copyright (C) 2017 Solucións Aloxa S.L. <info@aloxa.eu>
|
||||
# Alexandre Díaz <dev@redneboa.es>
|
||||
#
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU 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 General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
##############################################################################
|
||||
from datetime import timedelta
|
||||
from .common import TestHotel
|
||||
from odoo.addons.hotel import date_utils
|
||||
|
||||
|
||||
class TestHotelRoom(TestHotel):
|
||||
|
||||
def test_cancel_folio(self):
|
||||
now_utc_dt = date_utils.now()
|
||||
|
||||
org_reserv_start_utc_dt = now_utc_dt + timedelta(days=3)
|
||||
org_reserv_end_utc_dt = org_reserv_start_utc_dt + timedelta(days=6)
|
||||
folio = self.create_folio(self.user_hotel_manager, self.partner_2)
|
||||
reservation_a = self.create_reservation(
|
||||
self.user_hotel_manager,
|
||||
folio,
|
||||
org_reserv_start_utc_dt,
|
||||
org_reserv_end_utc_dt,
|
||||
self.hotel_room_double_200,
|
||||
"Reservation Test #1")
|
||||
reservation_b = self.create_reservation(
|
||||
self.user_hotel_manager,
|
||||
folio,
|
||||
org_reserv_start_utc_dt,
|
||||
org_reserv_end_utc_dt,
|
||||
self.hotel_room_simple_100,
|
||||
"Reservation Test #2")
|
||||
self.assertEqual(len(folio.room_lines), 2, 'Invalid room lines count')
|
||||
folio.action_cancel()
|
||||
self.assertEqual(folio.state, 'cancel', 'Invalid folio state')
|
||||
for rline in folio.room_lines:
|
||||
self.assertEqual(rline.state, 'cancelled',
|
||||
'Invalid reservation state')
|
||||
Reference in New Issue
Block a user