Ejemplos de _find_topics_with_wrong_rp en Python

Ejemplo n.º 1

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

def test_find_topics_with_wrong_rp_empty():
    result = _find_topics_with_wrong_rp(
        topics={},
        zk=None,
        default_min_isr=None,
    )

    assert result == []

Ejemplo n.º 2

Archivo: test_replication_factor.py Proyecto: udaysagar2177/kafka-utils

def test_find_topics_with_wrong_rp_empty():
    result = _find_topics_with_wrong_rp(
        topics={},
        zk=None,
        default_min_isr=None,
    )

    assert result == []

Ejemplo n.º 3

Archivo: test_replication_factor.py Proyecto: udaysagar2177/kafka-utils

def test_find_topics_with_wrong_rp_ok(mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=None,
    )

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)
    assert result == []

Ejemplo n.º 4

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

def test_find_topics_with_wrong_rp_without_min_isr_in_zk_use_default(mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=1,
    )

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)
    assert result == []

Ejemplo n.º 5

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

def test_find_topics_with_wrong_rp_returns_all_topics(mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=1,
    )

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)

    def dict_comparator(d):
        return sorted(d.items())

    assert sorted(result, key=dict_comparator) == sorted(TOPICS_WITH_WRONG_RP, key=dict_comparator)

Ejemplo n.º 6

Archivo: test_replication_factor.py Proyecto: udaysagar2177/kafka-utils

def test_find_topics_with_wrong_rp_returns_all_topics(mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=1,
    )

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)

    def dict_comparator(d):
        return sorted(d.items())

    assert sorted(result, key=dict_comparator) == sorted(TOPICS_WITH_WRONG_RP,
                                                         key=dict_comparator)

Ejemplo n.º 7

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

def test_find_topics_with_wrong_rp_not_empty_with_default_min_isr(mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=2,
    )
    topic1 = {
        'replication_factor': 2,
        'min_isr': 2,
        'topic': 'topic_1',
    }

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)
    assert result == [topic1]

Ejemplo n.º 8

Archivo: test_replication_factor.py Proyecto: udaysagar2177/kafka-utils

def test_find_topics_with_wrong_rp_not_empty_with_default_min_isr(
        mock_min_isr, mock_zk):
    result = _find_topics_with_wrong_rp(
        topics=TOPICS_STATE,
        zk=mock_zk,
        default_min_isr=2,
    )
    topic1 = {
        'replication_factor': 2,
        'min_isr': 2,
        'topic': 'topic_1',
    }

    calls = [mock.call(mock_zk, 'topic_0'), mock.call(mock_zk, 'topic_1')]
    mock_min_isr.assert_has_calls(calls, any_order=True)
    assert result == [topic1]