configure jaas for topics.sh w/ sasl

Author: dpkp

test/fixtures.py

@@ -414,21 +414,26 @@ def _create_zk_chroot(self):
             raise RuntimeError("Failed to create Zookeeper chroot node")
         self.out("Kafka chroot created in Zookeeper!")
 
+    def render_jaas(self):
+        self.jaas_config = self._jaas_config()
+        jaas_conf = self.tmp_dir.join("kafka_server_jaas.conf")
+        jaas_conf_template = self.test_resource("kafka_server_jaas.conf")
+        self.render_template(jaas_conf_template, jaas_conf, vars(self))
+        return jaas_conf.strpath
+
     def start(self):
         # Configure Kafka child process
         properties = self.tmp_dir.join("kafka.properties")
-        jaas_conf = self.tmp_dir.join("kafka_server_jaas.conf")
         properties_template = self.test_resource("kafka.properties")
-        jaas_conf_template = self.test_resource("kafka_server_jaas.conf")
 
         # Consider replacing w/ run_script('kafka-server-start.sh', ...)
         args = self.kafka_run_class_args("kafka.Kafka", properties.strpath)
         env = self.kafka_run_class_env()
         if self.sasl_enabled:
+            jaas_conf = self.render_jaas()
             opts = env.get('KAFKA_OPTS', '').strip()
-            opts += ' -Djava.security.auth.login.config={}'.format(jaas_conf.strpath)
+            opts += ' -Djava.security.auth.login.config={}'.format(jaas_conf)
             env['KAFKA_OPTS'] = opts
-            self.render_template(jaas_conf_template, jaas_conf, vars(self))
 
         timeout = 5
         max_timeout = 120
@@ -631,6 +636,11 @@ def get_topic_names(self):
         cmd = self.run_script('kafka-topics.sh', *args)
         env = self.kafka_run_class_env()
         env.pop('KAFKA_LOG4J_OPTS')
+        if self.sasl_enabled:
+            jaas_conf = self.render_jaas()
+            opts = env.get('KAFKA_OPTS', '').strip()
+            opts += ' -Djava.security.auth.login.config={}'.format(jaas_conf)
+            env['KAFKA_OPTS'] = opts
         proc = subprocess.Popen(cmd, env=env, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         stdout, stderr = proc.communicate()
         if proc.returncode != 0: