def test_noderolesdata():
"""Check proper deserialization for a node roles data."""
dict = {
"roleDimensions": [
{
"name":
"dim1",
"type":
"CUSTOM",
"roles": [
{
"name": "role1",
"regex": "regex",
},
{
"name": "role2",
"regex": "regex",
},
]
},
]
}
nodeRoleData = NodeRolesData.from_dict(dict)
assert len(nodeRoleData.roleDimensions) == 1
assert len(nodeRoleData.roleDimensions[0].roles) == 2
assert nodeRoleData.roleDimensions[0].roles[0].name == "role1"
def test_noderolesdata():
"""Check proper deserialization for a node roles data."""
dict = {"roleDimensionOrder": ["dim1", "dim2"], "roleMappings": []}
nodeRoleData = NodeRolesData.from_dict(dict)
assert nodeRoleData.defaultDimension is None
assert nodeRoleData.roleDimensionOrder == ["dim1", "dim2"]
assert nodeRoleData.roleMappings == []
def test_noderolesdata():
"""Check proper deserialization for a node roles data."""
dict = {
"roleDimensions": [
{
"name":
"dim1",
"type":
"CUSTOM",
"roles": [
{
"name": "role1",
"regex": "regex",
},
{
"name": "role2",
"regex": "regex",
},
],
"roleDimensionMappings": [{
"regex": "regex",
}, {
"regex": "regex2",
"groups": [],
"canonicalRoleNames": {},
"caseSensitive": True
}]
},
]
}
nodeRoleData = NodeRolesData.from_dict(dict)
assert len(nodeRoleData.roleDimensions) == 1
assert len(nodeRoleData.roleDimensions[0].roles) == 2
assert nodeRoleData.roleDimensions[0].roles[0].name == "role1"
assert len(nodeRoleData.roleDimensions[0].roleDimensionMappings) == 2
assert nodeRoleData.roleDimensions[0].roleDimensionMappings[
0].regex == "regex"
assert nodeRoleData.roleDimensions[0].roleDimensionMappings[0].groups == [
1
]
assert nodeRoleData.roleDimensions[0].roleDimensionMappings[
0].canonicalRoleNames == {}
assert nodeRoleData.roleDimensions[0].roleDimensionMappings[
0].caseSensitive is False
def bf_get_snapshot_node_roles():
# type: () -> NodeRolesData
"""Returns the definitions and assignments of node roles for the active network and snapshot."""
return NodeRolesData.from_dict(
restv2helper.get_snapshot_node_roles(bf_session))
def bf_get_snapshot_inferred_node_roles():
# type: () -> NodeRolesData
"""Gets suggested definitions and hypothetical assignments of node roles for the active network and snapshot."""
return NodeRolesData.from_dict(
restv2helper.get_snapshot_inferred_node_roles(bf_session))
