partner-contact/base_contact/tests/test_base_contact.py

# -*- coding: utf-8 ⁻*-
##############################################################################
#
#    OpenERP, Open Source Business Applications
#    Copyright (C) 2013-TODAY OpenERP S.A. (<http://openerp.com>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

from openerp.tests import common


class Test_Base_Contact(common.TransactionCase):

    def setUp(self):
        """*****setUp*****"""
        super(Test_Base_Contact, self).setUp()
        cr, uid = self.cr, self.uid
        ModelData = self.registry('ir.model.data')
        self.partner = self.registry('res.partner')

        # Get test records reference
        for attr, module, name in [
                ('main_partner_id', 'base', 'main_partner'),
                ('bob_contact_id', 'base_contact', 'res_partner_contact1'),
                ('bob_job1_id', 'base_contact', 'res_partner_contact1_work_position1'),
                ('roger_contact_id', 'base', 'res_partner_main2'),
                ('roger_job2_id', 'base_contact', 'res_partner_main2_position_consultant')]:
            r = ModelData.get_object_reference(cr, uid, module, name)
            setattr(self, attr, r[1] if r else False)

    def test_00_show_only_standalone_contact(self):
        """Check that only standalone contact are shown if context explicitly state to not display all positions"""
        cr, uid = self.cr, self.uid
        ctx = {'search_show_all_positions': False}
        partner_ids = self.partner.search(cr, uid, [], context=ctx)
        partner_ids.sort()
        self.assertTrue(self.bob_job1_id not in partner_ids)
        self.assertTrue(self.roger_job2_id not in partner_ids)

    def test_01_show_all_positions(self):
        """Check that all contact are show if context is empty or explicitly state to display all positions"""
        cr, uid = self.cr, self.uid

        partner_ids = self.partner.search(cr, uid, [], context=None)
        self.assertTrue(self.bob_job1_id in partner_ids)
        self.assertTrue(self.roger_job2_id in partner_ids)

        ctx = {'search_show_all_positions': True}
        partner_ids = self.partner.search(cr, uid, [], context=ctx)
        self.assertTrue(self.bob_job1_id in partner_ids)
        self.assertTrue(self.roger_job2_id in partner_ids)

    def test_02_reading_other_contact_one2many_show_all_positions(self):
        """Check that readonly partner's ``other_contact_ids`` return all values whatever the context"""
        cr, uid = self.cr, self.uid

        def read_other_contacts(pid, context=None):
            return self.partner.read(cr, uid, [pid], ['other_contact_ids'], context=context)[0]['other_contact_ids']

        def read_contacts(pid, context=None):
            return self.partner.read(cr, uid, [pid], ['child_ids'], context=context)[0]['child_ids']

        ctx = None
        self.assertEqual(read_other_contacts(self.bob_contact_id, context=ctx), [self.bob_job1_id])
        ctx = {'search_show_all_positions': False}
        self.assertEqual(read_other_contacts(self.bob_contact_id, context=ctx), [self.bob_job1_id])
        ctx = {'search_show_all_positions': True}
        self.assertEqual(read_other_contacts(self.bob_contact_id, context=ctx), [self.bob_job1_id])

        ctx = None
        self.assertTrue(self.bob_job1_id in read_contacts(self.main_partner_id, context=ctx))
        ctx = {'search_show_all_positions': False}
        self.assertTrue(self.bob_job1_id in read_contacts(self.main_partner_id, context=ctx))
        ctx = {'search_show_all_positions': True}
        self.assertTrue(self.bob_job1_id in read_contacts(self.main_partner_id, context=ctx))

    def test_03_search_match_attached_contacts(self):
        """Check that searching partner also return partners having attached contacts matching search criteria"""
        cr, uid = self.cr, self.uid
        # Bob's contact has one other position which is related to 'Your Company'
        # so search for all contacts working for 'Your Company' should contain bob position.
        partner_ids = self.partner.search(cr, uid, [('parent_id', 'ilike', 'Your Company')], context=None)
        self.assertTrue(self.bob_job1_id in partner_ids)

        # but when searching without 'all positions', we should get the position standalone contact instead.
        ctx = {'search_show_all_positions': False}
        partner_ids = self.partner.search(cr, uid, [('parent_id', 'ilike', 'Your Company')], context=ctx)
        self.assertTrue(self.bob_contact_id in partner_ids)