partner-contact/partner_address_version/tests/test_address_version.py

# -*- coding: utf-8 -*-
# Copyright 2018 Akretion - Benoît Guillot
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo.tests import SavepointCase
from collections import OrderedDict
import hashlib
from odoo.exceptions import UserError


class TestAddressVersion(SavepointCase):

    @classmethod
    def setUpClass(cls):
        super(TestAddressVersion, cls).setUpClass()
        cls.partner_vals = OrderedDict([
            ('name', u'Name'),
            ('street', u'Street'),
            ('street2', u'Street2'),
            ('zip', u'Zip'),
            ('city', u'City'),
            ('country_id', cls.env.ref('base.fr'))
        ])
        create_vals = cls.partner_vals.copy()
        create_vals['country_id'] = cls.env.ref('base.fr').id
        create_vals_2 = create_vals.copy()
        cls.partner = cls.env['res.partner'].create(create_vals)
        cls.partner_2 = cls.env['res.partner'].create(create_vals_2)
        cls.partner_vals.update({'parent_id': cls.partner.id})

    def test_hash(self):
        test_hash = hashlib.md5(str(self.partner_vals)).hexdigest()
        self.assertEqual(test_hash, self.partner.get_version_hash())

    def test_create_version_partner(self):
        new_partner = self.partner.get_address_version()
        self.assertEqual(new_partner.active, False)
        self.assertNotEqual(new_partner.id, self.partner.id)
        self.assertEqual(new_partner.parent_id.id, self.partner.id)

    def test_get_version_hash(self):
        self.partner.version_hash = self.partner.get_version_hash()
        self.partner.active = False
        version_partner = self.partner.get_address_version()
        self.assertEqual(version_partner.id, self.partner.id)

    def test_write_versioned_partner(self):
        new_partner = self.partner.get_address_version()
        with self.assertRaises(UserError):
            new_partner.street = 'New street'

    def test_same_address_different_parent(self):
        new_partner = self.partner.get_address_version()
        new_partner_2 = self.partner_2.get_address_version()
        for field in self.partner.get_version_fields():
            if field == 'parent_id':
                continue
            self.assertEqual(new_partner[field], new_partner_2[field])
        self.assertNotEqual(new_partner.id, new_partner_2.id)
        self.assertNotEqual(
            new_partner.version_hash,
            new_partner_2.version_hash
        )