Changing set_workload call to match latest changes in ccm

GregBestland · GregBestland · commit b408c166c93a · 2016-02-10T11:20:57.000-06:00
diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py
@@ -267,8 +267,7 @@ def use_cluster(cluster_name, nodes, ipformat=None, start=True, workloads=[]):
         if start:
             if(len(workloads) > 0):
                 for node in CCM_CLUSTER.nodes.values():
-                    for workload in workloads:
-                        node.set_workload(workload)
+                        node.set_workloads(workloads)
             log.debug("Starting CCM cluster: {0}".format(cluster_name))
             CCM_CLUSTER.start(wait_for_binary_proto=True, wait_other_notice=True, jvm_args=jvm_args)
             #Added to wait for slow nodes to start up
