def add_brokers_from_zk(cluster, zk):
for b in zk.get_children("/brokers/ids"):
broker_data, bstat = zk.get("/brokers/ids/{0}".format(b))
cluster.add_broker(Broker.create_from_json(int(b), broker_data))
if cluster.num_brokers() == 0:
raise ZookeeperException(
"The cluster specified does not have any brokers")
def test_broker_create_from_json_extended(self):
jsonstr = '{"jmx_port":-1,"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092"],"host":"10.0.0.10","version":3,"port":9092}'
broker2 = Broker.create_from_json(1, jsonstr)
assert broker2.jmx_port == -1
assert broker2.timestamp == "1466985807242"
assert broker2.endpoints == ["PLAINTEXT://10.0.0.10:9092"]
assert broker2.version == 3
assert broker2.port == 9092
def test_broker_get_endpoint(self):
jsonstr = ('{"jmx_port":-1,"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092", "SSL://10.0.0.10:2834"],'
'"host":"10.0.0.10","version":3,"port":9092}')
broker = Broker.create_from_json(1, jsonstr)
endpoint = broker.get_endpoint("SSL")
assert endpoint.protocol == "SSL"
assert endpoint.hostname == "10.0.0.10"
assert endpoint.port == 2834
def test_broker_get_endpoint(self):
jsonstr = (
'{"jmx_port":-1,"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092", "SSL://10.0.0.10:2834"],'
'"host":"10.0.0.10","version":3,"port":9092}')
broker = Broker.create_from_json(1, jsonstr)
endpoint = broker.get_endpoint("SSL")
assert endpoint.protocol == "SSL"
assert endpoint.hostname == "10.0.0.10"
assert endpoint.port == 2834
def test_broker_create_from_json_extended(self):
jsonstr = '{"jmx_port":-1,"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092"],"host":"10.0.0.10","version":3,"port":9092}'
broker2 = Broker.create_from_json(1, jsonstr)
assert broker2.jmx_port == -1
assert broker2.timestamp == "1466985807242"
assert len(broker2.endpoints) == 1
assert broker2.endpoints[0].protocol == 'PLAINTEXT'
assert broker2.endpoints[0].hostname == '10.0.0.10'
assert broker2.endpoints[0].port == 9092
assert broker2.version == 3
assert broker2.port == 9092
def test_broker_create_from_json_bad_jmx_port(self):
jsonstr = '{"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092"],"host":"10.0.0.10","version":3,"port":9092}'
broker2 = Broker.create_from_json(1, jsonstr)
assert broker2.hostname == '10.0.0.10'
def test_broker_create_from_json_bad_jmx_port(self):
jsonstr = '{"timestamp":"1466985807242","endpoints":["PLAINTEXT://10.0.0.10:9092"],"host":"10.0.0.10","version":3,"port":9092}'
broker2 = Broker.create_from_json(1, jsonstr)
assert broker2.hostname == '10.0.0.10'
