openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Don't replace custom query handler if graph is being used

Browse Source
This commit is contained in:
GregBestland
2016-10-06 16:28:06 -05:00
parent 8289de0b38
commit cf633e8f12
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/integration/__init__.py
View File

@@ -342,8 +342,10 @@ def use_cluster(cluster_name, nodes, ipformat=None, start=True, workloads=[]):
try:
jvm_args = []
# This will enable the Mirroring query handler which will echo our custom payload k,v pairs back
if PROTOCOL_VERSION >= 4:
jvm_args = [" -Dcassandra.custom_query_handler_class=org.apache.cassandra.cql3.CustomPayloadMirroringQueryHandler"]
if 'graph' not in workloads:
if PROTOCOL_VERSION >= 4:
jvm_args = [" -Dcassandra.custom_query_handler_class=org.apache.cassandra.cql3.CustomPayloadMirroringQueryHandler"]
if(len(workloads) > 0):
for node in CCM_CLUSTER.nodes.values():
node.set_workloads(workloads)