fix tests

Author: JoeyRubas

mittab/libs/tests/views/test_return_to.py

@@ -52,9 +52,27 @@ def test_with_return_to_keeps_urls_clean():
 
 
 def test_login_middleware_stores_return_to_for_get_requests():
-    request = RequestFactory().get("/pairings/status/?round=1", HTTP_HOST="testserver")
-    mock_groups = type("Groups", (), {"filter": lambda *args, **kwargs: type("QuerySet", (), {"exists": lambda: False})()})()
-    request.user = type("User", (), {"is_anonymous": False, "is_authenticated": True, "groups": mock_groups})()
+    request = RequestFactory().get(
+        "/pairings/status/?round=1", HTTP_HOST="testserver"
+    )
+    mock_groups = type(
+        "Groups",
+        (),
+        {
+            "filter": lambda *args, **kwargs: type(
+                "QuerySet", (), {"exists": lambda: False}
+            )()
+        },
+    )()
+    request.user = type(
+        "User",
+        (),
+        {
+            "is_anonymous": False,
+            "is_authenticated": True,
+            "groups": mock_groups,
+        },
+    )()
     request.session = {}
 
     response = Login(lambda req: HttpResponse("ok"))(request)
@@ -69,8 +87,24 @@ def test_login_middleware_prefers_internal_referer_for_return_to():
         HTTP_HOST="testserver",
         HTTP_REFERER="http://testserver/batch_checkin/",
     )
-    mock_groups = type("Groups", (), {"filter": lambda *args, **kwargs: type("QuerySet", (), {"exists": lambda: False})()})()
-    request.user = type("User", (), {"is_anonymous": False, "is_authenticated": True, "groups": mock_groups})()
+    mock_groups = type(
+        "Groups",
+        (),
+        {
+            "filter": lambda *args, **kwargs: type(
+                "QuerySet", (), {"exists": lambda: False}
+            )()
+        },
+    )()
+    request.user = type(
+        "User",
+        (),
+        {
+            "is_anonymous": False,
+            "is_authenticated": True,
+            "groups": mock_groups,
+        },
+    )()
     request.session = {}
 
     response = Login(lambda req: HttpResponse("ok"))(request)