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_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
