def test_validate_format_version_absent():
assignment = {
"partitions":
[{"partition": 0, "topic": u't1', "replicas": [0, 1, 2]}]
}
# 'version' key missing: Verify Validation failed
assert _validate_format(assignment) is False
def test_validate_format_partitions_empty():
# Empty partitions-list
assignment = {
"version": 1,
"partitions": []
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_type_replica_brokers():
# replica-brokers are not int
assignment = {
"version": 1,
"partitions": [{"partition": 0, "topic": u't1', "replicas": ['0', 1]}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_replica_empty():
# replica-brokers are empty
assignment = {
"version": 1,
"partitions": [{"partition": 0, "topic": u't1', "replicas": []}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_type_replicas():
# replicas is not list
assignment = {
"version": 1,
"partitions": [{"partition": 0, "topic": u't1', "replicas": 0}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_type_topic():
# topic-id is not string
assignment = {
"version": 1,
"partitions": [{"partition": 0, "topic": 2, "replicas": [0, 1, 2]}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_type_partition():
# partition-id is not int
assignment = {
"version": 1,
"partitions": [{"partition": '0', "topic": u't2', "replicas": [0, 1, 2]}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_type_partitions():
# 'partitions' is not list
assignment = {
"version": 1,
"partitions": {"partition": 0, "topic": u't2', "replicas": [0, 1, 2]}
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_version_wrong():
# Invalid version 2
assignment = {
"version": 2,
"partitions":
[{"partition": 0, "topic": u't1', "replicas": [0, 1, 2]}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_invalid_key():
# Invalid key "cluster"
assignment = {
"cluster": "invalid_key",
"partitions":
[{"partition": 0, "topic": u't1', "replicas": [0, 1, 2]}]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format():
assignment = {
"version": 1,
"partitions": [
{"partition": 0, "topic": u't1', "replicas": [0, 1]},
{"partition": 1, "topic": u't1', "replicas": [1, 2]},
{"partition": 0, "topic": u't2', "replicas": [1, 2]}
]
}
# Verify correct format
assert _validate_format(assignment) is True
def test_validate_format_invalid_key_partition():
# Invalid/unwanted key 'isr'
assignment = {
"version": 1,
"partitions":
[
{"topic": u't1', "replicas": [0, 1, 2]},
{"isr": [0, 1, 2], "partition": 0, "topic": u't2', "replicas": [0, 1, 2]}
]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_missing_key_replicas():
# 'replicas' key missing in partitions
assignment = {
"version": 1,
"partitions":
[
{"partition": 0, "topic": u't1'},
{"partition": 0, "topic": u't2', "replicas": [0, 1, 2]}
]
}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_empty():
# Empty assignment
assignment = {}
# Verify validation failed
assert _validate_format(assignment) is False
def test_validate_format_partitions_absent():
assignment = {"version": 1}
# 'partitions' key missing: Verify Validation failed
assert _validate_format(assignment) is False
