def test_validate_plan_duplicate_replica_brokers():
# Duplicate replica-brokers [0, 0, 1]
assignment = {
"version": 1,
"partitions": [{"partition": 0, "topic": u't1', "replicas": [0, 0, 1]}]
}
# Verify validation failed
assert _validate_plan(assignment) is False
def test_validate_plan():
assignment = {
"version": 1,
"partitions": [
{"partition": 0, "topic": u't1', "replicas": [0, 1]},
{"partition": 1, "topic": u't1', "replicas": [1, 2]},
]
}
# Verify valid assignment
assert _validate_plan(assignment) is True
def test_validate_plan_duplicate_partitions():
# Duplicate partition (t1, 0)
assignment = {
"version": 1,
"partitions": [
{"partition": 0, "topic": u't1', "replicas": [0, 1]},
{"partition": 0, "topic": u't1', "replicas": [1, 2]},
]
}
# Verify validation failed
assert _validate_plan(assignment) is False
def test_validate_plan_different_replication_factor():
# Replication-factor (t1, 0): 2
# Replication-factor (t1, 1): 3
assignment = {
"version": 1,
"partitions": [
{"partition": 0, "topic": u't1', "replicas": [0, 2, 1]},
{"partition": 1, "topic": u't1', "replicas": [0, 1]}
]
}
# Verify validation failed
assert _validate_plan(assignment) is False
