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'