Fix takeover backend

d120 · May 25, 2020 · 1c2d132 · 1c2d132
1 parent 81d266d
commit 1c2d132
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/feedback/auth.py b/src/feedback/auth.py
@@ -27,7 +27,7 @@ def authenticate(self, request, vid, token):
 
 
 class TakeoverBackend(ModelBackend):
-    def authenticate(self, user, current_user=None, reset=False):
+    def authenticate(self, request, user, current_user=None, reset=False):
         if reset or (current_user and current_user.is_superuser):
             return user
         return None

diff --git a/src/feedback/tests/test_auth.py b/src/feedback/tests/test_auth.py
@@ -37,10 +37,10 @@ def setUp(self):
         self.uj.is_superuser = True
 
     def test_authenticate(self):
-        self.assertFalse(self.b.authenticate(user=self.ub))
-        self.assertFalse(self.b.authenticate(user=self.ub, current_user=self.ub))
-        self.assertEqual(self.b.authenticate(user=self.ub, current_user=self.uj), self.ub)
-        self.assertEqual(self.b.authenticate(user=self.ub, reset=True), self.ub)
+        self.assertFalse(self.b.authenticate(request=None, user=self.ub))
+        self.assertFalse(self.b.authenticate(request=None, user=self.ub, current_user=self.ub))
+        self.assertEqual(self.b.authenticate(request=None, user=self.ub, current_user=self.uj), self.ub)
+        self.assertEqual(self.b.authenticate(request=None, user=self.ub, reset=True), self.ub)
 
 
 class FSAccountBackendTest(TestCase):