fix unit test

ocadotechnology · Oct 2, 2023 · ff0750a · ff0750a
1 parent b123c05
commit ff0750a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/backend/api/tests/test_views.py b/backend/api/tests/test_views.py
@@ -29,7 +29,11 @@ def test_post__otp(self):
 
         assert response.status_code == 200
         self.assertDictEqual(
-            response.json(), {"auth_factors": [AuthFactor.Type.OTP]}
+            response.json(),
+            {
+                "auth_factors": [AuthFactor.Type.OTP],
+                "otp_bypass_token_exists": False,
+            },
         )
 
         self.user.userprofile.otp_secret = pyotp.random_base32()
@@ -45,7 +49,13 @@ def test_post__otp(self):
             )
 
         assert response.status_code == 200
-        self.assertDictEqual(response.json(), {"auth_factors": []})
+        self.assertDictEqual(
+            response.json(),
+            {
+                "auth_factors": [],
+                "otp_bypass_token_exists": False,
+            },
+        )
 
 
 class TestClearExpiredView(CronTestCase):