From cf633e8f12123395fc3e772ad08838bd114f6d7e Mon Sep 17 00:00:00 2001 From: GregBestland Date: Thu, 6 Oct 2016 16:28:06 -0500 Subject: [PATCH] Don't replace custom query handler if graph is being used --- tests/integration/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py index 9dbfca3e..8202b6fa 100644 --- a/tests/integration/__init__.py +++ b/tests/integration/__init__.py @@ -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)