Fix missing next URL from initial login form data

Author: codingjoe

mailauth/views.py

@@ -35,6 +35,12 @@ def form_valid(self, form):
     def get_success_url(self):
         return resolve_url(self.success_url)
 
+    def get_initial(self):
+        return {
+            self.redirect_field_name: self.request.GET.get(self.redirect_field_name),
+            **super().get_initial(),
+        }
+
 
 INTERNAL_LOGIN_URL_TOKEN = 'login-token'
 

tests/test_views.py

@@ -1,5 +1,16 @@
 from django.urls import reverse
 
+from mailauth.views import LoginView
+
+
+class TestLoginView:
+
+    def test_get_initial(self, rf):
+        view = LoginView()
+        view.request = rf.get("/", data={"next": "foo/bar"})
+
+        assert view.get_initial() == {"next": "foo/bar"}
+
 
 class TestLoginTokenView: