def test_get_node_roles(): try: network_name = 'n1' bf_set_network(network_name) assert bf_get_node_roles() == NodeRolesData([]) finally: bf_delete_network(network_name)
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_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)