Fix test_use_requested_compression() to use Protocol version 4

Per review comments, reverted the changes to test, but changed it to explicitly use v4 of the protocol.

Author: mykaul

tests/unit/test_connection.py

@@ -33,8 +33,8 @@
 
 class ConnectionTest(unittest.TestCase):
 
-    def make_connection(self):
-        c = Connection(DefaultEndPoint('1.2.3.4'))
+    def make_connection(self, **kwargs):
+        c = Connection(DefaultEndPoint('1.2.3.4'), **kwargs)
         c._socket = Mock()
         c._socket.send.side_effect = lambda x: len(x)
         return c
@@ -186,17 +186,18 @@ def test_requested_compression_not_available(self, *args):
         assert isinstance(args[0], ProtocolError)
 
     def test_use_requested_compression(self, *args):
-        c = self.make_connection()
+        c = self.make_connection(protocol_version=4)
         c._requests = {0: (c._handle_options_response, ProtocolHandler.decode_message, [])}
         c.defunct = Mock()
-        # request LZ4 compression
-        c.compression = "lz4"
+        # request snappy compression
+        c.compression = "snappy"
 
         locally_supported_compressions.pop('lz4', None)
         locally_supported_compressions.pop('snappy', None)
         locally_supported_compressions['lz4'] = ('lz4compress', 'lz4decompress')
         locally_supported_compressions['snappy'] = ('snappycompress', 'snappydecompress')
 
+        # the server only supports snappy
         options_buf = BytesIO()
         write_stringmultimap(options_buf, {
             'CQL_VERSION': ['3.0.3'],
@@ -206,8 +207,7 @@ def test_use_requested_compression(self, *args):
 
         c.process_msg(_Frame(version=4, flags=0, stream=0, opcode=SupportedMessage.opcode, body_offset=9, end_pos=9 + len(options)), options)
 
-
-        assert c.decompressor == locally_supported_compressions['lz4'][1]
+        assert c.decompressor == locally_supported_compressions['snappy'][1]
 
     def test_disable_compression(self, *args):
         c = self.make_connection()