build.tests.helpers

Complexity

Total Complexity

5

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

Importance

Changes

0

5 Functions

Rating	Name	Duplication	Size	Complexity
A	get_controller_mock()	0	6	1
A	get_interface_mock()	0	6	1
A	get_connection_mock()	0	8	1
A	get_kytos_event_mock()	0	11	1
A	get_switch_mock()	0	3	1

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

from pyof.utils import unpack
from kytos.core import Controller
from kytos.core.config import KytosConfig
from kytos.core.connection import Connection, ConnectionState
from kytos.core.interface import Interface
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(interface_name, port, *args, **kwargs):
    """Return a interface mock."""
    switch = get_switch_mock(0x04)
    switch.connection = Mock()
    iface = Interface(interface_name, port, switch, *args, **kwargs)
    return iface


def get_switch_mock(dpid="00:00:00:00:00:00:00:01"):
    """Return a switch mock."""
    return Switch(dpid)


def get_connection_mock(of_version, target_switch, state=ConnectionState.NEW):
    """Return a connection mock."""
    connection = Connection(Mock(), Mock(), Mock())
    connection.switch = target_switch
    connection.state = state
    connection.protocol.version = of_version
    connection.protocol.unpack = unpack
    return connection


def get_kytos_event_mock(**kwargs):
    """Return a kytos event mock."""
    destination = kwargs.get('destination')
    message = kwargs.get('message')
    source = kwargs.get('source')

    event = MagicMock()
    event.source = source
    event.content = {'destination': destination,
                     'message': message}
    return event