Fix back link for initial set end user page

uktrade · Jan 17, 2025 · d6de5fa · d6de5fa
1 parent 2718c8f
commit d6de5fa
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/exporter/applications/views/parties/end_users.py b/exporter/applications/views/parties/end_users.py
@@ -142,6 +142,13 @@ class SetPartyView(LoginRequiredMixin, BaseSessionWizardView):
     def application(self):
         return get_application(self.request, self.kwargs["pk"])
 
+    # back_link_url only needs to be set for the first question in the form wizard,
+    # for subsequent questions the back link is generated through another mechanism
+    def get_context_data(self, form, **kwargs):
+        context = super().get_context_data(form, **kwargs)
+        context.update({"back_link_url": reverse("applications:add_end_user", kwargs=self.kwargs)})
+        return context
+
     def get_form_kwargs(self, step=None):
         kwargs = super().get_form_kwargs(step)