build.tests.unit.test_scheduler

Complexity

Total Complexity

4

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	TestScheduler.test_remove_job()	0	8	1
A	TestScheduler.test_get_job()	0	6	1
A	TestScheduler.test_remove_job_fail()	0	7	2

"""Module to test the scheduler.py"""
from unittest import TestCase
from unittest.mock import MagicMock
from napps.amlight.sdntrace_cp.scheduler import Scheduler
from apscheduler.jobstores.base import JobLookupError


class TestScheduler(TestCase):
    """Test class Scheduler"""

    def test_remove_job(self):
        """Test remove_job with existent id"""
        scheduler = Scheduler()
        trigger_args = {'trigger': 'interval', 'seconds': 2}
        scheduler.add_callable('mock_id', MagicMock, **trigger_args)
        self.assertIsNotNone(scheduler.get_job('mock_id'))
        scheduler.remove_job('mock_id')
        self.assertIsNone(scheduler.get_job('mock_id'))

    def test_remove_job_fail(self):
        """Test remove_job with non existent id"""
        scheduler = Scheduler()
        try:
            scheduler.remove_job('mock_id')
        except JobLookupError:
            self.fail("scheduler.remove_job() raised an error")

    def test_get_job(self):
        """Test get_job and add_callable"""
        scheduler = Scheduler()
        trigger_args = {'trigger': 'interval', 'seconds': 2}
        scheduler.add_callable('mock_id', MagicMock, **trigger_args)
        self.assertIsNotNone(scheduler.get_job('mock_id'))