def test_find_topics_with_wrong_rp_empty(): result = _find_topics_with_wrong_rp( topics={}, zk=None, default_min_isr=None, ) assert result == []
def test_find_topics_with_wrong_rp_empty(): result = _find_topics_with_wrong_rp( topics={}, zk=None, default_min_isr=None, ) assert result == []
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 == []
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 == []
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)
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)
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]
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]