From a7567ad8c678315ae92448199fae8fb40afecaed Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Sun, 10 May 2020 18:45:16 +0200
Subject: [PATCH] stages/identification: add recovery support

---
 passbook/core/templates/login/base.html | 11 ++++++++++-
 passbook/stages/identification/stage.py | 11 ++++++++---
 passbook/stages/identification/tests.py | 18 +++++++++++++++++-
 3 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/passbook/core/templates/login/base.html b/passbook/core/templates/login/base.html
index 15f99a209..6d1070dae 100644
--- a/passbook/core/templates/login/base.html
+++ b/passbook/core/templates/login/base.html
@@ -63,12 +63,21 @@
                     </li>
                     {% endfor %}
                 </ul>
-                {% if enroll_url %}
+                {% if enroll_url or recovery_url %}
                 <div class="pf-c-login__main-footer-band">
+                    {% if enroll_url %}
                     <p class="pf-c-login__main-footer-band-item">
                         {% trans 'Need an account?' %}
                         <a href="{{ enroll_url }}">{% trans 'Sign up.' %}</a>
                     </p>
+                    {% endif %}
+                    {% if recovery_url %}
+                    <p class="pf-c-login__main-footer-band-item">
+                        <a href="{{ recovery_url }}">
+                            {% trans 'Forgot username or password?' %}
+                        </a>
+                    </p>
+                    {% endif %}
                 </div>
                 {% endif %}
             </footer>
diff --git a/passbook/stages/identification/stage.py b/passbook/stages/identification/stage.py
index 81ca912f5..5c1a05e0a 100644
--- a/passbook/stages/identification/stage.py
+++ b/passbook/stages/identification/stage.py
@@ -34,14 +34,19 @@ class IdentificationStageView(FormView, AuthenticationStage):
         return [current_stage.template]
 
     def get_context_data(self, **kwargs):
-        # Check for related enrollment flow, add URL to view
+        # Check for related enrollment and recovery flow, add URL to view
         enrollment_flow = self.executor.flow.related_flow(FlowDesignation.ENROLLMENT)
         if enrollment_flow:
-            url = reverse(
+            kwargs["enroll_url"] = reverse(
                 "passbook_flows:flow-executor",
                 kwargs={"flow_slug": enrollment_flow.slug},
             )
-            kwargs["enroll_url"] = url
+        recovery_flow = self.executor.flow.related_flow(FlowDesignation.RECOVERY)
+        if recovery_flow:
+            kwargs["recovery_url"] = reverse(
+                "passbook_flows:flow-executor",
+                kwargs={"flow_slug": recovery_flow.slug},
+            )
 
         # Check all enabled source, add them if they have a UI Login button.
         kwargs["sources"] = []
diff --git a/passbook/stages/identification/tests.py b/passbook/stages/identification/tests.py
index 4b98a098a..f3389c1ec 100644
--- a/passbook/stages/identification/tests.py
+++ b/passbook/stages/identification/tests.py
@@ -82,7 +82,7 @@ class TestIdentificationStage(TestCase):
         """Test that enrollment flow is linked correctly"""
         flow = Flow.objects.create(
             name="enroll-test",
-            slug="unique-string",
+            slug="unique-enrollment-string",
             designation=FlowDesignation.ENROLLMENT,
         )
 
@@ -93,3 +93,19 @@ class TestIdentificationStage(TestCase):
         )
         self.assertEqual(response.status_code, 200)
         self.assertIn(flow.slug, response.rendered_content)
+
+    def test_recovery_flow(self):
+        """Test that recovery flow is linked correctly"""
+        flow = Flow.objects.create(
+            name="enroll-test",
+            slug="unique-recovery-string",
+            designation=FlowDesignation.RECOVERY,
+        )
+
+        response = self.client.get(
+            reverse(
+                "passbook_flows:flow-executor", kwargs={"flow_slug": self.flow.slug}
+            ),
+        )
+        self.assertEqual(response.status_code, 200)
+        self.assertIn(flow.slug, response.rendered_content)