make pinax-referrals tolerant to the ending slash in URL

README.md

@@ -407,6 +407,10 @@ You may need to do this if you use a custom user model and upgrade Django.
 
 ## Change Log
 
+### 4.3.0 (2023-06-07)
+
+* added tolerance to the ending slash in URL
+
 ### 4.2.0 (2023-01-12)
 
 * Added referral model instance parameter to `generate_code` callback (`PINAX_REFERRALS_CODE_GENERATOR_CALLBACK` variable)

pinax/referrals/tests/tests.py

@@ -3,7 +3,7 @@
 from django.contrib.auth import get_user_model
 from django.test import TestCase
 
-from pinax.referrals.models import Referral
+from pinax.referrals.models import Referral, ReferralResponse
 
 
 class Tests(TestCase):
@@ -74,3 +74,24 @@ def test_process_referral_authenticated_next_unsafe(self):
         self.assertEqual(referral.responses.count(), 1)
         referral_response = referral.responses.first()
         self.assertEqual(referral_response.user, referred_user)
+
+    def test_process_referral_slash_both_variants(self):
+        referral = Referral.create(redirect_to="https://example.com/")
+        self.assertFalse(referral.responses.exists())
+        response = self.client.get(referral.url.rstrip("/"))
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(response["Location"], "https://example.com/")
+        self.assertEqual(referral.responses.count(), 1)
+        referral_response = referral.responses.first()
+        self.assertIsNone(referral_response.user)
+        self.assertEqual(response.cookies["pinax-referral"].value, f"{referral.code}:{referral_response.session_key}")
+
+        ReferralResponse.objects.all().delete()
+
+        response = self.client.get(referral.url.rstrip("/") + "/")
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(response["Location"], "https://example.com/")
+        self.assertEqual(referral.responses.count(), 1)
+        referral_response = referral.responses.first()
+        self.assertIsNone(referral_response.user)
+        self.assertEqual(response.cookies["pinax-referral"].value, f"{referral.code}:{referral_response.session_key}")

pinax/referrals/urls.py

@@ -1,10 +1,10 @@
-from django.urls import path
+from django.urls import path, re_path
 
 from .views import create_referral, process_referral
 
 app_name = "pinax_referrals"
 
 urlpatterns = [
     path("", create_referral, name="create_referral"),
-    path("<str:code>", process_referral, name="process_referral")
+    re_path(r"^(?P<code>[\w-]+)/?$", process_referral, name="process_referral"),
 ]