build.tests.unit.test_search_results

Complexity

Total Complexity

10

Size/Duplication

Total Lines	76
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	TestSearchResults.setup()	0	8	1
A	TestSearchResults.get_path()	0	4	1
A	TestSearchResults.get_path_constrained()	0	6	1
A	TestSearchResults.add_interfaces()	0	8	2
A	TestSearchResults.add_metadata_to_link()	0	5	1
A	TestSearchResults.create_link()	0	7	1
A	TestSearchResults.test_setup()	0	3	1
A	TestSearchResults.create_switch()	0	4	1
A	TestSearchResults.generate_topology()	0	6	1

"""Module to test the KytosGraph in graph.py."""
from unittest import TestCase

import networkx as nx
# Core modules to import
from kytos.core.interface import Interface
from kytos.core.link import Link
from kytos.core.switch import Switch

# module under test
from graph import KytosGraph


class TestSearchResults(TestCase):
    """Tests for the graph class."""

    def setup(self):
        """Setup for most tests"""
        switches, links = self.generate_topology()
        self.graph = KytosGraph()
        self.graph.clear()
        self.graph.update_nodes(switches)
        self.graph.update_links(links)
        self.graph.set_path_fun(nx.shortest_simple_paths)

    def get_path(self, source, destination):
        """Return the shortest path"""
        results = self.graph.shortest_paths(source, destination)
        return results

    def get_path_constrained(self, source, destination, depth=None,
                             **metrics):
        """Return the constrained shortest path"""
        return self.graph.constrained_flexible_paths(source, destination,
                                                     depth,
                                                     **metrics)

    def test_setup(self):
        """Provides information on default test setup"""
        self.setup()

    @ staticmethod
    def generate_topology():
        """Generates a predetermined topology"""
        switches = {}
        links = {}
        return (switches, links)

    @ staticmethod
    def create_switch(name, switches):
        '''Add a new switch to the list of switches'''
        switches[name] = Switch(name)

    @ staticmethod
    def add_interfaces(count, switch, interfaces):
        '''Add a new interface to the list of interfaces'''
        for i in range(1, count + 1):
            str1 = "{}:{}".format(switch.dpid, i)
            interface = Interface(str1, i, switch)
            interfaces[str1] = interface
            switch.update_interface(interface)

    @ staticmethod
    def create_link(interface_a, interface_b, interfaces, links):
        '''Add a new link between two interfaces into the list of links'''
        compounded = "{}|{}".format(interface_a, interface_b)
        final_name = compounded
        links[final_name] = Link(
            interfaces[interface_a], interfaces[interface_b])

    @ staticmethod
    def add_metadata_to_link(interface_a, interface_b, metrics, links):
        '''Add metadata to an existing link in the list of links'''
        compounded = "{}|{}".format(interface_a, interface_b)
        links[compounded].extend_metadata(metrics)