Added protocol version to test_session_host_parameter

bjmb · bjmb · commit a16f5f7b8553 · 2017-03-23T09:23:52.000-04:00
diff --git a/tests/integration/standard/test_cluster.py b/tests/integration/standard/test_cluster.py
@@ -193,10 +193,10 @@ def test_session_host_parameter(self):
         @test_category connection
         """
         with self.assertRaises(NoHostAvailable):
-            Session(Cluster(), [])
+            Session(Cluster(protocol_version=PROTOCOL_VERSION), [])
         with self.assertRaises(NoHostAvailable):
-            Session(Cluster(), [Host("1.2.3.4", SimpleConvictionPolicy)])
-        session = Session(Cluster(), [Host(x, SimpleConvictionPolicy) for x in
+            Session(Cluster(protocol_version=PROTOCOL_VERSION), [Host("1.2.3.4", SimpleConvictionPolicy)])
+        session = Session(Cluster(protocol_version=PROTOCOL_VERSION), [Host(x, SimpleConvictionPolicy) for x in
                                       ("127.0.0.1", "127.0.0.2", "1.2.3.4")])
         session.shutdown()
 
