mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-11-22 06:02:08 +00:00
101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
# Copyright 2016 ACSONE SA/NV (<http://acsone.eu>)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)nses/agpl).
|
|
|
|
import datetime
|
|
|
|
from dateutil.rrule import MONTHLY
|
|
|
|
from flectra.exceptions import ValidationError
|
|
from flectra.tests.common import Form, TransactionCase
|
|
|
|
|
|
class DateRangeGeneratorTest(TransactionCase):
|
|
def setUp(self):
|
|
super(DateRangeGeneratorTest, self).setUp()
|
|
self.generator = self.env["date.range.generator"]
|
|
self.type = self.env["date.range.type"].create(
|
|
{"name": "Fiscal year", "company_id": False, "allow_overlap": False}
|
|
)
|
|
|
|
self.company = self.env["res.company"].create({"name": "Test company"})
|
|
self.company_2 = self.env["res.company"].create(
|
|
{"name": "Test company 2", "parent_id": self.company.id}
|
|
)
|
|
self.typeB = self.env["date.range.type"].create(
|
|
{
|
|
"name": "Fiscal year B",
|
|
"company_id": self.company.id,
|
|
"allow_overlap": False,
|
|
}
|
|
)
|
|
|
|
def test_generate(self):
|
|
generator = self.generator.create(
|
|
{
|
|
"date_start": "1943-01-01",
|
|
"name_prefix": "1943-",
|
|
"type_id": self.type.id,
|
|
"duration_count": 3,
|
|
"unit_of_time": str(MONTHLY),
|
|
"count": 4,
|
|
}
|
|
)
|
|
generator.action_apply()
|
|
ranges = self.env["date.range"].search([("type_id", "=", self.type.id)])
|
|
self.assertEqual(len(ranges), 4)
|
|
range4 = ranges[3]
|
|
self.assertEqual(range4.date_start, datetime.date(1943, 10, 1))
|
|
self.assertEqual(range4.date_end, datetime.date(1943, 12, 31))
|
|
self.assertEqual(range4.type_id, self.type)
|
|
|
|
def test_generator_multicompany_1(self):
|
|
with self.assertRaises(ValidationError):
|
|
self.generator.create(
|
|
{
|
|
"date_start": "1943-01-01",
|
|
"name_prefix": "1943-",
|
|
"type_id": self.typeB.id,
|
|
"duration_count": 3,
|
|
"unit_of_time": str(MONTHLY),
|
|
"count": 4,
|
|
"company_id": self.company_2.id,
|
|
}
|
|
)
|
|
|
|
def test_generator_form(self):
|
|
"""Test validation and onchange functionality"""
|
|
form = Form(self.env["date.range.generator"])
|
|
form.type_id = self.type
|
|
form.unit_of_time = str(MONTHLY)
|
|
form.duration_count = 10
|
|
form.date_end = "2021-01-01"
|
|
# Setting count clears date_end
|
|
form.count = 10
|
|
self.assertFalse(form.date_end)
|
|
# Setting date_end clears count
|
|
form.date_end = "2021-01-01"
|
|
self.assertFalse(form.count)
|
|
form.count = 10
|
|
form.name_prefix = "PREFIX"
|
|
# Cannot generate ranges with an invalid name_expr
|
|
with self.assertRaisesRegex(ValidationError, "Invalid name expression"):
|
|
form.name_expr = "'not valid"
|
|
# Setting name_expr clears name_prefix
|
|
form.name_expr = "'PREFIX%s' % index"
|
|
self.assertFalse(form.name_prefix)
|
|
self.assertEqual(form.range_name_preview, "PREFIX01")
|
|
wizard = form.save()
|
|
|
|
# Cannot generate ranges without count and without date_end
|
|
wizard.date_end = False
|
|
wizard.count = False
|
|
with self.assertRaisesRegex(ValidationError, "end date.*number of ranges"):
|
|
wizard.action_apply()
|
|
|
|
wizard.count = 10
|
|
# Cannot generate ranges without a prefix and without an expression
|
|
wizard.name_expr = False
|
|
wizard.name_prefix = False
|
|
with self.assertRaisesRegex(ValidationError, "prefix.*expression"):
|
|
wizard.action_apply()
|