def test_compress_streaming_response(self):
"""
Compression is performed on responses with streaming content.
"""
r = CompressionMiddleware().process_response(self.req,
self.stream_resp)
self.assertEqual(brotli.decompress(b''.join(r)),
b''.join(self.sequence))
self.assertEqual(r.get('Content-Encoding'), 'br')
self.assertFalse(r.has_header('Content-Length'))
self.assertEqual(r.get('Vary'), 'Accept-Encoding')
def test_middleware_compress_response(self):
fake_request = FakeRequestAcceptsBrotli()
response_content = UTF8_LOREM_IPSUM_IN_CZECH
fake_response = FakeResponse(content=response_content)
compression_middleware = CompressionMiddleware()
response = compression_middleware.process_response(
fake_request, fake_response)
decompressed_response = brotli.decompress(
response.content) # type: bytes
self.assertEqual(response_content,
decompressed_response.decode(encoding='utf-8'))
self.assertEqual(response.get('Vary'), 'Accept-Encoding')
def test_middleware_wont_compress_response_if_response_is_small(self):
fake_request = FakeRequestAcceptsBrotli()
response_content = "Hello World"
self.assertLess(len(response_content), 200) # a < b
fake_response = FakeResponse(content=response_content)
compression_middleware = CompressionMiddleware()
response = compression_middleware.process_response(
fake_request, fake_response)
self.assertEqual(response_content,
response.content.decode(encoding='utf-8'))
self.assertFalse(response.has_header('Vary'))
def test_middleware_wont_compress_if_client_not_accept(self):
fake_request = FakeLegacyRequest()
response_content = UTF8_LOREM_IPSUM_IN_CZECH
fake_response = FakeResponse(content=response_content)
compression_middleware = CompressionMiddleware()
response = compression_middleware.process_response(
fake_request, fake_response)
django_gzip_middleware = GZipMiddleware()
gzip_response = django_gzip_middleware.process_response(
fake_request, fake_response)
self.assertEqual(response_content,
response.content.decode(encoding='utf-8'))
self.assertEqual(response.get('Vary'), 'Accept-Encoding')
def test_middleware_wont_compress_if_response_is_already_compressed(self):
fake_request = FakeRequestAcceptsBrotli()
response_content = UTF8_LOREM_IPSUM_IN_CZECH
fake_response = FakeResponse(content=response_content)
compression_middleware = CompressionMiddleware()
django_gzip_middleware = GZipMiddleware()
gzip_response = django_gzip_middleware.process_response(
fake_request, fake_response)
response = compression_middleware.process_response(
fake_request, gzip_response)
self.assertEqual(
response_content,
gzip_decompress(response.content).decode(encoding='utf-8'))
self.assertEqual(response.get('Vary'), 'Accept-Encoding')
def test_etag_is_updated_if_present(self):
fake_request = FakeRequestAcceptsBrotli()
response_content = UTF8_LOREM_IPSUM_IN_CZECH * 5
fake_etag_content = "\"foo\""
fake_response = FakeResponse(content=response_content,
headers={"ETag": fake_etag_content})
self.assertEqual(fake_response['ETag'], fake_etag_content)
compression_middleware = CompressionMiddleware()
response = compression_middleware.process_response(
fake_request, fake_response)
decompressed_response = brotli.decompress(
response.content) # type: bytes
self.assertEqual(response_content,
decompressed_response.decode(encoding='utf-8'))
# note: this is where we differ from django-brotli
# django-brotli's expectation:
### self.assertEqual(response['ETag'], '"foo;br\\"')
# Django's expectation:
self.assertEqual(response['ETag'], 'W/"foo"')