partner-contact/portal_partner_select_all/tests/test_portal_partner_select_all.py

# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo.tests import common, Form


class TestPortalPartnerSelctAll(common.TransactionCase):

    def setUp(self):
        super().setUp()

        Partner = self.env["res.partner"]
        self.partner1 = Partner.create({'name': 'P1', 'email': 'p1@p1'})
        self.partner2 = Partner.create({'name': 'P2', 'email': 'p2@p2'})
        self.partner3 = Partner.create({'name': 'P3', 'email': 'p3@p3'})
        self.wizard_all = self.env["portal.wizard"].with_context({
            'active_ids': [self.partner1.id, self.partner2.id]
        }).create({})
        self.wizard_default = self.env["portal.wizard"].with_context({
            'active_ids': [self.partner1.id, self.partner2.id, self.partner3.id]
        }).create({})

    def test_portal_partner_select_all_wizard(self):
        # check selecting all
        wizard_all_form = Form(self.wizard_all)
        wizard_all_form.set_all_users = True
        w = wizard_all_form.save()
        w.action_apply()
        # partner should have user records with assigned portal group
        self.assertTrue(self.partner1.user_ids, "Partner should have user")
        self.assertTrue(self.partner2.user_ids, "Partner should have user")
        self.assertTrue(self.partner1.user_ids[0].has_group('base.group_portal'))
        self.assertTrue(self.partner2.user_ids[0].has_group('base.group_portal'))

        # checking toogle
        wizard_default_form = Form(self.wizard_default)
        wizard_default_form.set_all_users = True
        wizard_default_form.set_all_users = False
        w = wizard_default_form.save()
        w.action_apply()
        self.assertFalse(self.partner3.user_ids, "Partner shouldn't have a user")