You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
2.6 KiB
63 lines
2.6 KiB
# -*- coding: utf-8 -*-
|
|
# © 2016 Therp BV <http://therp.nl>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
from psycopg2.extensions import AsIs
|
|
from dateutil.rrule import MONTHLY, rrule
|
|
from openerp import SUPERUSER_ID, fields, models
|
|
from openerp.tests.common import TransactionCase
|
|
from openerp.addons.field_rrule import FieldRRule
|
|
from openerp.addons.field_rrule.field_rrule import SerializableRRuleSet
|
|
|
|
|
|
class RRuleTest(models.TransientModel):
|
|
_name = 'test.field.rrule'
|
|
|
|
# either use a default in object notation
|
|
rrule_with_default = FieldRRule(default=[{
|
|
"type": "rrule",
|
|
"dtstart": '2016-01-02 00:00:00',
|
|
"count": 1,
|
|
"freq": MONTHLY,
|
|
"interval": 1,
|
|
"bymonthday": [1],
|
|
}])
|
|
# or pass a SerializableRRuleSet.
|
|
# Rember that this class is callable, so passing it directly as default
|
|
# would actually pass an rruleset, because odoo uses the result of the
|
|
# callable. But in __call__, we check for this case, so nothing to do
|
|
rrule_with_default2 = FieldRRule(default=SerializableRRuleSet(
|
|
rrule(
|
|
dtstart=fields.Datetime.from_string('2016-01-02 00:00:00'),
|
|
interval=1,
|
|
freq=MONTHLY,
|
|
count=1,
|
|
bymonthday=[1],
|
|
)))
|
|
# also fiddle with an empty one
|
|
rrule = FieldRRule()
|
|
|
|
|
|
class TestFieldRrule(TransactionCase):
|
|
def test_field_rrule(self):
|
|
model = RRuleTest._build_model(self.registry, self.cr)
|
|
model._prepare_setup(self.cr, SUPERUSER_ID, False)
|
|
model._setup_base(self.cr, SUPERUSER_ID, False)
|
|
model._setup_fields(self.cr, SUPERUSER_ID)
|
|
model._auto_init(self.cr)
|
|
record_id = model.create(self.cr, SUPERUSER_ID, {'rrule': None})
|
|
self.cr.execute(
|
|
'select rrule, rrule_with_default, rrule_with_default2 from '
|
|
'%s where id=%s', (AsIs(model._table), record_id))
|
|
data = self.cr.fetchall()[0]
|
|
self.assertEqual(data[0], 'null')
|
|
self.assertEqual(data[1], data[2])
|
|
record = model.browse(self.cr, SUPERUSER_ID, record_id)
|
|
self.assertFalse(record.rrule)
|
|
self.assertTrue(record.rrule_with_default)
|
|
self.assertTrue(record.rrule_with_default2)
|
|
self.assertEqual(record.rrule_with_default, record.rrule_with_default2)
|
|
self.assertEqual(record.rrule_with_default.count(), 1)
|
|
self.assertFalse(record.rrule_with_default.after(
|
|
fields.Datetime.from_string('2016-02-01 00:00:00')))
|
|
self.assertTrue(record.rrule_with_default.after(
|
|
fields.Datetime.from_string('2016-02-01 00:00:00'), inc=True))
|