def test_get_snapshot_node_role_dimension(network, roles_snapshot):
bf_set_network(network)
bf_set_snapshot(roles_snapshot)
node_roles = NodeRolesData([NodeRoleDimension('dim1')])
bf_put_node_roles(node_roles)
# should not crash
bf_get_snapshot_node_role_dimension('dim1')
def test_put_node_roles():
try:
network_name = 'n1'
bf_set_network(network_name)
node_roles = NodeRolesData([NodeRoleDimension('dim1')])
bf_put_node_roles(node_roles)
assert bf_get_node_roles() == node_roles
finally:
bf_delete_network(network_name)
def test_get_snapshot_node_roles(network, roles_snapshot):
bf_set_network(network)
bf_set_snapshot(roles_snapshot)
dimension_name = 'dim1'
node_roles = NodeRolesData([NodeRoleDimension(dimension_name)])
bf_put_node_roles(node_roles)
# there should be 1 role dimension
snapshot_node_roles = bf_get_snapshot_node_roles()
assert len(snapshot_node_roles.roleDimensions) == 1
assert snapshot_node_roles.roleDimensions[0].name == dimension_name
def test_get_snapshot_node_role_dimension(network, roles_snapshot):
bf_set_network(network)
bf_set_snapshot(roles_snapshot)
mapping = RoleMapping(name="mapping",
regex="regex",
roleDimensionGroups={"dim1": [1]})
node_roles = NodeRolesData(roleDimensionOrder=["dim1"],
roleMappings=[mapping])
bf_put_node_roles(node_roles)
# should not crash
bf_get_snapshot_node_role_dimension("dim1")
def test_add_node_roles_data():
try:
network_name = "n1"
bf_set_network(network_name)
mappings = [
RoleMapping("mapping1", "(.*)-(.*)", {"type": [1], "index": [2]}, {})
]
roles_data = NodeRolesData(None, ["type", "index"], mappings)
bf_put_node_roles(roles_data)
assert bf_get_node_roles() == roles_data
finally:
bf_delete_network(network_name)
def test_put_node_roles():
try:
network_name = "n1"
bf_set_network(network_name)
mapping = RoleMapping(
name="mapping", regex="(regex)", roleDimensionGroups={"dim1": [1]}
)
node_roles = NodeRolesData(
defaultDimension="dim1", roleDimensionOrder=["dim1"], roleMappings=[mapping]
)
bf_put_node_roles(node_roles)
assert bf_get_node_roles() == node_roles
finally:
bf_delete_network(network_name)
def test_get_snapshot_node_roles(network, roles_snapshot):
bf_set_network(network)
bf_set_snapshot(roles_snapshot)
dimension_name = "dim1"
mapping = RoleMapping(name="mapping",
regex="regex",
roleDimensionGroups={dimension_name: [1]})
node_roles = NodeRolesData(roleDimensionOrder=[dimension_name],
roleMappings=[mapping])
bf_put_node_roles(node_roles)
# there should be 1 role dimension
snapshot_node_roles = bf_get_snapshot_node_roles()
assert len(snapshot_node_roles.roleDimensionOrder) == 1
assert snapshot_node_roles.roleDimensionOrder[0] == dimension_name
