build.tests.unit.helpers

Complexity

Total Complexity

6

Size/Duplication

Total Lines	45
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	get_napp_urls()	0	23	4
A	get_controller_mock()	0	6	1
A	get_app_test_client()	0	4	1

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

from kytos.core.config import KytosConfig
from kytos.core import Controller


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


def get_napp_urls(napp):
    """Return the kytos/topology urls.

    The urls will be like:

    urls = [
        (options, methods, url)
    ]

    """
    controller = napp.controller
    controller.api_server.register_napp_endpoints(napp)

    urls = []
    for rule in controller.api_server.app.url_map.iter_rules():
        options = {}
        for arg in rule.arguments:
            options[arg] = "[{0}]".format(arg)

        if f'{napp.username}/{napp.name}' in str(rule):
            urls.append((options, rule.methods, f'{str(rule)}'))

    return urls


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