tests: py2/3 unified, using pytest, automated on Travis
diff --git a/tests/test_encoding.py b/tests/test_encoding.py
new file mode 100644
index 0000000..df991a1
--- /dev/null
+++ b/tests/test_encoding.py
@@ -0,0 +1,99 @@
+import httplib2
+import tests
+
+
+def test_gzip_head():
+ # Test that we don't try to decompress a HEAD response
+ http = httplib2.Http()
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'gzip', 'content-length': 42},
+ )
+ with tests.server_const_bytes(response) as uri:
+ response, content = http.request(uri, 'HEAD')
+ assert response.status == 200
+ assert int(response['content-length']) != 0
+ assert content == b''
+
+
+def test_gzip_get():
+ # Test that we support gzip compression
+ http = httplib2.Http()
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'gzip'},
+ body=tests.gzip_compress(b'properly compressed'),
+ )
+ with tests.server_const_bytes(response) as uri:
+ response, content = http.request(uri, 'GET')
+ assert response.status == 200
+ assert 'content-encoding' not in response
+ assert '-content-encoding' in response
+ assert int(response['content-length']) == len(b'properly compressed')
+ assert content == b'properly compressed'
+
+
+def test_gzip_post_response():
+ http = httplib2.Http()
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'gzip'},
+ body=tests.gzip_compress(b'properly compressed'),
+ )
+ with tests.server_const_bytes(response) as uri:
+ response, content = http.request(uri, 'POST', body=b'')
+ assert response.status == 200
+ assert 'content-encoding' not in response
+ assert '-content-encoding' in response
+
+
+def test_gzip_malformed_response():
+ http = httplib2.Http()
+ # Test that we raise a good exception when the gzip fails
+ http.force_exception_to_status_code = False
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'gzip'},
+ body=b'obviously not compressed',
+ )
+ with tests.server_const_bytes(response, request_count=2) as uri:
+ with tests.assert_raises(httplib2.FailedToDecompressContent):
+ http.request(uri, 'GET')
+
+ # Re-run the test with out the exceptions
+ http.force_exception_to_status_code = True
+
+ response, content = http.request(uri, 'GET')
+ assert response.status == 500
+ assert response.reason.startswith('Content purported')
+
+
+def test_deflate_get():
+ # Test that we support deflate compression
+ http = httplib2.Http()
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'deflate'},
+ body=tests.deflate_compress(b'properly compressed'),
+ )
+ with tests.server_const_bytes(response) as uri:
+ response, content = http.request(uri, 'GET')
+ assert response.status == 200
+ assert 'content-encoding' not in response
+ assert int(response['content-length']) == len(b'properly compressed')
+ assert content == b'properly compressed'
+
+
+def test_deflate_malformed_response():
+ # Test that we raise a good exception when the deflate fails
+ http = httplib2.Http()
+ http.force_exception_to_status_code = False
+ response = tests.http_response_bytes(
+ headers={'content-encoding': 'deflate'},
+ body=b'obviously not compressed',
+ )
+ with tests.server_const_bytes(response, request_count=2) as uri:
+ with tests.assert_raises(httplib2.FailedToDecompressContent):
+ http.request(uri, 'GET')
+
+ # Re-run the test with out the exceptions
+ http.force_exception_to_status_code = True
+
+ response, content = http.request(uri, 'GET')
+ assert response.status == 500
+ assert response.reason.startswith('Content purported')