def _inner(*args, **kwargs):
"""
Alters the response.
"""
response = func(*args, **kwargs)
remove_headers_from_response(response, *headers)
return response
def test_removes_intended_headers(self):
fake_request = HttpRequest()
fake_response = HttpResponse()
fake_response['Vary'] = 'Cookie'
fake_response['Accept-Encoding'] = 'gzip'
remove_headers_from_response(fake_response, 'Vary')
result = self.middleware.process_response(fake_request, fake_response)
self.assertNotIn('Vary', result)
self.assertEquals('gzip', result['Accept-Encoding'])
def test_doesnt_barf_removing_nonexistent_headers(self):
fake_request = HttpRequest()
fake_response = HttpResponse()
fake_response['Vary'] = 'Cookie'
fake_response['Accept-Encoding'] = 'gzip'
remove_headers_from_response(fake_response, 'Vary', 'FakeHeaderWeeee')
result = self.middleware.process_response(fake_request, fake_response)
self.assertNotIn('Vary', result)
self.assertEquals('gzip', result['Accept-Encoding'])
def test_removes_intended_headers(self):
fake_request = HttpRequest()
fake_response = HttpResponse()
fake_response['Vary'] = 'Cookie'
fake_response['Accept-Encoding'] = 'gzip'
remove_headers_from_response(fake_response, 'Vary')
result = self.middleware.process_response(fake_request, fake_response)
assert 'Vary' not in result
assert 'gzip' == result['Accept-Encoding']
def test_removes_intended_headers(self):
fake_request = HttpRequest()
fake_response = HttpResponse()
fake_response["Vary"] = "Cookie"
fake_response["Accept-Encoding"] = "gzip"
remove_headers_from_response(fake_response, "Vary")
result = self.middleware.process_response(fake_request, fake_response)
self.assertNotIn("Vary", result)
self.assertEquals("gzip", result["Accept-Encoding"])
def test_doesnt_barf_removing_nonexistent_headers(self):
fake_request = HttpRequest()
fake_response = HttpResponse()
fake_response["Vary"] = "Cookie"
fake_response["Accept-Encoding"] = "gzip"
remove_headers_from_response(fake_response, "Vary", "FakeHeaderWeeee")
result = self.middleware.process_response(fake_request, fake_response)
self.assertNotIn("Vary", result)
self.assertEquals("gzip", result["Accept-Encoding"])
