Using Role url reversal to support all custom lookup_fields (#86)

src/neapolitan/views.py

@@ -296,11 +296,9 @@ def get_success_url(self):
             % self.__class__.__name__
         )
         if self.role == Role.DELETE:
-            success_url = reverse(f"{self.url_base}-list")
+            success_url = Role.LIST.reverse(self)
         else:
-            success_url = reverse(
-                f"{self.url_base}-detail", kwargs={"pk": self.object.pk}
-            )
+            success_url = Role.DETAIL.reverse(self, self.object)
         return success_url
 
     # Pagination and filtering

tests/tests.py

@@ -1,4 +1,5 @@
 import os
+import uuid
 
 from django.core.management import call_command
 from django.http import HttpResponse
@@ -168,6 +169,21 @@ def test_custom_lookup_field(self):
         self.assertEqual(response.status_code, 200)
         self.assertContains(response, self.main_collection.name)
 
+    def test_custom_lookup_field_create(self):
+        create_url = reverse("named_collections-create")
+
+        # Submit the form.
+        response = self.client.post(
+            create_url,
+            {
+                "code": uuid.uuid4(),
+                "name": "The Carlton Collection",
+            },
+            follow=True,
+        )
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.resolver_match.url_name, "named_collections-detail")
+
     def test_lookup_url_converter(self):
         """Test view.lookup_url_converter"""
         response = self.client.get(f"/named_collections/{self.main_collection.id}/")