Python bf_get_node_roles Examples

Example #1

File: test_network_funcs.py Project: mirzawaqasahmed/pybatfish

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)

Example #2

File: test_network_funcs.py Project: mirzawaqasahmed/pybatfish

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)

Example #3

File: test_network_funcs.py Project: batfish/pybatfish

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)

Example #4

File: test_network_funcs.py Project: batfish/pybatfish

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)