Exemplo n.º 1
0
def test_txtmeta_parse():
    """
    Test the txtmeta parsing feature of the TopologyManager object.
    """
    topology = TopologyManager(engine='debug')
    dictmeta = topology.parse(TOPOLOGY, load=False)

    expected = {
        'nodes': [{
            'attributes': OrderedDict([('shell', 'vtysh')]),
            'nodes': ['sw1', 'sw2']
        }, {
            'attributes': OrderedDict([('type', 'host')]),
            'nodes': ['hs1']
        }, {
            'attributes': OrderedDict(),
            'nodes': ['hs2']
        }],
        'ports': [],
        'links': [{
            'attributes': OrderedDict(),
            'endpoints': (('sw1', '1'), ('hs1', '1'))
        }, {
            'attributes': OrderedDict(),
            'endpoints': (('sw1', 'a'), ('hs1', 'a'))
        }, {
            'attributes': OrderedDict([('attr1', 1)]),
            'endpoints': (('sw1', '4'), ('hs2', 'a'))
        }]
    }

    ddiff = DeepDiff(dictmeta, expected)
    assert not ddiff
Exemplo n.º 2
0
def test_build():
    """
    Test building and unbuilding a topology using the Mininet engine.
    """
    topology = TopologyManager(engine='debug')

    # Create topology using the NMLManager
    sw1 = topology.nml.create_node(identifier='sw1', name='My Switch 1')
    hs1 = topology.nml.create_node(identifier='hs1',
                                   name='My Host 1',
                                   type='host')

    sw1p1 = topology.nml.create_biport(sw1)
    sw1p2 = topology.nml.create_biport(sw1)
    sw1p3 = topology.nml.create_biport(sw1)  # noqa
    hs1p1 = topology.nml.create_biport(hs1)
    hs1p2 = topology.nml.create_biport(hs1)
    hs1p3 = topology.nml.create_biport(hs1)  # noqa

    sw1p1_hs1p1 = topology.nml.create_bilink(sw1p1, hs1p1)  # noqa
    sw1p2_hs1p2 = topology.nml.create_bilink(sw1p2, hs1p2)  # noqa

    # Build topology
    topology.build()

    # Get an engine node
    assert topology.get('sw1') is not None
    assert topology.get('hs1') is not None

    # Unbuild topology
    topology.unbuild()
def test_txtmeta_parse():
    """
    Test the txtmeta parsing feature of the TopologyManager object.
    """
    topology = TopologyManager(engine='debug')
    dictmeta = topology.parse(TOPOLOGY, load=False)

    expected = {
        'nodes': [
            {
                'attributes': OrderedDict([('shell', 'vtysh')]),
                'nodes': ['sw1', 'sw2']
            },
            {
                'attributes': OrderedDict([('type', 'host')]),
                'nodes': ['hs1']
            },
            {
                'attributes': OrderedDict(),
                'nodes': ['hs2']
            }
        ],
        'ports': [],
        'links': [
            {
                'attributes': OrderedDict(),
                'endpoints': (('sw1', '1'), ('hs1', '1'))
            },
            {
                'attributes': OrderedDict(),
                'endpoints': (('sw1', 'a'), ('hs1', 'a'))
            },
            {
                'attributes': OrderedDict([('attr1', 1)]),
                'endpoints': (('sw1', '4'), ('hs2', 'a'))
            }
        ]
    }

    ddiff = DeepDiff(dictmeta, expected)
    assert not ddiff
def test_build():
    """
    Test building and unbuilding a topology using the Mininet engine.
    """
    topology = TopologyManager(engine='debug')

    # Create topology using the NMLManager
    sw1 = topology.nml.create_node(identifier='sw1', name='My Switch 1')
    hs1 = topology.nml.create_node(identifier='hs1', name='My Host 1',
                                   type='host')

    sw1p1 = topology.nml.create_biport(sw1)
    sw1p2 = topology.nml.create_biport(sw1)
    sw1p3 = topology.nml.create_biport(sw1)  # noqa
    hs1p1 = topology.nml.create_biport(hs1)
    hs1p2 = topology.nml.create_biport(hs1)
    hs1p3 = topology.nml.create_biport(hs1)  # noqa

    sw1p1_hs1p1 = topology.nml.create_bilink(sw1p1, hs1p1)  # noqa
    sw1p2_hs1p2 = topology.nml.create_bilink(sw1p2, hs1p2)  # noqa

    # Build topology
    topology.build()

    # Get an engine node
    assert topology.get('sw1') is not None
    assert topology.get('hs1') is not None

    # Unbuild topology
    topology.unbuild()
