partner-contact/partner_contact_in_several_companies/tests/test_partner_contact_in_several_companies.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 PartnerContactInSeveralCompaniesCase(common.TransactionCase):

    def setUp(self):
        """*****setUp*****"""
        super(PartnerContactInSeveralCompaniesCase, 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',
                 'partner_contact_in_several_companies',
                 'res_partner_contact1'),
                ('bob_job1_id',
                 'partner_contact_in_several_companies',
                 'res_partner_contact1_work_position1'),
                ('roger_contact_id', 'base', 'res_partner_main2'),
                ('roger_job2_id',
                 'partner_contact_in_several_companies',
                 '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.assertIn(
            self.bob_job1_id,
            read_contacts(self.main_partner_id, context=ctx),
        )
        ctx = {'search_show_all_positions': False}
        self.assertIn(
            self.bob_job1_id,
            read_contacts(self.main_partner_id, context=ctx),
        )
        ctx = {'search_show_all_positions': True}
        self.assertIn(
            self.bob_job1_id,
            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
        # 'YourCompany'
        # so search for all contacts working for 'YourCompany' should contain
        # bob position.
        partner_ids = self.partner.search(
            cr, uid,
            [('parent_id', 'ilike', 'YourCompany')],
            context=None
        )
        self.assertIn(self.bob_job1_id, 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', 'YourCompany')],
            context=ctx
        )
        self.assertIn(self.bob_contact_id, partner_ids, )

    def test_04_contact_creation(self):
        """Check that we're begin to create a contact"""
        cr, uid = self.cr, self.uid

        # Create a contact using only name
        new_contact_id = self.partner.create(cr, uid, {'name': 'Bob Egnops'})
        self.assertEqual(
            self.partner.browse(cr, uid, new_contact_id).contact_type,
            'standalone',
        )

        # Create a contact with only contact_id
        new_contact_id = self.partner.create(
            cr, uid, {'contact_id': self.bob_contact_id}
        )
        new_contact = self.partner.browse(cr, uid, new_contact_id)
        self.assertEqual(new_contact.name, 'Bob Egnops')
        self.assertEqual(new_contact.contact_type, 'attached')

        # Create a contact with both contact_id and name;
        # contact's name should override provided value in that case
        new_contact_id = self.partner.create(
            cr, uid, {'contact_id': self.bob_contact_id, 'name': 'Rob Egnops'}
        )
        self.assertEqual(
            self.partner.browse(cr, uid, new_contact_id).name,
            'Bob Egnops'
        )

        # Reset contact to standalone
        self.partner.write(cr, uid, [new_contact_id], {'contact_id': False})
        self.assertEqual(
            self.partner.browse(cr, uid, new_contact_id).contact_type,
            'standalone',
        )

    def test_05_contact_fields_sync(self):
        """Check that contact's fields are correctly synced between
        parent contact or related contacts
        """
        cr, uid = self.cr, self.uid

        # Test DOWNSTREAM sync
        self.partner.write(
            cr, uid, [self.bob_contact_id], {'name': 'Rob Egnops'}
        )
        self.assertEqual(
            self.partner.browse(cr, uid, self.bob_job1_id).name,
            'Rob Egnops',
        )

        # Test UPSTREAM sync
        self.partner.write(cr, uid, [self.bob_job1_id], {'name': 'Bob Egnops'})
        self.assertEqual(
            self.partner.browse(cr, uid, self.bob_contact_id).name,
            'Bob Egnops',
        )