Additional tests.

* Test REFERRAL_AUTO_CREATE set to False. * Test REFERRAL_CASE_SENSITVE set to True and False.
byteweaver · Feb 18, 2015 · 2388fe7 · 2388fe7
1 parent eacd8b0
commit 2388fe7
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/referral/tests/test_middlewares.py b/referral/tests/test_middlewares.py
@@ -20,3 +20,33 @@ def test_process_request_new_ref(self):
         self.request.GET[settings.GET_PARAMETER] = "new_ref"
         self.ref_middleware.process_request(self.request)
         self.assertEqual(Referrer.objects.all()[0].name, "new_ref")
+
+    def test_process_request_new_ref_no_autocreate(self):
+        settings.AUTO_CREATE = False
+        self.request.GET[settings.GET_PARAMETER] = "new_ref"
+        self.ref_middleware.process_request(self.request)
+        self.assertEqual(Referrer.objects.count(), 0)
+        settings.AUTO_CREATE = True
+
+    def test_referral_case_sensitive(self):
+        settings.CASE_SENSITIVE = True
+        self.request.GET[settings.GET_PARAMETER] = "new_ref"
+        self.ref_middleware.process_request(self.request)
+        self.assertEqual(Referrer.objects.count(), 1)
+
+        self.request.GET[settings.GET_PARAMETER] = "NEW_REF"
+        self.ref_middleware.process_request(self.request)
+        self.assertEqual(Referrer.objects.count(), 2)
+
+        self.assertEqual(Referrer.objects.filter(name='new_ref').count(), 1)
+        self.assertEqual(Referrer.objects.filter(name='NEW_REF').count(), 1)
+        settings.CASE_SENSITIVE = False
+
+    def test_referral_case_insensitive(self):
+        self.request.GET[settings.GET_PARAMETER] = "new_ref"
+        self.ref_middleware.process_request(self.request)
+        self.assertEqual(Referrer.objects.count(), 1)
+
+        self.request.GET[settings.GET_PARAMETER] = "NEW_REF"
+        self.ref_middleware.process_request(self.request)
+        self.assertEqual(Referrer.objects.count(), 1)