add failing heuristic test for 301 and 304 response

Author: eads

tests/conftest.py

@@ -59,6 +59,9 @@ def update_etag(self, env, start_response):
         start_response('200 OK', headers)
         return [pformat(env).encode("utf8")]
 
+    def conditional_get(self, env, start_response):
+        return start_response('304 Not Modified', [])
+
     def etag(self, env, start_response):
         headers = [
             ('Etag', self.etag_string),

tests/test_expires_heuristics.py

@@ -36,6 +36,26 @@ def test_no_header_change_means_no_warning_header(self, url):
         assert not self.heuristic.warning.called
 
 
+class TestHeuristicWith3xxResponse(object):
+    def setup(self):
+
+        class DummyHeuristic(BaseHeuristic):
+            def update_headers(self, resp):
+                return {'x-dummy-header': 'foobar'}
+
+        self.sess = CacheControl(Session(), heuristic=DummyHeuristic())
+
+    def test_heuristic_applies_to_301(self, url):
+        the_url = url + 'permanent_redirect'
+        resp = self.sess.get(the_url)
+        assert 'x-dummy-header' in resp.headers
+
+    def test_heuristic_applies_to_304(self, url):
+        the_url = url + 'conditional_get'
+        resp = self.sess.get(the_url)
+        assert 'x-dummy-header' in resp.headers
+
+
 class TestUseExpiresHeuristic(object):
 
     def test_expires_heuristic_arg(self):