Adapt test module

pull/1798/head
Akim Juillerat 2020-04-01 11:16:57 +02:00
parent 9fcaf2db9e
commit 09206fb73e
13 changed files with 165 additions and 153 deletions

View File

@ -1,2 +1,2 @@
from . import base_time_weekday from . import time_weekday
from . import base_time_window from . import time_window_mixin

View File

@ -1 +1,2 @@
This module provides base classes to manage time windows. This module provides base classes and models to manage time windows through
`time.window.mixin`.

View File

@ -0,0 +1,30 @@
Example implementation for the mixin can be found in module `test_base_time_window`.
As a time window will always be linked to a related model thourgh a M2o relation,
when defining the new model inheriting the mixin, one should pay attention to the
following points in order to have the overlapping check work properly:
- Define class property `_overlap_check_field`: This must state the M2o field to
use for the to check of overlapping time window records linked to a specific
record of the related model.
- Add the M2o field to the related model in the `api.constrains`:
For example:
.. code-block:: python
class PartnerTimeWindow(models.Model):
_name = 'partner.time.window'
_inherit = 'time.window.mixin'
partner_id = fields.Many2one(
res.partner', required=True, index=True, ondelete='cascade'
)
_overlap_check_field = 'partner_id'
@api.constrains('partner_id')
def check_window_no_overlaps(self):
return super().check_window_no_overlaps()

View File

@ -1,2 +1 @@
# Copyright 2020 Camptocamp SA from . import models
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)

View File

@ -8,7 +8,7 @@
"author": "ACSONE SA/NV, Camptocamp, Odoo Community Association (OCA)", "author": "ACSONE SA/NV, Camptocamp, Odoo Community Association (OCA)",
"license": "AGPL-3", "license": "AGPL-3",
"website": "https://github.com/OCA/server-tools", "website": "https://github.com/OCA/server-tools",
"depends": ["base"], "depends": ["base_time_window"],
"data": [], "data": ["security/ir.model.access.xml"],
"installable": True, "installable": True,
} }

View File

@ -1,2 +1,2 @@
# Copyright 2020 Camptocamp SA from . import partner_time_window
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl) from . import res_partner

View File

@ -4,82 +4,30 @@
import math import math
from psycopg2.extensions import AsIs
from odoo import _, api, fields, models from odoo import _, api, fields, models
from odoo.exceptions import ValidationError
class DeliveryWindow(models.Model): class TestPartnerTimeWindow(models.Model):
_name = "delivery.window" _name = "test.partner.time.window"
_description = "Delivery Window" _inherit = "time.window.mixin"
_order = "partner_id, start" _description = "Test partner time Window"
_overlap_check_field = 'partner_id'
start = fields.Float("From", required=True)
end = fields.Float("To", required=True)
week_day_ids = fields.Many2many(
comodel_name="alc.delivery.week.day", required=True
)
partner_id = fields.Many2one( partner_id = fields.Many2one(
"res.partner", required=True, index=True, ondelete='cascade' "res.partner", required=True, index=True, ondelete='cascade'
) )
@api.constrains("start", "end", "week_day_ids") @api.constrains("partner_id")
def check_window_no_onverlaps(self): def check_window_no_overlaps(self):
week_days_field = self._fields["week_day_ids"] return super().check_window_no_overlaps()
for record in self:
if record.start > record.end:
raise ValidationError(
_("%s must be > %s")
% (
self.float_to_time_repr(record.end),
self.float_to_time_repr(record.start),
)
)
# here we use a plain SQL query to benefit of the numrange
# function available in PostgresSQL
# (http://www.postgresql.org/docs/current/static/rangetypes.html)
SQL = """
SELECT
id
FROM
%(table)s w
join %(relation)s as d
on d.%(relation_window_fkey)s = w.id
WHERE
NUMRANGE(w.start::numeric, w.end::numeric) &&
NUMRANGE(%(start)s::numeric, %(end)s::numeric)
AND w.id != %(window_id)s
AND d.%(relation_week_day_fkey)s in %(week_day_ids)s
AND w.partner_id = %(partner_id)s"""
self.env.cr.execute(
SQL,
dict(
table=AsIs(self._table),
relation=AsIs(week_days_field.relation),
relation_window_fkey=AsIs(week_days_field.column1),
relation_week_day_fkey=AsIs(week_days_field.column2),
start=record.start,
end=record.end,
window_id=record.id,
week_day_ids=tuple(record.week_day_ids.ids),
partner_id=record.partner_id.id,
),
)
res = self.env.cr.fetchall()
if res:
other = self.browse(res[0][0])
raise ValidationError(
_("%s overlaps %s")
% (record.display_name, other.display_name)
)
@api.depends("start", "end", "week_day_ids") @api.depends("start", "end", "weekday_ids")
def _compute_display_name(self): def _compute_display_name(self):
for record in self: for record in self:
"{days}: From {start} to {end}".format( record.display_name = _("{days}: From {start} to {end}").format(
days=", ".join(record.week_day_ids.mapped("display_name")), days=", ".join(record.weekday_ids.mapped("display_name")),
start=self.float_to_time_repr(record.start), start=self.float_to_time_repr(record.start),
end=self.float_to_time_repr(record.end), end=self.float_to_time_repr(record.end),
) )

