authentik/passbook/policies/group_membership/tests.py

"""evaluator tests"""
from django.test import TestCase
from guardian.shortcuts import get_anonymous_user

from passbook.core.models import Group
from passbook.policies.group_membership.models import GroupMembershipPolicy
from passbook.policies.types import PolicyRequest


class TestGroupMembershipPolicy(TestCase):
    """GroupMembershipPolicy tests"""

    def setUp(self):
        self.request = PolicyRequest(user=get_anonymous_user())

    def test_invalid(self):
        """user not in group"""
        group = Group.objects.create(name="test")
        policy: GroupMembershipPolicy = GroupMembershipPolicy.objects.create(
            group=group
        )
        self.assertFalse(policy.passes(self.request).passing)

    def test_valid(self):
        """user in group"""
        group = Group.objects.create(name="test")
        group.users.add(get_anonymous_user())
        group.save()
        policy: GroupMembershipPolicy = GroupMembershipPolicy.objects.create(
            group=group
        )
        self.assertTrue(policy.passes(self.request).passing)