build.tests.helpers

Complexity

Total Complexity

6

Size/Duplication

Total Lines	69
Duplicated Lines	0 %

Importance

Changes

0

6 Functions

Rating	Name	Duplication	Size	Complexity
A	get_topology_mock()	0	18	1
A	get_switch_mock()	0	5	1
A	get_link_mock()	0	7	1
A	get_test_client()	0	4	1
A	get_controller_mock()	0	6	1
A	get_interface_mock()	0	8	1

"""Module to help to create tests."""
from unittest.mock import MagicMock, Mock, create_autospec

from kytos.core import Controller
from kytos.core.config import KytosConfig
from kytos.core.interface import Interface
from kytos.core.link import Link
from kytos.core.switch import Switch


def get_controller_mock():
    """Return a controller mock."""
    options = KytosConfig().options['daemon']
    controller = Controller(options)
    controller.log = Mock()
    return controller


def get_interface_mock(name, port_number, switch):
    """Return a interface mock."""
    interface = create_autospec(Interface)
    interface.id = "{}:{}".format(switch.dpid, port_number)
    type(interface).name = name
    type(interface).port_number = port_number
    type(interface).switch = switch
    return interface


def get_link_mock(endpoint_a, endpoint_b):
    """Return a link mock."""
    link = create_autospec(Link)
    type(link).endpoint_a = endpoint_a
    type(link).endpoint_b = endpoint_b
    type(link).metadata = {"A": "B"}
    return link


def get_switch_mock(dpid):
    """Return a switch mock."""
    switch = create_autospec(Switch)
    type(switch).dpid = dpid
    return switch


def get_topology_mock():
    """Create a default topology."""
    switch = get_switch_mock("00:00:00:00:00:00:00:01")
    endpoint_a = get_interface_mock("s1-eth1", 1, switch)
    endpoint_b = get_interface_mock("s1-eth2", 2, switch)
    endpoint_c = get_interface_mock("s1-eth1", 3, switch)
    endpoint_d = get_interface_mock("s1-eth2", 4, switch)
    link_1 = get_link_mock(endpoint_a, endpoint_b)
    link_2 = get_link_mock(endpoint_c, endpoint_d)
    switch.interfaces = {endpoint_a.id: endpoint_a,
                         endpoint_b.id: endpoint_b,
                         endpoint_c.id: endpoint_c,
                         endpoint_d.id: endpoint_d}

    topology = MagicMock()
    topology.links = {"1": link_1, "2": link_2}
    topology.switches = {"00:00:00:00:00:00:00:01": switch}
    return topology


def get_test_client(controller, napp):
    """Return a flask api test client."""
    controller.api_server.register_napp_endpoints(napp)
    return controller.api_server.app.test_client()