File: test_res_partner_iap.py

package info (click to toggle)
odoo 18.0.0%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 878,716 kB
  • sloc: javascript: 927,937; python: 685,670; xml: 388,524; sh: 1,033; sql: 415; makefile: 26
file content (77 lines) | stat: -rw-r--r-- 3,473 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

import psycopg2

from odoo.addons.mail.tests.common import MailCommon
from odoo.tools import mute_logger


class TestResPartnerIap(MailCommon):

    @mute_logger("odoo.sql_db")
    def test_res_partner_iap_constraint(self):
        partner = self.partner_employee

        self.env["res.partner.iap"].search([("partner_id", "=", partner.id)]).unlink()
        self.env["res.partner.iap"].create({"partner_id": partner.id, "iap_enrich_info": "test info"})

        with self.assertRaises(psycopg2.IntegrityError, msg="Can create only one partner IAP per partner"):
            self.env["res.partner.iap"].create({"partner_id": partner.id, "iap_enrich_info": "test info"})

    def test_res_partner_iap_compute_iap_enrich_info(self):
        partner = self.partner_employee

        self.assertFalse(partner.iap_enrich_info)

        partner_iap = self.env["res.partner.iap"].create({"partner_id": partner.id, "iap_enrich_info": "test info"})
        partner.invalidate_recordset()
        self.assertEqual(partner.iap_enrich_info, "test info")

        partner_iap.unlink()
        partner.iap_enrich_info = "test info 2"

        partner_iap = self.env["res.partner.iap"].search([("partner_id", "=", partner.id)])
        self.assertTrue(partner_iap, "Should have create the <res.partner.iap>")
        self.assertEqual(partner_iap.iap_enrich_info, "test info 2")

        partner.iap_enrich_info = "test info 3"
        partner_iap.invalidate_recordset()
        new_partner_iap = self.env["res.partner.iap"].search([("partner_id", "=", partner.id)])
        self.assertEqual(new_partner_iap, partner_iap, "Should have write on the existing one")
        self.assertEqual(partner_iap.iap_enrich_info, "test info 3")

    def test_res_partner_iap_creation(self):
        partner = self.env['res.partner'].create({
            'name': 'Test partner',
            'iap_enrich_info': 'enrichment information',
            'iap_search_domain': 'qsd@example.com',
        })

        partner.invalidate_recordset()

        self.assertEqual(partner.iap_enrich_info, 'enrichment information')
        self.assertEqual(partner.iap_search_domain, 'qsd@example.com')

        partner_iap = self.env['res.partner.iap'].search([('partner_id', '=', partner.id)])
        self.assertEqual(len(partner_iap), 1, 'Should create only one <res.partner.iap>')
        self.assertEqual(partner_iap.iap_enrich_info, 'enrichment information')
        self.assertEqual(partner_iap.iap_search_domain, 'qsd@example.com')

    def test_res_partner_iap_writing(self):
        partner = self.env['res.partner'].create({
            'name': 'Test partner 2',
        })
        partner.write({
            'iap_enrich_info': 'second information',
            'iap_search_domain': 'xyz@example.com',
        })
        partner_iap = self.env['res.partner.iap'].search([('partner_id', '=', partner.id)])
        self.assertEqual(len(partner_iap), 1, 'Should create only one <res.partner.iap>')
        self.assertEqual(partner_iap.iap_enrich_info, 'second information')
        self.assertEqual(partner_iap.iap_search_domain, 'xyz@example.com')

        partner.iap_search_domain = "only write on domain"
        partner_iap.invalidate_recordset()
        self.assertEqual(partner_iap.iap_enrich_info, 'second information')
        self.assertEqual(partner_iap.iap_search_domain, 'only write on domain')