View File

@ -11,12 +11,10 @@ class ResPartner(models.Model):
_inherit = "res.partner" _inherit = "res.partner"
delivery_window_ids = fields.One2many( time_window_ids = fields.One2many(
comodel_name="delivery.window", comodel_name="test.partner.time.window",
inverse_name="partner_id", inverse_name="partner_id",
string="Delivery windows", string="Time windows",
help="If specified, delivery is only possible into the specified "
"time windows. (Leaves empty if no restriction)",
) )
def get_delivery_windows(self, day_name): def get_delivery_windows(self, day_name):

View File

@ -0,0 +1,2 @@
* Laurent Mignon <laurent.mignon@acsone.eu>
* Akim Juillerat <akim.juillerat@camptocamp.com>

View File

@ -0,0 +1 @@
This module provides unittests for module `base_time_window`.

View File

@ -3,9 +3,9 @@
License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). --> License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). -->
<odoo> <odoo>
<record model="ir.model.access" id="delivery_window_access_read"> <record model="ir.model.access" id="test_partner_time_window_access_read">
<field name="name">delivery.window access read</field> <field name="name">test.partner.time.window access read</field>
<field name="model_id" ref="model_delivery_window"/> <field name="model_id" ref="model_test_partner_time_window"/>
<field name="group_id" ref="base.group_user"/> <field name="group_id" ref="base.group_user"/>
<field name="perm_read" eval="1"/> <field name="perm_read" eval="1"/>
<field name="perm_create" eval="0"/> <field name="perm_create" eval="0"/>
@ -13,10 +13,10 @@
<field name="perm_unlink" eval="0"/> <field name="perm_unlink" eval="0"/>
</record> </record>
<record model="ir.model.access" id="delivery_window_access_manage"> <record model="ir.model.access" id="test_partner_time_window_access_manage">
<field name="name">delivery.window access read</field> <field name="name">test.partner.time.window access read</field>
<field name="model_id" ref="model_delivery_window"/> <field name="model_id" ref="model_test_partner_time_window"/>
<field name="group_id" ref="sales_team.group_sale_manager"/> <field name="group_id" ref="base.group_partner_manager"/>
<field name="perm_read" eval="1"/> <field name="perm_read" eval="1"/>
<field name="perm_create" eval="1"/> <field name="perm_create" eval="1"/>
<field name="perm_write" eval="1"/> <field name="perm_write" eval="1"/>

View File

