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')
|
||||
|
||||
Reference in New Issue
Block a user