make servers/*/api_versions.txt and make servers/*/messages

Author: dpkp

Makefile

@@ -93,6 +93,14 @@ servers/%/kafka-bin: servers/dist/$$(call kafka_artifact_name,$$*) | servers/dis
 	tar xzvf $< -C $@ --strip-components 1
 	if [[ "$*" < "1" ]]; then make servers/patch-libs/$*; fi
 
+servers/%/api_versions.txt: servers/$$*/kafka-bin
+	KAFKA_VERSION=$* python -m test.fixtures get_api_versions >servers/$*/api_versions.txt
+
+servers/%/messages: servers/$$*/kafka-bin
+	cd servers/$*/ && jar xvf kafka-bin/libs/kafka-clients-$*.jar common/message/
+	mv servers/$*/common/message/ servers/$*/messages/
+	rmdir servers/$*/common
+
 servers/patch-libs/%: servers/dist/jakarta.xml.bind-api-2.3.3.jar | servers/$$*/kafka-bin
 	cp $< servers/$*/kafka-bin/libs/
 

test/fixtures.py

@@ -686,3 +686,37 @@ def get_producers(self, cnt, **params):
         params = self._enrich_client_params(params, client_id='producer')
         for client in self._create_many_clients(cnt, KafkaProducer, **params):
             yield client
+
+
+def get_api_versions():
+    import logging
+    logging.basicConfig(level=logging.ERROR)
+
+    from test.fixtures import ZookeeperFixture, KafkaFixture
+    zk = ZookeeperFixture.instance()
+    k = KafkaFixture.instance(0, zk)
+
+    from kafka import KafkaClient
+    client = KafkaClient(bootstrap_servers='localhost:{}'.format(k.port))
+    client.check_version()
+
+    from pprint import pprint
+
+    pprint(client.get_api_versions())
+
+    client.close()
+    k.close()
+    zk.close()
+
+
+if __name__ == '__main__':
+    import sys
+    if len(sys.argv) < 2:
+        print("Commands: get_api_versions")
+        exit(0)
+    cmd = sys.argv[1]
+    if cmd == 'get_api_versions':
+        get_api_versions()
+    else:
+        print("Unknown cmd: %s", cmd)
+        exit(1)