Exemplo n.º 5
0
def test_autoport():
    """
    Test the autoport feature.
    """
    topodesc = """
        [port_number=5] hs1:oobm
        hs1:a -- hs2:x
        hs1:2 -- hs2:2
        hs1:4 -- hs2:4
        hs1:b -- hs2:y
    """

    topology = TopologyManager(engine='debug')
    topology.parse(topodesc)
    topology.build()

    assert topology.get('hs1') is not None
    assert topology.get('hs2') is not None

    ports = {k: dict(v) for k, v in topology.ports.items()}
    expected = {
        'hs1': {
            'oobm': 'oobm',
            'a': 'a',
            '2': '2',
            '4': '4',
            'b': 'b',
        },
        'hs2': {
            'x': 'x',
            '2': '2',
            '4': '4',
            'y': 'y',
        }
    }

    topology.unbuild()

    ddiff = DeepDiff(ports, expected)
    assert not ddiff
def test_autoport():
    """
    Test the autoport feature.
    """
    topodesc = """
        [port_number=5] hs1:oobm
        hs1:a -- hs2:x
        hs1:2 -- hs2:2
        hs1:4 -- hs2:4
        hs1:b -- hs2:y
    """

    topology = TopologyManager(engine='debug')
    topology.parse(topodesc)
    topology.build()

    assert topology.get('hs1') is not None
    assert topology.get('hs2') is not None

    ports = {k: dict(v) for k, v in topology.ports.items()}
    expected = {
        'hs1': {
            'oobm': 'oobm',
            'a': 'a',
            '2': '2',
            '4': '4',
            'b': 'b',
        },
        'hs2': {
            'x': 'x',
            '2': '2',
            '4': '4',
            'y': 'y',
        }
    }

    topology.unbuild()

    ddiff = DeepDiff(ports, expected)
    assert not ddiff
Exemplo n.º 7
0
def test_build():
    """
    Test building and unbuilding a topology using the Mininet engine.
    """
    # Create a graph
    graph = TopologyGraph()

    # Create some nodes
    graph.create_node('sw1', name='My Switch 1')
    graph.create_node('hs1', name='My Host 1', type='host')

    # Create some ports
    graph.create_port('p1', 'sw1')
    graph.create_port('p2', 'sw1')
    graph.create_port('p3', 'sw1')

    graph.create_port('p1', 'hs1')
    graph.create_port('p2', 'hs1')
    graph.create_port('p3', 'hs1')

    # Create the links
    graph.create_link('sw1', 'p1', 'hs1', 'p1')
    graph.create_link('sw1', 'p2', 'hs1', 'p2')

    # Check consistency of the graph
    graph.check_consistency()

    # Create the topology and set the graph
    topology = TopologyManager(engine='debug')
    topology.graph = graph

    # Build the topology
    topology.build()

    # Get an engine node
    assert topology.get('sw1') is not None
    assert topology.get('hs1') is not None

    # Unbuild topology
    topology.unbuild()
def test_attribute_injection(tmpdir):
    """
    Test the configuration file is parsed correctly.
    """
    workdir = str(tmpdir)
    search_path = str(tmpdir.mkdir('test'))

    try:
        # Write matching topologies
        for count, content in enumerate(TOPOLOGY_MATCHES):
            output_filename = join(
                search_path, 'test_topology_match_{}.py'.format(count)
            )
            with open(output_filename, 'w') as fd:
                fd.write('TOPOLOGY = """\n')
                fd.write(content)
                fd.write('"""')

        # Write the attributes injection file
        injection_path = join(workdir, 'attributes_injection.json')
        with open(injection_path, 'w') as fd:
            fd.write(INJECTION_FILE.format(search_path=search_path))

        # Change keys of the expected parsed injection file
        expected = OrderedDict()
        for key, value in EXPECTED_PARSED_INJECTION_FILE.items():
            expected[key.format(search_path=search_path)] = value

        # Actually parse the injection file
        actual = parse_attribute_injection(
            injection_path, search_path=search_path
        )

        # Compare the actual and the expected
        differences = DeepDiff(actual, expected)
        # from pprint import pprint
        # print('--------------------')
        # pprint(actual)
        # print('--------------------')
        # pprint(expected)
        # print('--------------------')
        assert not differences

        # Test building with injected attributes
        injected_attributes = actual[
            '{search_path}/test_topology_match_0.py'.format(
                search_path=search_path
            )
        ]

        topology = TopologyManager()
        topology.parse(
            TOPOLOGY_TO_BUILD,
            inject=injected_attributes
        )
        topology.build()

        sw9 = topology.get('sw9')
        sw2 = topology.get('sw2')
        hs3 = topology.get('hs3')
        hs2 = topology.get('hs2')

        assert sw9.metadata == {
            'shell': 'vtysh',
            'name': 'Switch 9'
        }
        # [shell=vtysh name="new_name" image=image_for_sw2] sw2
        assert sw2.metadata == {
            'image': 'image_for_sw2',
            'shell': 'vtysh',
            'name': 'new_name'
        }
        assert hs3.metadata == {
            'type': 'host',
            'name': 'Host 3'
        }
        assert hs2.metadata == {
            'type': 'host',
            'name': 'Host 2',
            'image': 'image_for_all_hosts',
            'hardware': 'hardware_for_sw1_sw3_hs1_hs2'
        }

    finally:
        try:
            rmtree(workdir)
        except:
            pass
