admin: fix linting in api tests

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-05-10 20:14:21 +02:00
parent 6f58fdf158
commit 51e3453dca
1 changed files with 10 additions and 7 deletions

View File

@ -1,5 +1,4 @@
"""test admin api""" """test admin api"""
from authentik.events.monitored_tasks import TaskResultStatus
from json import loads from json import loads
from django.test import TestCase from django.test import TestCase
@ -8,6 +7,7 @@ from django.urls import reverse
from authentik import __version__ from authentik import __version__
from authentik.core.models import Group, User from authentik.core.models import Group, User
from authentik.core.tasks import clean_expired_models from authentik.core.tasks import clean_expired_models
from authentik.events.monitored_tasks import TaskResultStatus
class TestAdminAPI(TestCase): class TestAdminAPI(TestCase):
@ -35,17 +35,20 @@ class TestAdminAPI(TestCase):
"""Test Task API (read single)""" """Test Task API (read single)"""
clean_expired_models.delay() clean_expired_models.delay()
response = self.client.get( response = self.client.get(
reverse("authentik_api:admin_system_tasks-detail", kwargs={ reverse(
"pk": "clean_expired_models" "authentik_api:admin_system_tasks-detail",
})) kwargs={"pk": "clean_expired_models"},
)
)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
body = loads(response.content) body = loads(response.content)
self.assertEqual(body["status"], TaskResultStatus.SUCCESSFUL.name) self.assertEqual(body["status"], TaskResultStatus.SUCCESSFUL.name)
self.assertEqual(body["task_name"], "clean_expired_models") self.assertEqual(body["task_name"], "clean_expired_models")
response = self.client.get( response = self.client.get(
reverse("authentik_api:admin_system_tasks-detail", kwargs={ reverse(
"pk": "qwerqwer" "authentik_api:admin_system_tasks-detail", kwargs={"pk": "qwerqwer"}
})) )
)
self.assertEqual(response.status_code, 404) self.assertEqual(response.status_code, 404)
def test_tasks_retry(self): def test_tasks_retry(self):