base_name_search_improved/tests/test_name_search.py

# -*- coding: utf-8 -*-
# © 2016 Daniel Reis
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from openerp.tests.common import TransactionCase, at_install, post_install


@at_install(False)
@post_install(True)
class NameSearchCase(TransactionCase):

    def setUp(self):
        super(NameSearchCase, self).setUp()
        phone_field = self.env.ref('base.field_res_partner_phone')
        model_partner = self.env.ref('base.model_res_partner')
        model_partner.name_search_ids = phone_field
        self.Partner = self.env['res.partner']
        self.partner1 = self.Partner.create(
            {'name': 'Luigi Verconti',
             'customer': True,
             'phone': '+351 555 777 333'})
        self.partner2 = self.Partner.create(
            {'name': 'Ken Shabby',
             'customer': True,
             'phone': '+351 555 333 777'})
        self.partner3 = self.Partner.create(
            {'name': 'Johann Gambolputty of Ulm',
             'supplier': True,
             'phone': '+351 777 333 555'})

    def test_RelevanceOrderedResults(self):
        """Return results ordered by relevance"""
        res = self.Partner.name_search('555 777')
        self.assertEqual(
            res[0][0], self.partner1.id,
            'Match full string honoring spaces')
        self.assertEqual(
            res[1][0], self.partner2.id,
            'Match words honoring order of appearance')
        self.assertEqual(
            res[2][0], self.partner3.id,
            'Match all words, regardless of order of appearance')

    def test_NameSearchMustMatchAllWords(self):
        """Must Match All Words"""
        res = self.Partner.name_search('ulm 555 777')
        self.assertFalse(res)

    def test_MustHonorDomain(self):
        """Must also honor a provided Domain"""
        res = self.Partner.name_search('+351', args=[('supplier', '=', True)])
        gambulputty = self.partner3.id
        self.assertEqual(len(res), 1)
        self.assertEqual(res[0][0], gambulputty)