# Copyright 2017 Onestein (<http://www.onestein.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo.tests.common import TransactionCase


class TestOnchangeHelper(TransactionCase):
    def test01_partner_parent(self):
        main_partner = self.env.ref('base.main_partner')
        input_vals = dict(parent_id=main_partner.id, type='contact')
        updated_vals = self.env['res.partner'].play_onchanges(
            input_vals, ['parent_id']
        )
        self.assertIn('country_id', updated_vals)
        self.assertIn('state_id', updated_vals)
        self.assertIn('street', updated_vals)
        self.assertIn('zip', updated_vals)

        self.assertEqual(
            updated_vals['country_id'], main_partner.country_id.id
        )
        self.assertEqual(updated_vals['state_id'], main_partner.state_id.id)
        self.assertEqual(updated_vals['street'], main_partner.street)
        self.assertEqual(updated_vals['zip'], main_partner.zip)

    def test02_partner_country(self):
        partner_demo = self.env.ref('base.partner_demo')
        input_vals = {'country_id': self.env.ref('base.us').id}
        updated_vals = partner_demo.play_onchanges(input_vals, ['country_id'])
        self.assertIn('country_id', updated_vals)

    def test_playing_onchange_on_model(self):
        result = self.env['res.partner'].play_onchanges(
            {'company_type': 'company'}, ['company_type']
        )
        self.assertEqual(result['is_company'], True)