partner-contact/partner_contact_address_default/tests/test_partner_contact_address_default.py

# Copyright 2020 Tecnativa - Carlos Dauden
# Copyright 2020 Tecnativa - Sergio Teruel
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo.tests import common


class TestPartnerContactAddressDefault(common.TransactionCase):

    def setUp(self):
        super().setUp()
        self.Partner = self.env['res.partner']
        self.partner = self.Partner.create({
            'name': 'Parent partner',
        })
        self.partner_child_delivery1 = self.Partner.create({
            'name': 'Child delivery 1',
            'type': 'delivery',
            'parent_id': self.partner.id,
        })
        self.partner_child_delivery2 = self.Partner.create({
            'name': 'Child delivery 2',
            'type': 'delivery',
            'parent_id': self.partner.id,
        })
        self.partner_child_invoice = self.Partner.create({
            'name': 'Child invoice',
            'type': 'invoice',
            'parent_id': self.partner.id,
        })

    def test_contact_address_default(self):
        self.partner.partner_delivery_id = self.partner
        self.partner.partner_invoice_id = self.partner
        res = self.partner.address_get()
        self.assertEqual(res['delivery'], self.partner)
        self.assertEqual(res['invoice'], self.partner)

        self.partner_child_delivery2.partner_delivery_id =\
            self.partner_child_delivery2
        self.partner_child_delivery2.partner_invoice_id =\
            self.partner_child_delivery2
        res = self.partner_child_delivery2.address_get()
        self.assertEqual(res['delivery'], self.partner_child_delivery2)
        self.assertEqual(res['invoice'], self.partner_child_delivery2)