@ -1,2 +1 @@
# Copyright 2020 Camptocamp SA from . import test_base_time_window
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)

View File

@ -6,182 +6,216 @@ from odoo.exceptions import ValidationError
from odoo.tests import SavepointCase from odoo.tests import SavepointCase
class TestDeliveryWindow(SavepointCase): class TestTimeWindow(SavepointCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
super(TestDeliveryWindow, cls).setUpClass() super().setUpClass()
cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))
cls.partner_1 = cls.env['res.partner'].create({'name': 'partner 1'}) cls.partner_1 = cls.env['res.partner'].create({'name': 'partner 1'})
cls.partner_2 = cls.env['res.partner'].create({'name': 'patner 2'}) cls.partner_2 = cls.env['res.partner'].create({'name': 'partner 2'})
cls.DeliveryWindow = cls.env["delivery.window"] cls.TimeWindow = cls.env["test.partner.time.window"]
cls.monday = cls.env.ref( cls.monday = cls.env.ref(
"partner_delivery_window.delivery_weed_day_monday" "base_time_window.time_weekday_monday"
) )
cls.sunday = cls.env.ref( cls.sunday = cls.env.ref(
"partner_delivery_window.delivery_weed_day_sunday" "base_time_window.time_weekday_sunday"
) )
def test_00(self): def test_00(self):
""" """
Data: Data:
A partner without delivery window A partner without time window
Test Case: Test Case:
Add a delivery window Add a time window
Expected result: Expected result:
A delivery window is created for the partner A time window is created for the partner
""" """
self.assertFalse(self.partner_1.delivery_window_ids) self.assertFalse(self.partner_1.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
self.assertTrue(self.partner_1.delivery_window_ids) self.assertTrue(self.partner_1.time_window_ids)
delivery_window = self.partner_1.delivery_window_ids time_window = self.partner_1.time_window_ids
self.assertEqual(delivery_window.start, 10.0) self.assertEqual(time_window.start, 10.0)
self.assertEqual(delivery_window.end, 12.0) self.assertEqual(time_window.end, 12.0)
self.assertEqual(delivery_window.week_day_ids, self.monday) self.assertEqual(time_window.weekday_ids, self.monday)
def test_01(self): def test_01(self):
""" """
Data: Data:
A partner without delivery window A partner without time window
Test Case: Test Case:
1 Add a delivery window 1 Add a time window
2 unlink the partner 2 unlink the partner
Expected result: Expected result:
1 A delivery window is created for the partner 1 A time window is created for the partner
2 The delivery window is removed 2 The time window is removed
""" """
partner_id = self.partner_1.id partner_id = self.partner_1.id
self.assertFalse(self.partner_1.delivery_window_ids) self.assertFalse(self.partner_1.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
self.assertTrue(self.partner_1.delivery_window_ids) self.assertTrue(self.partner_1.time_window_ids)
delivery_window = self.DeliveryWindow.search( time_window = self.TimeWindow.search(
[("partner_id", "=", partner_id)] [("partner_id", "=", partner_id)]
) )
self.assertTrue(delivery_window) self.assertTrue(time_window)
self.partner_1.unlink() self.partner_1.unlink()
self.assertFalse(delivery_window.exists()) self.assertFalse(time_window.exists())
def test_02(self): def test_02(self):
""" """
Data: Data:
A partner without delivery window A partner without time window
Test Case: Test Case:
1 Add a delivery window 1 Add a time window
2 Add a second delivery window that overlaps the first one (same day) 2 Add a second time window that overlaps the first one (same day)
Expected result: Expected result:
1 A delivery window is created for the partner 1 A time window is created for the partner
2 ValidationError is raised 2 ValidationError is raised
""" """
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
with self.assertRaises(ValidationError): with self.assertRaises(ValidationError):
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 11.0, "start": 11.0,
"end": 13.0, "end": 13.0,
"week_day_ids": [(4, self.monday.id), (4, self.sunday.id)], "weekday_ids": [(4, self.monday.id), (4, self.sunday.id)],
} }
) )
def test_03(self): def test_03(self):
""" """
Data: Data:
A partner without delivery window A partner without time window
Test Case: Test Case:
1 Add a delivery window 1 Add a time window
2 Add a second delivery window that overlaps the first one (another day) 2 Add a second time window that overlaps the first one (another day)
Expected result: Expected result:
1 A delivery window is created for the partner 1 A time window is created for the partner
2 A second delivery window is created for the partner 2 A second time window is created for the partner
""" """
self.assertFalse(self.partner_1.delivery_window_ids) self.assertFalse(self.partner_1.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
self.assertTrue(self.partner_1.delivery_window_ids) self.assertTrue(self.partner_1.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 11.0, "start": 11.0,
"end": 13.0, "end": 13.0,
"week_day_ids": [(4, self.sunday.id)], "weekday_ids": [(4, self.sunday.id)],
} }
) )
self.assertEquals(len(self.partner_1.delivery_window_ids), 2) self.assertEquals(len(self.partner_1.time_window_ids), 2)
def test_04(self): def test_04(self):
""" """
Data: Data:
Partner 1 without delivery window Partner 1 without time window
Partner 2 without delivery window Partner 2 without time window
Test Case: Test Case:
1 Add a delivery window to partner 1 1 Add a time window to partner 1
2 Add the same delivery window to partner 2 2 Add the same time window to partner 2
Expected result: Expected result:
1 A delivery window is created for the partner 1 1 A time window is created for the partner 1
1 A delivery window is created for the partner 2 1 A time window is created for the partner 2
""" """
self.assertFalse(self.partner_1.delivery_window_ids) self.assertFalse(self.partner_1.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
self.assertTrue(self.partner_1.delivery_window_ids) self.assertTrue(self.partner_1.time_window_ids)
self.assertFalse(self.partner_2.delivery_window_ids) self.assertFalse(self.partner_2.time_window_ids)
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_2.id, "partner_id": self.partner_2.id,
"start": 10.0, "start": 10.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
self.assertTrue(self.partner_2.delivery_window_ids) self.assertTrue(self.partner_2.time_window_ids)
def test_05(self): def test_05(self):
""""" """""
Data: Data:
Partner 1 without delivery window Partner 1 without time window
Test Case: Test Case:
Add a delivery window to partner 1 with end > start Add a time window to partner 1 with end > start
Expected result: Expected result:
ValidationError is raised ValidationError is raised
""" """
with self.assertRaises(ValidationError): with self.assertRaises(ValidationError):
self.DeliveryWindow.create( self.TimeWindow.create(
{ {
"partner_id": self.partner_1.id, "partner_id": self.partner_1.id,
"start": 14.0, "start": 14.0,
"end": 12.0, "end": 12.0,
"week_day_ids": [(4, self.monday.id)], "weekday_ids": [(4, self.monday.id)],
} }
) )
def test_06(self):
"""""
Data:
Partner 1 with time window on monday
Partner 2 with time window on monday
Test Case:
Change time window from Partner 1 to Partner 2
Expected result:
ValidationError is raised
"""
self.assertFalse(self.partner_1.time_window_ids)
p1_timewindow = self.TimeWindow.create(
{
"partner_id": self.partner_1.id,
"start": 10.0,
"end": 12.0,
"weekday_ids": [(4, self.monday.id)],
}
)
self.assertTrue(self.partner_1.time_window_ids)
self.assertFalse(self.partner_2.time_window_ids)
self.TimeWindow.create(
{
"partner_id": self.partner_2.id,
"start": 10.0,
"end": 12.0,
"weekday_ids": [(4, self.monday.id)],
}
)
self.assertTrue(self.partner_2.time_window_ids)
with self.assertRaises(ValidationError):
p1_timewindow.partner_id = self.partner_2