Adding Agent unit test and asserting calls to logging.

This commit is contained in:
Mitsuo Takaki
2016-04-29 23:49:51 -07:00
parent a6cf64e2b4
commit c7a28edac2
2 changed files with 38 additions and 1 deletions

31
tests/test_scheduler.py Normal file
View 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)