audrius/pms
1
0
Fork 1
mirror of https://github.com/OCA/pms.git synced 2025-01-29 00:17:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

[REF] pms: setup test data in class

Browse Source
This commit is contained in:
Robin Keunen
2023-06-22 16:47:02 +02:00
committed by Darío Lodeiros
parent 68a4984909
commit 4aa8814410
25 changed files with 375 additions and 350 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pms/tests/common.py
View File

@@ -2,36 +2,37 @@ from odoo.tests import common
class TestPms(common.SavepointCase):
def setUp(self):
super().setUp()
self.availability_plan1 = self.env["pms.availability.plan"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.availability_plan1 = cls.env["pms.availability.plan"].create(
{"name": "Availability Plan 1"}
)
self.pricelist1 = self.env["product.pricelist"].create(
cls.pricelist1 = cls.env["product.pricelist"].create(
{
"name": "Pricelist 1",
"availability_plan_id": self.availability_plan1.id,
"availability_plan_id": cls.availability_plan1.id,
}
)
self.company1 = self.env["res.company"].create(
cls.company1 = cls.env["res.company"].create(
{
"name": "Company 1",
}
)
self.pms_property1 = self.env["pms.property"].create(
cls.pms_property1 = cls.env["pms.property"].create(
{
"name": "Property 1",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.room_type_class1 = self.env["pms.room.type.class"].create(
cls.room_type_class1 = cls.env["pms.room.type.class"].create(
{
"name": "Room Type Class 1",
"default_code": "RTC1",
}
)
for pricelist in self.env["product.pricelist"].search([]):
for pricelist in cls.env["product.pricelist"].search([]):
if not pricelist.availability_plan_id:
pricelist.availability_plan_id = self.availability_plan1.id
pricelist.availability_plan_id = cls.availability_plan1.id
pricelist.is_pms_available = True

7
pms/tests/test_automated_mails.py
View File

@@ -4,9 +4,10 @@ from .common import TestPms
class TestPmsAutomatedMails(TestPms):
def setUp(self):
super().setUp()
self.template = self.env["mail.template"].search(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.template = cls.env["mail.template"].search(
[("name", "=", "Confirmed Reservation")]
)

75
pms/tests/test_pms_availability_plan_rules.py
View File

@@ -7,114 +7,115 @@ from .common import TestPms
class TestPmsRoomTypeAvailabilityRules(TestPms):
def setUp(self):
super().setUp()
self.pms_property2 = self.env["pms.property"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.pms_property2 = cls.env["pms.property"].create(
{
"name": "Property 2",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.pricelist2 = self.env["product.pricelist"].create(
cls.pricelist2 = cls.env["product.pricelist"].create(
{
"name": "test pricelist 1",
"pms_property_ids": [
(4, self.pms_property1.id),
(4, self.pms_property2.id),
(4, cls.pms_property1.id),
(4, cls.pms_property2.id),
],
"availability_plan_id": self.availability_plan1.id,
"availability_plan_id": cls.availability_plan1.id,
"is_pms_available": True,
}
)
# pms.sale.channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",
}
)
# pms.availability.plan
self.test_room_type_availability1 = self.env["pms.availability.plan"].create(
cls.test_room_type_availability1 = cls.env["pms.availability.plan"].create(
{
"name": "Availability plan for TEST",
"pms_pricelist_ids": [(6, 0, [self.pricelist2.id])],
"pms_pricelist_ids": [(6, 0, [cls.pricelist2.id])],
}
)
# pms.property
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "MY PMS TEST",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist2.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist2.id,
}
)
self.pricelist2.write(
cls.pricelist2.write(
{
"pms_property_ids": [
(4, self.pms_property3.id),
(4, cls.pms_property3.id),
],
}
)
# pms.room.type
self.test_room_type_single = self.env["pms.room.type"].create(
cls.test_room_type_single = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property3.id],
"pms_property_ids": [cls.pms_property3.id],
"name": "Single Test",
"default_code": "SNG_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# pms.room.type
self.test_room_type_double = self.env["pms.room.type"].create(
cls.test_room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [
(4, self.pms_property3.id),
(4, cls.pms_property3.id),
],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# pms.room
self.test_room1_double = self.env["pms.room"].create(
cls.test_room1_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property3.id,
"pms_property_id": cls.pms_property3.id,
"name": "Double 201 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
# pms.room
self.test_room2_double = self.env["pms.room"].create(
cls.test_room2_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property3.id,
"pms_property_id": cls.pms_property3.id,
"name": "Double 202 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
self.test_room1_single = self.env["pms.room"].create(
cls.test_room1_single = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property3.id,
"pms_property_id": cls.pms_property3.id,
"name": "Single 101 test",
"room_type_id": self.test_room_type_single.id,
"room_type_id": cls.test_room_type_single.id,
"capacity": 1,
}
)
# pms.room
self.test_room2_single = self.env["pms.room"].create(
cls.test_room2_single = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property3.id,
"pms_property_id": cls.pms_property3.id,
"name": "Single 102 test",
"room_type_id": self.test_room_type_single.id,
"room_type_id": cls.test_room_type_single.id,
"capacity": 1,
}
)
# partner
self.partner1 = self.env["res.partner"].create(
{"name": "Charles", "property_product_pricelist": self.pricelist1}
cls.partner1 = cls.env["res.partner"].create(
{"name": "Charles", "property_product_pricelist": cls.pricelist1}
)
def test_availability_rooms_all(self):

13
pms/tests/test_pms_board_service.py
View File

@@ -7,18 +7,19 @@ from .common import TestPms
class TestBoardService(TestPms):
def setUp(self):
super().setUp()
self.company2 = self.env["res.company"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.company2 = cls.env["res.company"].create(
{
"name": "Company 2",
}
)
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "Property 3",
"company_id": self.company2.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company2.id,
"default_pricelist_id": cls.pricelist1.id,
}
)

39
pms/tests/test_pms_booking_engine.py
View File

@@ -8,61 +8,62 @@ from .common import TestPms
class TestPmsBookingEngine(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# CREATION OF ROOM TYPE (WITH ROOM TYPE CLASS)
self.test_room_type_double = self.env["pms.room.type"].create(
cls.test_room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"list_price": 40.0,
}
)
# pms.room
self.test_room1_double = self.env["pms.room"].create(
cls.test_room1_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 201 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
# pms.room
self.test_room2_double = self.env["pms.room"].create(
cls.test_room2_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 202 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
# pms.room
self.test_room3_double = self.env["pms.room"].create(
cls.test_room3_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 203 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
# pms.room
self.test_room4_double = self.env["pms.room"].create(
cls.test_room4_double = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 204 test",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
# res.partner
self.partner_id = self.env["res.partner"].create(
cls.partner_id = cls.env["res.partner"].create(
{
"name": "Miguel",
"mobile": "654667733",
@@ -71,7 +72,7 @@ class TestPmsBookingEngine(TestPms):
)
# pms.sale.channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

69
pms/tests/test_pms_checkin_partner.py
View File

@@ -13,43 +13,44 @@ _logger = logging.getLogger(__name__)
class TestPmsCheckinPartner(TestPms):
@freeze_time("2012-01-14")
def setUp(self):
super().setUp()
self.room_type1 = self.env["pms.room.type"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
today = datetime.date(2012, 1, 14)
cls.room_type1 = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Triple",
"default_code": "TRP",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Triple 101",
"room_type_id": self.room_type1.id,
"room_type_id": cls.room_type1.id,
"capacity": 3,
}
)
self.room1_2 = self.env["pms.room"].create(
cls.room1_2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Triple 111",
"room_type_id": self.room_type1.id,
"room_type_id": cls.room_type1.id,
"capacity": 3,
}
)
self.room1_3 = self.env["pms.room"].create(
cls.room1_3 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Triple 222",
"room_type_id": self.room_type1.id,
"room_type_id": cls.room_type1.id,
"capacity": 3,
}
)
self.host1 = self.env["res.partner"].create(
cls.host1 = cls.env["res.partner"].create(
{
"name": "Miguel",
"email": "miguel@example.com",
@@ -57,41 +58,41 @@ class TestPmsCheckinPartner(TestPms):
"gender": "male",
}
)
self.id_category = self.env["res.partner.id_category"].search(
cls.id_category = cls.env["res.partner.id_category"].search(
[("code", "=", "D")]
)
if not self.id_category:
self.id_category = self.env["res.partner.id_category"].create(
if not cls.id_category:
cls.id_category = cls.env["res.partner.id_category"].create(
{"name": "DNI", "code": "D"}
)
self.env["res.partner.id_number"].create(
cls.env["res.partner.id_number"].create(
{
"category_id": self.id_category.id,
"category_id": cls.id_category.id,
"name": "30065089H",
"valid_from": datetime.date.today(),
"partner_id": self.host1.id,
"valid_from": today,
"partner_id": cls.host1.id,
}
)
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",
}
)
reservation_vals = {
"checkin": datetime.date.today(),
"checkout": datetime.date.today() + datetime.timedelta(days=3),
"room_type_id": self.room_type1.id,
"partner_id": self.host1.id,
"checkin": today,
"checkout": today + datetime.timedelta(days=3),
"room_type_id": cls.room_type1.id,
"partner_id": cls.host1.id,
"adults": 3,
"pms_property_id": self.pms_property1.id,
"sale_channel_origin_id": self.sale_channel_direct1.id,
"pms_property_id": cls.pms_property1.id,
"sale_channel_origin_id": cls.sale_channel_direct1.id,
}
self.reservation_1 = self.env["pms.reservation"].create(reservation_vals)
self.checkin1 = self.env["pms.checkin.partner"].create(
cls.reservation_1 = cls.env["pms.reservation"].create(reservation_vals)
cls.checkin1 = cls.env["pms.checkin.partner"].create(
{
"partner_id": self.host1.id,
"reservation_id": self.reservation_1.id,
"partner_id": cls.host1.id,
"reservation_id": cls.reservation_1.id,
}
)

27
pms/tests/test_pms_folio.py
View File

@@ -12,43 +12,44 @@ class TestPmsFolio(TestPms):
# SetUp and Common Scenarios methods
def setUp(self):
@classmethod
def setUpClass(cls):
"""
- common + room_type_double with 2 rooms (double1 and double2) in pms_property1
"""
super().setUp()
super().setUpClass()
# create room type
self.room_type_double = self.env["pms.room.type"].create(
cls.room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"price": 25,
}
)
# create room
self.double1 = self.env["pms.room"].create(
cls.double1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 101",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
# create room
self.double2 = self.env["pms.room"].create(
cls.double2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 102",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
# make current journals payable
journals = self.env["account.journal"].search(
journals = cls.env["account.journal"].search(
[
("type", "in", ["bank", "cash"]),
]
@@ -56,7 +57,7 @@ class TestPmsFolio(TestPms):
journals.allowed_pms_payments = True
# create sale channel direct
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

49
pms/tests/test_pms_folio_invoice.py
View File

@@ -6,78 +6,79 @@ from .common import TestPms
class TestPmsFolioInvoice(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# create a room type availability
self.room_type_availability = self.env["pms.availability.plan"].create(
cls.room_type_availability = cls.env["pms.availability.plan"].create(
{"name": "Availability plan for TEST"}
)
# journal to simplified invoices
self.simplified_journal = self.env["account.journal"].create(
cls.simplified_journal = cls.env["account.journal"].create(
{
"name": "Simplified journal",
"code": "SMP",
"type": "sale",
"company_id": self.env.ref("base.main_company").id,
"company_id": cls.env.ref("base.main_company").id,
}
)
# create a property
self.property = self.env["pms.property"].create(
cls.property = cls.env["pms.property"].create(
{
"name": "MY PMS TEST",
"company_id": self.env.ref("base.main_company").id,
"default_pricelist_id": self.pricelist1.id,
"journal_simplified_invoice_id": self.simplified_journal.id,
"company_id": cls.env.ref("base.main_company").id,
"default_pricelist_id": cls.pricelist1.id,
"journal_simplified_invoice_id": cls.simplified_journal.id,
}
)
# create room type
self.room_type_double = self.env["pms.room.type"].create(
cls.room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.property.id],
"pms_property_ids": [cls.property.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"price": 25,
}
)
# create rooms
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.property.id,
"pms_property_id": cls.property.id,
"name": "Double 101",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
self.room2 = self.env["pms.room"].create(
cls.room2 = cls.env["pms.room"].create(
{
"pms_property_id": self.property.id,
"pms_property_id": cls.property.id,
"name": "Double 102",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
self.room3 = self.env["pms.room"].create(
cls.room3 = cls.env["pms.room"].create(
{
"pms_property_id": self.property.id,
"pms_property_id": cls.property.id,
"name": "Double 103",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
# res.partner
self.partner_id = self.env["res.partner"].create(
cls.partner_id = cls.env["res.partner"].create(
{
"name": "Miguel",
"vat": "45224522J",
"country_id": self.env.ref("base.es").id,
"country_id": cls.env.ref("base.es").id,
"city": "Madrid",
"zip": "28013",
"street": "Calle de la calle",
@@ -85,7 +86,7 @@ class TestPmsFolioInvoice(TestPms):
)
# create a sale channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

5
pms/tests/test_pms_folio_prices.py
View File

@@ -2,8 +2,9 @@ from odoo.tests.common import SavepointCase
class TestPmsFolioPrice(SavepointCase):
def setUp(self):
super(TestPmsFolioPrice, self).setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
def test_price_folio(self):
"""Test create reservation and services, and check price

49
pms/tests/test_pms_folio_sale_line.py
View File

@@ -6,54 +6,55 @@ from .common import TestPms
class TestPmsFolioSaleLine(TestPms):
def setUp(self):
@classmethod
def setUpClass(cls):
"""
- common + room_type_avalability_plan
"""
super().setUp()
super().setUpClass()
# create room type
self.room_type_double = self.env["pms.room.type"].create(
cls.room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"price": 25,
}
)
# create room
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 101",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
self.room2 = self.env["pms.room"].create(
cls.room2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 102",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
}
)
self.product_test1 = self.env["product.product"].create(
cls.product_test1 = cls.env["product.product"].create(
{
"name": "Test Product 1",
"per_day": True,
}
)
self.product_test2 = self.env["product.product"].create(
cls.product_test2 = cls.env["product.product"].create(
{
"name": "Test Product 1",
"per_day": True,
"consumed_on": "after",
}
)
self.board_service_test = self.board_service = self.env[
cls.board_service_test = cls.board_service = cls.env[
"pms.board.service"
].create(
{
@@ -61,28 +62,28 @@ class TestPmsFolioSaleLine(TestPms):
"default_code": "TPS",
}
)
self.env["pms.board.service.line"].create(
cls.env["pms.board.service.line"].create(
{
"pms_board_service_id": self.board_service_test.id,
"product_id": self.product_test1.id,
"pms_board_service_id": cls.board_service_test.id,
"product_id": cls.product_test1.id,
"amount": 8,
}
)
self.board_service_room_type = self.env["pms.board.service.room.type"].create(
cls.board_service_room_type = cls.env["pms.board.service.room.type"].create(
{
"pms_room_type_id": self.room_type_double.id,
"pms_board_service_id": self.board_service_test.id,
"pms_property_id": self.pms_property1.id,
"pms_room_type_id": cls.room_type_double.id,
"pms_board_service_id": cls.board_service_test.id,
"pms_property_id": cls.pms_property1.id,
}
)
self.extra_service = self.env["pms.service"].create(
cls.extra_service = cls.env["pms.service"].create(
{
"is_board_service": False,
"product_id": self.product_test2.id,
"product_id": cls.product_test2.id,
}
)
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

5
pms/tests/test_pms_invoice_refund.py
View File

@@ -6,5 +6,6 @@ freeze_time("2000-02-02")
class TestPmsInvoiceRefund(SavepointCase):
def setUp(self):
super(TestPmsInvoiceRefund, self).setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()

17
pms/tests/test_pms_multiproperty.py
View File

@@ -7,21 +7,22 @@ from .common import TestPms
class TestPmsMultiproperty(TestPms):
def setUp(self):
super().setUp()
self.pms_property2 = self.env["pms.property"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.pms_property2 = cls.env["pms.property"].create(
{
"name": "Pms_property_test2",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "Pms_property_test3",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)

5
pms/tests/test_pms_payment.py
View File

@@ -6,8 +6,9 @@ freeze_time("2000-02-02")
class TestPmsPayment(SavepointCase):
def setUp(self):
super(TestPmsPayment, self).setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# TODO: Test allowed manual payment
# create a journal with allowed_pms_payments = True and

55
pms/tests/test_pms_pricelist.py
View File

@@ -11,82 +11,83 @@ from .common import TestPms
@tagged("standard", "nice")
class TestPmsPricelist(TestPms):
def setUp(self):
super().setUp()
self.pms_property2 = self.env["pms.property"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.pms_property2 = cls.env["pms.property"].create(
{
"name": "Property_2",
"company_id": self.env.ref("base.main_company").id,
"default_pricelist_id": self.env.ref("product.list0").id,
"company_id": cls.env.ref("base.main_company").id,
"default_pricelist_id": cls.env.ref("product.list0").id,
}
)
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "Property_3",
"company_id": self.env.ref("base.main_company").id,
"default_pricelist_id": self.env.ref("product.list0").id,
"company_id": cls.env.ref("base.main_company").id,
"default_pricelist_id": cls.env.ref("product.list0").id,
}
)
self.room_type1 = self.env["pms.room.type"].create(
cls.room_type1 = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id, self.pms_property2.id],
"pms_property_ids": [cls.pms_property1.id, cls.pms_property2.id],
"name": "Single",
"default_code": "SIN",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"list_price": 30,
}
)
# pms.room
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Single 101",
"room_type_id": self.room_type1.id,
"room_type_id": cls.room_type1.id,
"capacity": 2,
}
)
self.pricelist2 = self.env["product.pricelist"].create(
cls.pricelist2 = cls.env["product.pricelist"].create(
{
"name": "pricelist_2",
"pms_property_ids": [self.pms_property1.id, self.pms_property2.id],
"availability_plan_id": self.availability_plan1.id,
"pms_property_ids": [cls.pms_property1.id, cls.pms_property2.id],
"availability_plan_id": cls.availability_plan1.id,
"is_pms_available": True,
}
)
# product.product 1
self.product1 = self.env["product.product"].create({"name": "Test Breakfast"})
cls.product1 = cls.env["product.product"].create({"name": "Test Breakfast"})
# pms.board.service
self.board_service1 = self.env["pms.board.service"].create(
cls.board_service1 = cls.env["pms.board.service"].create(
{
"name": "Test Only Breakfast",
"default_code": "CB1",
}
)
# pms.board.service.line
self.board_service_line1 = self.env["pms.board.service.line"].create(
cls.board_service_line1 = cls.env["pms.board.service.line"].create(
{
"product_id": self.product1.id,
"pms_board_service_id": self.board_service1.id,
"product_id": cls.product1.id,
"pms_board_service_id": cls.board_service1.id,
}
)
# pms.board.service.room.type
self.board_service_room_type1 = self.env["pms.board.service.room.type"].create(
cls.board_service_room_type1 = cls.env["pms.board.service.room.type"].create(
{
"pms_room_type_id": self.room_type1.id,
"pms_board_service_id": self.board_service1.id,
"pms_room_type_id": cls.room_type1.id,
"pms_board_service_id": cls.board_service1.id,
}
)
self.partner1 = self.env["res.partner"].create({"name": "Carles"})
cls.partner1 = cls.env["res.partner"].create({"name": "Carles"})
# create a sale channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

27
pms/tests/test_pms_res_users.py
View File

@@ -4,38 +4,39 @@ from .common import TestPms
class TestPmsResUser(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# create a company and properties
self.company_A = self.env["res.company"].create(
cls.company_A = cls.env["res.company"].create(
{
"name": "Pms_Company1",
}
)
self.company_B = self.env["res.company"].create(
cls.company_B = cls.env["res.company"].create(
{
"name": "Pms_Company2",
}
)
self.property_A1 = self.env["pms.property"].create(
cls.property_A1 = cls.env["pms.property"].create(
{
"name": "Pms_property",
"company_id": self.company_A.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company_A.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.property_A2 = self.env["pms.property"].create(
cls.property_A2 = cls.env["pms.property"].create(
{
"name": "Pms_property2",
"company_id": self.company_A.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company_A.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.property_B1 = self.env["pms.property"].create(
cls.property_B1 = cls.env["pms.property"].create(
{
"name": "Pms_propertyB1",
"company_id": self.company_B.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company_B.id,
"default_pricelist_id": cls.pricelist1.id,
}
)

57
pms/tests/test_pms_reservation.py
View File

@@ -9,76 +9,77 @@ from .common import TestPms
class TestPmsReservations(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# create a room type availability
self.room_type_availability = self.env["pms.availability.plan"].create(
cls.room_type_availability = cls.env["pms.availability.plan"].create(
{
"name": "Availability plan for TEST",
"pms_pricelist_ids": [(6, 0, [self.pricelist1.id])],
"pms_pricelist_ids": [(6, 0, [cls.pricelist1.id])],
}
)
# create room type
self.room_type_double = self.env["pms.room.type"].create(
cls.room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
self.room_type_triple = self.env["pms.room.type"].create(
cls.room_type_triple = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Triple Test",
"default_code": "TRP_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# create rooms
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 101",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
"extra_beds_allowed": 1,
}
)
self.room2 = self.env["pms.room"].create(
cls.room2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 102",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
"extra_beds_allowed": 1,
}
)
self.room3 = self.env["pms.room"].create(
cls.room3 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 103",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
"extra_beds_allowed": 1,
}
)
self.room4 = self.env["pms.room"].create(
cls.room4 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Triple 104",
"room_type_id": self.room_type_triple.id,
"room_type_id": cls.room_type_triple.id,
"capacity": 3,
"extra_beds_allowed": 1,
}
)
self.partner1 = self.env["res.partner"].create(
cls.partner1 = cls.env["res.partner"].create(
{
"firstname": "Jaime",
"lastname": "García",
@@ -87,22 +88,22 @@ class TestPmsReservations(TestPms):
"gender": "male",
}
)
self.id_category = self.env["res.partner.id_category"].create(
cls.id_category = cls.env["res.partner.id_category"].create(
{"name": "DNI", "code": "D"}
)
self.sale_channel_direct = self.env["pms.sale.channel"].create(
cls.sale_channel_direct = cls.env["pms.sale.channel"].create(
{"name": "sale channel direct", "channel_type": "direct"}
)
self.sale_channel1 = self.env["pms.sale.channel"].create(
cls.sale_channel1 = cls.env["pms.sale.channel"].create(
{"name": "saleChannel1", "channel_type": "indirect"}
)
self.agency1 = self.env["res.partner"].create(
cls.agency1 = cls.env["res.partner"].create(
{
"name": "partner1",
"is_agency": True,
"invoice_to_agency": "always",
"default_commission": 15,
"sale_channel_id": self.sale_channel1.id,
"sale_channel_id": cls.sale_channel1.id,
}
)

17
pms/tests/test_pms_room.py
View File

@@ -7,22 +7,23 @@ from .common import TestPms
class TestPmsRoom(TestPms):
def setUp(self):
super().setUp()
self.pms_property2 = self.env["pms.property"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.pms_property2 = cls.env["pms.property"].create(
{
"name": "Property_2",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.room_type1 = self.env["pms.room.type"].create(
cls.room_type1 = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id, self.pms_property2.id],
"pms_property_ids": [cls.pms_property1.id, cls.pms_property2.id],
"name": "Single",
"default_code": "SIN",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
"list_price": 30,
}
)

19
pms/tests/test_pms_room_type.py
View File

@@ -7,26 +7,27 @@ from .common import TestPms
class TestRoomType(TestPms):
def setUp(self):
super().setUp()
self.pms_property2 = self.env["pms.property"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.pms_property2 = cls.env["pms.property"].create(
{
"name": "Property 2",
"company_id": self.company1.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company1.id,
"default_pricelist_id": cls.pricelist1.id,
}
)
self.company2 = self.env["res.company"].create(
cls.company2 = cls.env["res.company"].create(
{
"name": "Company 2",
}
)
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "Property 3",
"company_id": self.company2.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company2.id,
"default_pricelist_id": cls.pricelist1.id,
}
)

13
pms/tests/test_pms_room_type_class.py
View File

@@ -7,18 +7,19 @@ from .common import TestPms
class TestRoomTypeClass(TestPms):
def setUp(self):
super().setUp()
self.company2 = self.env["res.company"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.company2 = cls.env["res.company"].create(
{
"name": "Company 2",
}
)
self.pms_property3 = self.env["pms.property"].create(
cls.pms_property3 = cls.env["pms.property"].create(
{
"name": "Property 3",
"company_id": self.company2.id,
"default_pricelist_id": self.pricelist1.id,
"company_id": cls.company2.id,
"default_pricelist_id": cls.pricelist1.id,
}
)

31
pms/tests/test_pms_service.py
View File

@@ -8,38 +8,39 @@ from .common import TestPms
class TestPmsService(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# create room type
self.room_type_double = self.env["pms.room.type"].create(
cls.room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# create rooms
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 101",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
"extra_beds_allowed": 1,
}
)
self.room2 = self.env["pms.room"].create(
cls.room2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 102",
"room_type_id": self.room_type_double.id,
"room_type_id": cls.room_type_double.id,
"capacity": 2,
"extra_beds_allowed": 1,
}
)
self.partner1 = self.env["res.partner"].create(
cls.partner1 = cls.env["res.partner"].create(
{
"firstname": "María",
"lastname": "",
@@ -48,13 +49,13 @@ class TestPmsService(TestPms):
"gender": "male",
}
)
self.sale_channel_door = self.env["pms.sale.channel"].create(
cls.sale_channel_door = cls.env["pms.sale.channel"].create(
{"name": "Door", "channel_type": "direct"}
)
self.sale_channel_phone = self.env["pms.sale.channel"].create(
cls.sale_channel_phone = cls.env["pms.sale.channel"].create(
{"name": "Phone", "channel_type": "direct"}
)
self.sale_channel_mail = self.env["pms.sale.channel"].create(
cls.sale_channel_mail = cls.env["pms.sale.channel"].create(
{"name": "Mail", "channel_type": "direct"}
)

5
pms/tests/test_pms_simple_invoice.py
View File

@@ -6,5 +6,6 @@ freeze_time("2000-02-02")
class TestPmsInvoiceSimpleInvoice(SavepointCase):
def setUp(self):
super(TestPmsInvoiceSimpleInvoice, self).setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()

9
pms/tests/test_pms_wizard_massive_changes.py
View File

@@ -8,12 +8,13 @@ from .common import TestPms
class TestPmsWizardMassiveChanges(TestPms):
def setUp(self):
super().setUp()
self.availability_plan1 = self.env["pms.availability.plan"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.availability_plan1 = cls.env["pms.availability.plan"].create(
{
"name": "Availability plan for TEST",
"pms_pricelist_ids": [(6, 0, [self.pricelist1.id])],
"pms_pricelist_ids": [(6, 0, [cls.pricelist1.id])],
}
)

37
pms/tests/test_pms_wizard_split_join_swap_reservation.py
View File

@@ -6,58 +6,59 @@ from .common import TestPms
class TestPmsWizardSplitJoinSwapReservation(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# pms.availability.plan
self.test_availability_plan = self.env["pms.availability.plan"].create(
cls.test_availability_plan = cls.env["pms.availability.plan"].create(
{
"name": "Availability plan for TEST",
"pms_pricelist_ids": [(6, 0, [self.pricelist1.id])],
"pms_pricelist_ids": [(6, 0, [cls.pricelist1.id])],
}
)
# pms.room.type
self.test_room_type_single = self.env["pms.room.type"].create(
cls.test_room_type_single = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Single Test",
"default_code": "SNG_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# pms.room.type
self.test_room_type_double = self.env["pms.room.type"].create(
cls.test_room_type_double = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Double Test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
# create rooms
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 101",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
self.room2 = self.env["pms.room"].create(
cls.room2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Double 102",
"room_type_id": self.test_room_type_double.id,
"room_type_id": cls.test_room_type_double.id,
"capacity": 2,
}
)
self.partner1 = self.env["res.partner"].create({"name": "Antón"})
cls.partner1 = cls.env["res.partner"].create({"name": "Antón"})
# create a sale channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

21
pms/tests/test_product_template.py
View File

@@ -7,32 +7,33 @@ from .common import TestPms
class TestProductTemplate(TestPms):
def setUp(self):
super().setUp()
self.room_type = self.env["pms.room.type"].create(
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.room_type = cls.env["pms.room.type"].create(
{
"name": "Room type test",
"default_code": "DBL_Test",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
self.room = self.env["pms.room"].create(
cls.room = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Room test",
"room_type_id": self.room_type.id,
"room_type_id": cls.room_type.id,
"capacity": 2,
}
)
self.partner = self.env["res.partner"].create({"name": "partner1"})
self.board_service = self.env["pms.board.service"].create(
cls.partner = cls.env["res.partner"].create({"name": "partner1"})
cls.board_service = cls.env["pms.board.service"].create(
{
"name": "Board service test",
"default_code": "BST",
}
)
# create a sale channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",

49
pms/tests/test_shared_room.py
View File

@@ -7,17 +7,18 @@ from .common import TestPms
class TestPmsSharedRoom(TestPms):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(cls):
super().setUpClass()
# create a room type availability
self.room_type_availability = self.env["pms.availability.plan"].create(
cls.room_type_availability = cls.env["pms.availability.plan"].create(
{
"name": "Availability plan for TEST",
"pms_pricelist_ids": [(6, 0, [self.pricelist1.id])],
"pms_pricelist_ids": [(6, 0, [cls.pricelist1.id])],
}
)
self.bed_class = self.env["pms.room.type.class"].create(
cls.bed_class = cls.env["pms.room.type.class"].create(
{
"name": "Bed Class 1",
"default_code": "B1",
@@ -25,57 +26,57 @@ class TestPmsSharedRoom(TestPms):
)
# create room type
self.room_type_test = self.env["pms.room.type"].create(
cls.room_type_test = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Shared Test",
"default_code": "SHT",
"class_id": self.room_type_class1.id,
"class_id": cls.room_type_class1.id,
}
)
self.room_type_bed = self.env["pms.room.type"].create(
cls.room_type_bed = cls.env["pms.room.type"].create(
{
"pms_property_ids": [self.pms_property1.id],
"pms_property_ids": [cls.pms_property1.id],
"name": "Bed Type Test",
"default_code": "BTT",
"class_id": self.bed_class.id,
"class_id": cls.bed_class.id,
}
)
# create shared room
self.room1 = self.env["pms.room"].create(
cls.room1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "Shared 101",
"room_type_id": self.room_type_test.id,
"room_type_id": cls.room_type_test.id,
"capacity": 2,
}
)
# create beds in room1
self.r1bed1 = self.env["pms.room"].create(
cls.r1bed1 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "101 (1)",
"room_type_id": self.room_type_bed.id,
"room_type_id": cls.room_type_bed.id,
"capacity": 1,
"parent_id": self.room1.id,
"parent_id": cls.room1.id,
}
)
self.r1bed2 = self.env["pms.room"].create(
cls.r1bed2 = cls.env["pms.room"].create(
{
"pms_property_id": self.pms_property1.id,
"pms_property_id": cls.pms_property1.id,
"name": "101 (2)",
"room_type_id": self.room_type_bed.id,
"room_type_id": cls.room_type_bed.id,
"capacity": 2,
"parent_id": self.room1.id,
"parent_id": cls.room1.id,
}
)
# create partner
self.partner1 = self.env["res.partner"].create(
cls.partner1 = cls.env["res.partner"].create(
{
"firstname": "Jaime",
"lastname": "García",
@@ -86,7 +87,7 @@ class TestPmsSharedRoom(TestPms):
)
# create a sale channel
self.sale_channel_direct1 = self.env["pms.sale.channel"].create(
cls.sale_channel_direct1 = cls.env["pms.sale.channel"].create(
{
"name": "Door",
"channel_type": "direct",