Ejemplos de read_cluster_status en Python

Ejemplo n.º 1

Archivo: test_main.py Proyecto: mikewiebe/kafka-utils

def test_read_cluster_value_exit(mock_get):
    response = mock.Mock(status_code=404, spec=requests.Response)

    request = mock_get.return_value
    request.result.return_value = response

    with pytest.raises(SystemExit):
        p, b = main.read_cluster_status(["host1"], 80, "jolokia")

Ejemplo n.º 2

Archivo: test_main.py Proyecto: mikewiebe/kafka-utils

def test_read_cluster_value_server_down(mock_get):
    request = mock_get.return_value
    request.result.side_effect = RequestException

    p, b = main.read_cluster_status(["host1"], 80, "jolokia")

    assert p == 0  # 0 missing partitions
    assert b == 1  # 1 missing brokers

Ejemplo n.º 3

Archivo: test_main.py Proyecto: Yelp/kafka-utils

def test_read_cluster_value_server_down(mock_get):
    request = mock_get.return_value
    request.result.side_effect = RequestException

    p, b = main.read_cluster_status(["host1"], 80, "jolokia")

    assert p == 0   # 0 missing partitions
    assert b == 1   # 1 missing brokers

Ejemplo n.º 4

Archivo: test_main.py Proyecto: Yelp/kafka-utils

def test_read_cluster_value_exit(mock_get):
    response = mock.Mock(status_code=404, spec=requests.Response)

    request = mock_get.return_value
    request.result.return_value = response

    with pytest.raises(SystemExit):
        p, b = main.read_cluster_status(["host1"], 80, "jolokia")

Ejemplo n.º 5

Archivo: test_main.py Proyecto: mikewiebe/kafka-utils

def test_read_cluster_value_no_key(mock_get):
    response = mock.Mock(status_code=200, spec=requests.Response)
    response.json.return_value = {'wrong_key': 1}

    request = mock_get.return_value
    request.result.return_value = response

    p, b = main.read_cluster_status(["host1"], 80, "jolokia")

    assert p == 0  # 0 missing partitions
    assert b == 1  # 1 missing brokers

Ejemplo n.º 6

Archivo: test_main.py Proyecto: Yelp/kafka-utils

def test_read_cluster_value_no_key(mock_get):
    response = mock.Mock(status_code=200, spec=requests.Response)
    response.json.return_value = {'wrong_key': 1}

    request = mock_get.return_value
    request.result.return_value = response

    p, b = main.read_cluster_status(["host1"], 80, "jolokia")

    assert p == 0   # 0 missing partitions
    assert b == 1   # 1 missing brokers

Ejemplo n.º 7

def test_read_cluster_value_partitions(mock_get):
    response = mock.Mock(status_code=200, spec=requests.Response)
    response.json.return_value = {'value': 1}

    request = mock_get.return_value
    request.result.return_value = response

    p, b = main.read_cluster_status(["host1", "host2", "host3"], 80, "jmxproxy")

    assert p == 3   # 3 missing partitions
    assert b == 0   # 0 missing brokers