Files
authentik/authentik/policies/tests/test_bindings_api.py
Jens L 8949464294 root: reformat with latest black version and fix tests (#8376)
* format files

Signed-off-by: Jens Langhammer <jens@goauthentik.io>

* fix pyright

Signed-off-by: Jens Langhammer <jens@goauthentik.io>

* revert #8367

Signed-off-by: Jens Langhammer <jens@goauthentik.io>

* sigh

Signed-off-by: Jens Langhammer <jens@goauthentik.io>

---------

Signed-off-by: Jens Langhammer <jens@goauthentik.io>
2024-01-31 15:24:45 +01:00

54 lines
1.7 KiB
Python

"""Test bindings API"""
from django.urls import reverse
from rest_framework.test import APITestCase
from authentik.core.tests.utils import create_test_admin_user
from authentik.policies.models import PolicyBindingModel
class TestBindingsAPI(APITestCase):
"""Test bindings API"""
def setUp(self) -> None:
super().setUp()
self.pbm = PolicyBindingModel.objects.create()
self.user = create_test_admin_user()
self.group = self.user.ak_groups.first()
self.client.force_login(self.user)
def test_valid_binding(self):
"""Test valid binding"""
response = self.client.post(
reverse("authentik_api:policybinding-list"),
data={"target": self.pbm.pk, "user": self.user.pk, "order": 0},
)
self.assertEqual(response.status_code, 201)
def test_invalid_too_much(self):
"""Test invalid binding (too much)"""
response = self.client.post(
reverse("authentik_api:policybinding-list"),
data={
"target": self.pbm.pk,
"user": self.user.pk,
"group": self.group.pk,
"order": 0,
},
)
self.assertJSONEqual(
response.content.decode(),
{"non_field_errors": ["Only one of 'policy', 'group' or 'user' can be set."]},
)
def test_invalid_too_little(self):
"""Test invvalid binding (too little)"""
response = self.client.post(
reverse("authentik_api:policybinding-list"),
data={"target": self.pbm.pk, "order": 0},
)
self.assertJSONEqual(
response.content.decode(),
{"non_field_errors": ["One of 'policy', 'group' or 'user' must be set."]},
)