policies/reputation: fix tests

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>

authentik/policies/reputation/models.py

@@ -4,7 +4,6 @@ from django.db import models
 from django.utils.translation import gettext as _
 from rest_framework.serializers import BaseSerializer
 
-from authentik.core.models import User
 from authentik.lib.utils.http import get_client_ip
 from authentik.policies.models import Policy
 from authentik.policies.types import PolicyRequest, PolicyResult

authentik/policies/reputation/tasks.py

@@ -2,7 +2,6 @@
 from django.core.cache import cache
 from structlog.stdlib import get_logger
 
-from authentik.core.models import User
 from authentik.events.monitored_tasks import MonitoredTask, TaskResult, TaskResultStatus
 from authentik.policies.reputation.models import IPReputation, UserReputation
 from authentik.policies.reputation.signals import (

authentik/policies/reputation/tests.py

@@ -44,7 +44,9 @@ class TestReputationPolicy(TestCase):
         self.assertEqual(cache.get(CACHE_KEY_USER_PREFIX + self.test_username), -1)
         # Save cache and check db values
         save_user_reputation.delay().get()
-        self.assertEqual(UserReputation.objects.get(user=self.user).score, -1)
+        self.assertEqual(
+            UserReputation.objects.get(username=self.test_username).score, -1
+        )
 
     def test_policy(self):
         """Test Policy"""