build.tests.integration.helpers

Complexity

Total Complexity

3

Size/Duplication

Total Lines	40
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	get_controller_mock()	0	6	1
A	get_interface_mock()	0	6	1
A	get_switch_mock()	0	12	1

"""Module to help to create Integration tests."""
from unittest.mock import 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."""
    switch1 = get_switch_mock(0x04)
    switch1.connection = Mock()
    iface1 = Interface(interface_name, port, switch1, *args, **kwargs)
    return iface1


def get_switch_mock(of_version, connection_state=ConnectionState.NEW,
                    dpid="00:00:00:00:00:00:00:01"):
    """Return a switch mock."""
    switch = Switch(dpid)
    address = Mock()
    port = Mock()
    socket = Mock()
    switch.connection = Connection(address, port, socket)
    switch.connection.protocol.unpack = unpack
    switch.connection.protocol.version = of_version
    switch.connection.state = connection_state
    return switch