import re

with open('addons/education_lms/tests/test_education_lms.py', 'r') as f:
    text = f.read()

# Add base.group_portal to user_student_a and b
text = text.replace(
"""        self.user_student_a = self.env['res.users'].create({
            'name': 'Student A',
            'login': 'student_a',
            'groups_id': [(4, self.group_student.id)]
        })""",
"""        self.user_student_a = self.env['res.users'].create({
            'name': 'Student A',
            'login': 'student_a',
            'groups_id': [(4, self.group_student.id), (4, self.env.ref('base.group_portal').id)]
        })"""
)

text = text.replace(
"""        self.user_student_b = self.env['res.users'].create({
            'name': 'Student B',
            'login': 'student_b',
            'groups_id': [(4, self.group_student.id)]
        })""",
"""        self.user_student_b = self.env['res.users'].create({
            'name': 'Student B',
            'login': 'student_b',
            'groups_id': [(4, self.group_student.id), (4, self.env.ref('base.group_portal').id)]
        })"""
)

# Also remove the debug prints I added previously, to keep the test clean
text = re.sub(r' *print\(f"\\nDEBUG RULE PERM_READ.*?student_ids\n', '', text, flags=re.DOTALL)

with open('addons/education_lms/tests/test_education_lms.py', 'w') as f:
    f.write(text)