build.tests.unit.test_storehouse

Complexity

Total Complexity

6

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	TestStoreHouse.test_create_box()	0	5	1
A	TestStoreHouse.test_get_data()	0	7	1
A	TestStoreHouse.test_save_evc_callback_no_error()	0	9	1
A	TestStoreHouse.test_get_stored_box()	0	5	1
A	TestStoreHouse.test_save_evc_callback_with_error()	0	11	1
A	TestStoreHouse.setUp()	0	3	1

"""Module to test the schedule.py file."""
import datetime
from unittest import TestCase
from unittest.mock import patch, Mock

from apscheduler.triggers.cron import CronTrigger
from pytz import utc

from napps.kytos.mef_eline.models import EVC
from napps.kytos.mef_eline.storehouse import StoreHouse
from tests.helpers import get_controller_mock


class TestStoreHouse(TestCase):
    """Tests to verify StoreHouse class."""

    def setUp(self):
        """Execute steps before each tests."""
        self.storehouse = StoreHouse(get_controller_mock())

    def test_get_stored_box(self):
        """Test get_stored_box method."""
        self.storehouse.controller.buffers = Mock()
        self.storehouse.get_stored_box(1)
        self.storehouse.controller.buffers.app.put.assert_called_once()

    def test_create_box(self):
        """Test get_stored_box method."""
        self.storehouse.controller.buffers = Mock()
        self.storehouse.create_box()
        self.storehouse.controller.buffers.app.put.assert_called_once()

    @patch('napps.kytos.mef_eline.storehouse.log')
    def test_save_evc_callback_no_error(self, log_mock):
        """Test _save_evc_callback method."""
        self.storehouse._lock = Mock()
        data = Mock()
        data.box_id = 1
        self.storehouse._save_evc_callback('event', data, None)
        self.storehouse._lock.release.assert_called_once()
        log_mock.error.assert_not_called()

    @patch('napps.kytos.mef_eline.storehouse.log')
    def test_save_evc_callback_with_error(self, log_mock):
        """Test _save_evc_callback method."""
        self.storehouse._lock = Mock()
        self.storehouse.box = Mock()
        self.storehouse.box.box_id = 1
        data = Mock()
        data.box_id = 1
        self.storehouse._save_evc_callback('event', data, 'error')
        self.storehouse._lock.release.assert_called_once()
        log_mock.error.assert_called_once()

    @patch('napps.kytos.mef_eline.storehouse.StoreHouse.get_stored_box')
    def test_get_data(self, get_stored_box_mock):
        """Test get_data method."""
        self.storehouse.box = Mock()
        self.storehouse.box.box_id = 2
        self.storehouse.get_data()
        get_stored_box_mock.assert_called_once_with(2)