def test_merge_offsets_metadata_multiple(self): zk_offsets = { 'topic1': { 0: 6 }, } kafka_offsets = { 'topic1': { 0: 5 }, 'topic2': { 0: 15 }, } expected = { 'topic1': { 0: 6 }, 'topic2': { 0: 15 }, } topics = ['topic1', 'topic2'] result = merge_offsets_metadata(topics, zk_offsets, kafka_offsets) assert result == expected
def test_merge_offsets_metadata_empty(self): zk_offsets = {} kafka_offsets = {} expected = {} result = merge_offsets_metadata([], zk_offsets, kafka_offsets) assert result == expected
def test_merge_offsets_metadata(self): zk_offsets = { 'topic1': {0: 6}, } kafka_offsets = { 'topic1': {0: 5}, } expected = { 'topic1': {0: 6}, } topics = ['topic1'] result = merge_offsets_metadata(topics, zk_offsets, kafka_offsets) assert result == expected
def test_merge_offsets_metadata_kafka_only(self): zk_offsets = {} kafka_offsets = { 'topic1': { 0: 5 }, } expected = { 'topic1': { 0: 5 }, } topics = ['topic1'] result = merge_offsets_metadata(topics, zk_offsets, kafka_offsets) assert result == expected