# Copyright 2017 Creu Blanca <https://creublanca.es/>
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).


from datetime import date

from flectra.tests import common


class TestSequence(common.TransactionCase):
    def setUp(self):
        super().setUp()
        self.date = date(2018, 3, 14)

    def get_sequence(self, method):
        return self.env["ir.sequence"].create(
            {
                "name": "Test sequence",
                "implementation": "standard",
                "use_date_range": True,
                "range_reset": method,
                "padding": "5",
            }
        )

    def test_none(self):
        sequence = self.get_sequence(False)
        self.assertFalse(sequence.date_range_ids)
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )
        xrange = sequence.date_range_ids
        self.assertTrue(xrange)
        self.assertEqual(date(2018, 1, 1), xrange.date_from)
        self.assertEqual(date(2018, 12, 31), xrange.date_to)

    def test_daily(self):
        sequence = self.get_sequence("daily")
        self.assertFalse(sequence.date_range_ids)
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )
        xrange = sequence.date_range_ids
        self.assertTrue(xrange)
        self.assertEqual(self.date, xrange.date_from)
        self.assertEqual(self.date, xrange.date_to)

    def test_weekly(self):
        sequence = self.get_sequence("weekly")
        self.assertFalse(sequence.date_range_ids)
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )
        xrange = sequence.date_range_ids
        self.assertTrue(xrange)
        self.assertEqual(date(2018, 3, 12), xrange.date_from)
        self.assertEqual(date(2018, 3, 18), xrange.date_to)

    def test_monthly(self):
        sequence = self.get_sequence("monthly")
        self.assertFalse(sequence.date_range_ids)
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )
        xrange = sequence.date_range_ids
        self.assertTrue(xrange)
        self.assertEqual(date(2018, 3, 1), xrange.date_from)
        self.assertEqual(date(2018, 3, 31), xrange.date_to)

    def test_yearly(self):
        sequence = self.get_sequence("yearly")
        self.assertFalse(sequence.date_range_ids)
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )
        xrange = sequence.date_range_ids
        self.assertTrue(xrange)
        self.assertEqual(date(2018, 1, 1), xrange.date_from)
        self.assertEqual(date(2018, 12, 31), xrange.date_to)

    def test_monthly_existing(self):
        sequence = self.get_sequence("monthly")
        self.env["ir.sequence.date_range"].create(
            {
                "date_from": date(2018, 3, 1),
                "date_to": date(2018, 3, 10),
                "sequence_id": sequence.id,
            }
        )
        self.env["ir.sequence.date_range"].create(
            {
                "date_from": date(2018, 3, 20),
                "date_to": date(2018, 3, 25),
                "sequence_id": sequence.id,
            }
        )
        self.assertEqual(
            "00001", sequence.with_context(ir_sequence_date=self.date).next_by_id()
        )