partner-contact/partner_contact_age_range/tests/test_res_partner_age_range.py

# Copyright 2019-2020: Druidoo (<https://www.druidoo.io>)
# Copyright 2020 Ecosoft Co., Ltd (http://ecosoft.co.th/)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from datetime import datetime

from dateutil.relativedelta import relativedelta

from odoo.exceptions import ValidationError
from odoo.tests.common import TransactionCase


class TestRespartnerAgeRange(TransactionCase):
    def setUp(self):
        super().setUp()
        self.range_model = self.env["res.partner.age.range"]
        self.partner_model = self.env["res.partner"]
        self.baby_range = self.range_model.create(
            {"name": "baby", "age_from": 0, "age_to": 2}
        )
        self.partner = self.partner_model.create(
            {
                "name": "Test",
                "birthdate_date": datetime.today() - relativedelta(years=1, days=10),
            }
        )

    def test_age_from(self):
        age_from = self.range_model._default_age_from()
        toddler_range = self.range_model.create(
            {"name": "Toddler", "age_from": age_from, "age_to": 4}
        )
        self.assertEqual(toddler_range.age_from, self.baby_range.age_to + 1)

    def test_validate_range(self):
        with self.assertRaises(ValidationError):
            self.range_model.create({"name": "Child", "age_from": 1, "age_to": 12})
        with self.assertRaises(ValidationError):
            self.range_model.create({"name": "Teenager", "age_from": 16, "age_to": 15})

    def test_cron_update_age_range_id(self):
        self.partner_model._cron_update_age_range_id()
        self.assertEqual(self.partner.age_range_id, self.baby_range)