kafkatest: use cooperative-sticky for StickyAssignor

edenhill · edenhill · commit 69ec887d49f1 · 2020-10-26T09:16:06.000+01:00
diff --git a/confluent_kafka/kafkatest/verifiable_client.py b/confluent_kafka/kafkatest/verifiable_client.py
@@ -86,10 +86,12 @@ def set_config(conf, args):
             if n == 'partition.assignment.strategy':
                 # Convert Java class name to config value.
                 # "org.apache.kafka.clients.consumer.RangeAssignor" -> "range"
-                conf[n] = re.sub(r'org.apache.kafka.clients.consumer.(\w+)Assignor',
-                                 lambda x: x.group(1).lower(), v)
-            else:
-                conf[n] = v
+                v = re.sub(r'org.apache.kafka.clients.consumer.(\w+)Assignor',
+                           lambda x: x.group(1).lower(), v)
+                if v == 'sticky':
+                    v = 'cooperative-sticky'
+
+            conf[n] = v
 
     @staticmethod
     def read_config_file(path):
