def test_invalid_missing_replication(self):
testing_config = {
'test1': {
'PartitionCount': 1,
'Configs': {'test_config': 589,
'test_config': 581},
'hosts': ['1']
},
'test2': {'PartitionCount': 1,
'Configs': {}}
}
with pytest.raises(kafka_topic_enforcer.ConfigValidationError):
kafka_topic_enforcer.validate_config(testing_config)
def test_valid_config(self):
testing_config = {
'test1': {
'PartitionCount': 1,
'ReplicationFactor': 1,
'Configs': {'test_config': 589,
'test_config': 581},
'hosts': ['1']
},
'test2':
{'PartitionCount': 1,
'ReplicationFactor': 2,
'Configs': {}}
}
kafka_topic_enforcer.validate_config(testing_config)
def test_invalid_number_of_hosts(self):
testing_config = {
'test1': {
'PartitionCount': 1,
'ReplicationFactor': 1,
'Configs': {'test_config': 589,
'test_config': 581},
'hosts': ['1', '2']
},
'test2':
{'PartitionCount': 1,
'ReplicationFactor': 2,
'Configs': {}}
}
with pytest.raises(kafka_topic_enforcer.ConfigValidationError):
kafka_topic_enforcer.validate_config(testing_config)
