Make benchmark setup/teardown use minimal protocol version

aholmberg · aholmberg · commit 37cc21fd063e · 2015-06-17T10:47:22.000-05:00
diff --git a/benchmarks/base.py b/benchmarks/base.py
@@ -61,7 +61,7 @@
 def setup(hosts):
     log.info("Using 'cassandra' package from %s", cassandra.__path__)
 
-    cluster = Cluster(hosts)
+    cluster = Cluster(hosts, protocol_version=1)
     cluster.set_core_connections_per_host(HostDistance.LOCAL, 1)
     try:
         session = cluster.connect()
@@ -89,7 +89,7 @@ def setup(hosts):
 
 
 def teardown(hosts):
-    cluster = Cluster(hosts)
+    cluster = Cluster(hosts, protocol_version=1)
     cluster.set_core_connections_per_host(HostDistance.LOCAL, 1)
     session = cluster.connect()
     session.execute("DROP KEYSPACE " + KEYSPACE)
