Ticket #20099: 20099-test2.diff

tests/middleware/tests.py

@@ -320 +320 @@
         BrokenLinkEmailsMiddleware().process_response(self.req, self.resp)
         self.assertEqual(len(mail.outbox), 0)
 
+    def test_custom_is_ignorable_404(self):
+        class SubclassedMiddleware(BrokenLinkEmailsMiddleware):
+            """Ignoring based on UA only"""
+            ua_patterns = (re.compile(r'Spider.*'), re.compile(r'Robot.*'))
+            def is_ignorable_404(self, uri, user_agent):
+                return any(pattern.search(user_agent) for pattern in self.ua_patterns)
+
+        self.req.META['HTTP_REFERER'] = '/another/url/'
+        self.req.META['HTTP_USER_AGENT'] = 'Spider machine 3.4'
+        SubclassedMiddleware().process_response(self.req, self.resp)
+        self.assertEqual(len(mail.outbox), 0)
+        self.req.META['HTTP_USER_AGENT'] = 'My user agent'
+        SubclassedMiddleware().process_response(self.req, self.resp)
+        self.assertEqual(len(mail.outbox), 1)
+
 
 class ConditionalGetMiddlewareTest(TestCase):
     urls = 'middleware.cond_get_urls'