# Copyright 2016 ACSONE SA/NV (<https://acsone.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)

from flectra.tests.common import TransactionCase
from flectra.exceptions import ValidationError
from dateutil.rrule import MONTHLY


class DateRangeGeneratorTest(TransactionCase):
    post_install = True
    at_install = False

    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': 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, '1943-10-01')
        self.assertEqual(range4.date_end, '1943-12-31')
        self.assertEqual(range4.type_id, self.type)

    def test_generator_multicompany_1(self):
        generator = self.generator.new({
            'date_start': '1943-01-01',
            'name_prefix': '1943-',
            'type_id': self.typeB.id,
            'duration_count': 3,
            'unit_of_time': MONTHLY,
            'count': 4,
            'company_id': self.company.id,
        })
        generator._cache.update(generator._convert_to_cache(
            {'company_id': self.company_2.id}, update=True))
        generator._onchange_company_id()
        self.assertFalse(generator.type_id)

    def test_generator_multicompany_2(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': MONTHLY,
                'count': 4,
                'company_id': self.company_2.id,
            })

    def test_partner_id_domain(self):
        """Check here domain returned for partner_id
        in both date.range and date.range.generator"""
        date_range = self.env['date.range']
        generator = self.env['date.range.generator']
        date_type = self.env['date.range.type']
        month_type = date_type.create({
            'name': 'month type'
        })
        day_type = date_type.create({
            'name': 'day type',
            'parent_type_id': month_type.id,
        })
        month_range = date_range.create({
            'name': 'month range',
            'type_id': month_type.id,
            'date_start': '01-01-2050',
            'date_end': '02-01-2050',
        })
        # now trigger onchange in generator,
        # which would also trigger onchange in date_range
        values = {
            'date_start': month_range.date_start,
            'type_id': day_type.id,
        }
        on_change = generator._onchange_spec()
        domain = generator.onchange(
            values,
            ['type_id', 'date_start'],
            on_change,
        )
        # check that with this search domain,
        # only the month_range record is returned.
        self.assertEqual(
            date_range.search(domain['domain']['parent_id']),
            month_range,
        )
        # the same applies to date.range treeview
        values = {
            'date_start': month_range.date_start,
            'type_id': day_type.id,
        }
        on_change = date_range._onchange_spec()
        domain = date_range.onchange(
            values,
            ['type_id', 'date_start'],
            on_change,
        )
        self.assertEqual(
            date_range.search(domain['domain']['parent_id']),
            month_range,
        )