def test_init_scheduler_unknown_scheduler(self):
""" Test AgentRunner with parameters """
instance = Mock()
instance.config.scheduling.name = "BAD SCHEDULER"
co = AgentRunner(instance)
with self.assertRaisesRegexp(AgentRunnerException, "Scheduling strategies not found"):
co._init_scheduler()
def test_init_scheduler_none_scheduler(self):
""" Test AgentRunner scheduler instanciation """
instance = Mock()
instance.config.scheduling.name = None
co = AgentRunner(instance)
co._init_scheduler()
self.assertTrue(co.scheduler.__class__ is SCHEDULERS[None])
def test_start_no_instance(self):
""" Test AgentRunner with parameters """
sched = {"scheduling":{"name": "A SCHEDULER"}}
instance = None
mock = Mock()
co = AgentRunner(instance)
with self.assertRaisesRegexp(AgentRunnerException, "Can't init a runner for a None module !"):
co._init_scheduler()
def test_stop_no_scheduler(self):
""" Test AgentRunner with parameters """
sched = {"scheduling":{"name": "A SCHEDULER"}}
instance = Mock()
mock = Mock()
co = AgentRunner(instance)
co._init_scheduler = mock
with self.assertRaisesRegexp(AgentRunnerException, "No scheduler found"):
co.stop()
def test_start_no_scheduler(self):
""" Test AgentRunner with parameters """
instance = Mock()
mock = Mock()
instance.config.scheduling.name = "A SCHEDULER"
co = AgentRunner(instance)
co._init_scheduler = mock
with self.assertRaisesRegexp(AgentRunnerException, "No scheduler found"):
co.start()
def test_start_a_scheduler(self):
""" Test AgentRunner with parameters """
sched = {"scheduling":{"name": "A SCHEDULER"}}
mock = Mock()
mocked_sched = Mock()
instance = Mock()
co = AgentRunner(instance)
co._init_scheduler = mock
co.scheduler = mocked_sched
co.start()
mocked_sched.start.assert_called_with()