# Copyright 2016 Pedro M. Baeza <pedro.baeza@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo.tests import common
from odoo.tools.safe_eval import safe_eval


class TestDeduplicateFilter(common.TransactionCase):
    def setUp(self):
        super(TestDeduplicateFilter, self).setUp()
        self.partner_1 = self.env['res.partner'].create({
            'name': 'Partner 1',
            'email': 'partner1@example.org',
            'is_company': True,
            'parent_id': False,
        })
        self.partner_1.copy()
        self.partner_2 = self.env['res.partner'].create({
            'name': 'Partner 2',
            'email': 'partner2@example.org',
            'is_company': False,
            'parent_id': self.partner_1.id,
        })
        self.partner_2.copy()
        self.partner_3 = self.env['res.partner'].create({
            'name': 'Partner 3',
            'email': 'partner3@example.org',
            'is_company': False,
            'parent_id': False,
        })
        self.partner_3.copy()
        self.wizard = self.env['base.partner.merge.automatic.wizard'].create({
            'group_by_email': True,
        })

    def test_deduplicate_exclude_is_company(self):
        self.wizard.exclude_is_company = True
        self.wizard.action_start_manual_process()
        matched_founds = 0
        for line in self.wizard.line_ids:
            match_ids = safe_eval(line.aggr_ids)
            if self.partner_1.id in match_ids:
                self.assertTrue(False, 'Partner with is company not excluded')
            if self.partner_2.id in match_ids:
                matched_founds += 1
            if self.partner_3.id in match_ids:
                matched_founds += 1
        self.assertEqual(matched_founds, 2)

    def test_deduplicate_exclude_not_parent(self):
        self.wizard.exclude_not_parent = True
        self.wizard.action_start_manual_process()
        matched_founds = 0
        for line in self.wizard.line_ids:
            match_ids = safe_eval(line.aggr_ids)
            if self.partner_1.id in match_ids:
                self.assertTrue(False, 'Partner without parent not excluded')
            if self.partner_3.id in match_ids:
                self.assertTrue(False, 'Partner without parent not excluded')
            if self.partner_2.id in match_ids:
                matched_founds += 1
        self.assertEqual(matched_founds, 1)

    def test_deduplicate_exclude_parent(self):
        self.wizard.exclude_parent = True
        self.wizard.action_start_manual_process()
        matched_founds = 0
        for line in self.wizard.line_ids:
            match_ids = safe_eval(line.aggr_ids)
            if self.partner_2.id in match_ids:
                self.assertTrue(False, 'Partner with parent not excluded')
            if self.partner_1.id in match_ids:
                matched_founds += 1
            if self.partner_3.id in match_ids:
                matched_founds += 1
        self.assertEqual(matched_founds, 2)