mirror of
https://github.com/mtan93/cachet-url-monitor.git
synced 2026-03-08 05:31:58 +00:00
Adding Agent unit test and asserting calls to logging.
This commit is contained in:
31
tests/test_scheduler.py
Normal file
31
tests/test_scheduler.py
Normal file
@@ -0,0 +1,31 @@
|
||||
#!/usr/bin/env python
|
||||
import mock
|
||||
import unittest
|
||||
import sys
|
||||
sys.modules['schedule'] = mock.Mock()
|
||||
from cachet_url_monitor.scheduler import Agent
|
||||
|
||||
|
||||
class AgentTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.configuration = mock.Mock()
|
||||
self.agent = Agent(self.configuration)
|
||||
|
||||
def test_init(self):
|
||||
assert self.agent.configuration == self.configuration
|
||||
|
||||
def test_execute(self):
|
||||
evaluate = self.configuration.evaluate
|
||||
push_status_and_metrics = self.configuration.push_status_and_metrics
|
||||
self.agent.execute()
|
||||
|
||||
evaluate.assert_called_once()
|
||||
push_status_and_metrics.assert_called_once()
|
||||
|
||||
def test_start(self):
|
||||
every = sys.modules['schedule'].every
|
||||
self.configuration.data = {'frequency': 5}
|
||||
|
||||
self.agent.start()
|
||||
|
||||
every.assert_called_with(5)
|
||||
Reference in New Issue
Block a user