def test_unlink_relink():
    """
    Test the unlink and relink calls.

    Creates a topology with two hosts with a port each one and a named link
    between them. During execution the link gets down and up again and the
    connection is asserted in all stages.
    """

    topology = "[identifier=thelink] hs1:a -- hs2:b"

    mgr = TopologyManager(engine='docker')
    mgr.parse(topology)
    mgr.build()

    hs1 = mgr.get('hs1')
    hs2 = mgr.get('hs2')

    try:

        assert hs1 is not None
        assert hs2 is not None

        # Configure IPs
        hs1('ip link set dev a up')
        hs1('ip addr add 10.0.15.1/24 dev a')
        hs2('ip link set dev b up')
        hs2('ip addr add 10.0.15.2/24 dev b')

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2')
        assert '1 packets transmitted, 1 received' in ping_result

        # Unlink
        mgr.unlink('thelink')

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2')
        assert 'Network is unreachable' in ping_result

        # Relink
        mgr.relink('thelink')

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2')
        assert '1 packets transmitted, 1 received' in ping_result

    finally:
        mgr.unbuild()
def test_unlink_relink():
    """
    Test the unlink and relink calls.

    Creates a topology with two hosts with a port each one and a named link
    between them. During execution the link gets down and up again and the
    connection is asserted in all stages.
    """

    # Setup which shell to use
    shell = 'bash'

    topology = '[identifier=thelink] hs1:a -- hs2:b'

    mgr = TopologyManager(engine='docker')
    mgr.parse(topology)
    mgr.build()

    hs1 = mgr.get('hs1')
    hs2 = mgr.get('hs2')

    try:

        assert hs1 is not None
        assert hs2 is not None

        # Configure IPs
        hs1('ip link set dev a up', shell=shell)
        hs1('ip addr add 10.0.15.1/24 dev a', shell=shell)
        hs2('ip link set dev b up', shell=shell)
        hs2('ip addr add 10.0.15.2/24 dev b', shell=shell)

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2', shell=shell)
        assert '1 packets transmitted, 1 received' in ping_result

        # Unlink
        mgr.unlink('thelink')

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2', shell=shell)
        assert 'Network is unreachable' in ping_result

        # Relink
        mgr.relink('thelink')

        # Test connection
        ping_result = hs1('ping -c 1 10.0.15.2', shell=shell)
        assert '1 packets transmitted, 1 received' in ping_result

    finally:
        mgr.unbuild()
def test_attribute_injection(tmpdir):
    """
    Test the configuration file is parsed correctly.
    """
    workdir = str(tmpdir)
    search_path = str(tmpdir.mkdir('test'))
    subfolder = str(tmpdir.mkdir('test/subfolder'))

    try:
        # Write matching topologies
        for basepath, matches in ((search_path, TOPOLOGY_MATCHES),
                                  (subfolder, TOPOLOGY_MATCHES_FOLDER)):
            for count, content in matches.items():
                output_filename = join(
                    basepath, 'test_topology_match_{}.py'.format(count))
                with open(output_filename, 'w') as fd:
                    fd.write('TOPOLOGY = """\n')
                    fd.write(content)
                    fd.write('"""')

        # Write the attributes injection file
        injection_path = join(workdir, 'attributes_injection.json')
        with open(injection_path, 'w') as fd:
            fd.write(INJECTION_FILE.format(search_path=search_path))

        # Change keys of the expected parsed injection file
        expected = OrderedDict()
        for key, value in EXPECTED_PARSED_INJECTION_FILE.items():
            expected[key.format(search_path=search_path)] = value

        # Actually parse the injection file
        actual = parse_attribute_injection(injection_path,
                                           search_paths=[search_path])

        # Compare the actual and the expected
        differences = DeepDiff(actual, expected)
        assert not differences

        # Test building with injected attributes
        injected_attributes = actual['{search_path}/test_topology_match_0.py'.
                                     format(search_path=search_path)]

        topology = TopologyManager()
        topology.parse(TOPOLOGY_TO_BUILD, inject=injected_attributes)
        topology.build()

        sw9 = topology.get('sw9')
        sw2 = topology.get('sw2')
        sw8 = topology.get('sw8')
        hs3 = topology.get('hs3')
        hs2 = topology.get('hs2')

        assert sw9.metadata == {'shell': 'vtysh', 'name': 'Switch 9'}
        assert sw2.metadata == {
            'image': 'image_for_sw2',
            'shell': 'vtysh',
            'name': 'new_name'
        }
        assert sw8.metadata == {
            'type': 'switch',
            'name': 'Switch 8',
            'chassis': 'chassis_for_sw8'
        }
        assert hs3.metadata == {'type': 'host', 'name': 'Host 3'}
        assert hs2.metadata == {
            'type': 'host',
            'name': 'Host 2',
            'image': 'image_for_all_hosts',
            'hardware': 'hardware_for_sw1_sw3_hs1_hs2'
        }

    finally:
        try:
            rmtree(workdir)
        except:
            pass