Fixing failing test. Using @mock.path decorator, instead of overriding the sys.modules.

This commit is contained in:
Mitsuo Takaki
2016-05-19 08:26:18 -07:00
parent ca358eab2b
commit a83abfd1d3
4 changed files with 35 additions and 22 deletions

View File

@@ -1,7 +1,7 @@
endpoint: endpoint:
url: http://localhost:8080/swagger url: http://localhost:8080/swagger
method: GET method: GET
timeout: 0.1 timeout: 0.01
expectation: expectation:
- type: HTTP_STATUS - type: HTTP_STATUS
status: 200 status: 200

View File

@@ -1,20 +1,33 @@
#!/usr/bin/env python #!/usr/bin/env python
import sys
import unittest
import cachet_url_monitor.status import cachet_url_monitor.status
import mock import mock
import unittest
import sys
from requests import ConnectionError, HTTPError, Timeout from requests import ConnectionError, HTTPError, Timeout
sys.modules['requests'] = mock.Mock() sys.modules['requests'] = mock.Mock()
sys.modules['logging'] = mock.Mock() sys.modules['logging'] = mock.Mock()
from cachet_url_monitor.configuration import Configuration from cachet_url_monitor.configuration import Configuration
class ConfigurationTest(unittest.TestCase): class ConfigurationTest(unittest.TestCase):
def setUp(self): def setUp(self):
def getLogger(name): def getLogger(name):
self.mock_logger = mock.Mock() self.mock_logger = mock.Mock()
return self.mock_logger return self.mock_logger
sys.modules['logging'].getLogger = getLogger sys.modules['logging'].getLogger = getLogger
def get(url, headers):
get_return = mock.Mock()
get_return.ok = True
get_return.json = mock.Mock()
get_return.json.return_value = {'data': {'status': 1}}
return get_return
sys.modules['requests'].get = get
self.configuration = Configuration('config.yml') self.configuration = Configuration('config.yml')
sys.modules['requests'].Timeout = Timeout sys.modules['requests'].Timeout = Timeout
sys.modules['requests'].ConnectionError = ConnectionError sys.modules['requests'].ConnectionError = ConnectionError
@@ -27,6 +40,7 @@ class ConfigurationTest(unittest.TestCase):
def test_evaluate(self): def test_evaluate(self):
def total_seconds(): def total_seconds():
return 0.1 return 0.1
def request(method, url, timeout=None): def request(method, url, timeout=None):
response = mock.Mock() response = mock.Mock()
response.status_code = 200 response.status_code = 200
@@ -43,6 +57,7 @@ class ConfigurationTest(unittest.TestCase):
def test_evaluate_with_failure(self): def test_evaluate_with_failure(self):
def total_seconds(): def total_seconds():
return 0.1 return 0.1
def request(method, url, timeout=None): def request(method, url, timeout=None):
response = mock.Mock() response = mock.Mock()
# We are expecting a 200 response, so this will fail the expectation. # We are expecting a 200 response, so this will fail the expectation.

View File

@@ -1,9 +1,10 @@
#!/usr/bin/env python #!/usr/bin/env python
import unittest
import mock import mock
import re import re
import unittest
from cachet_url_monitor.configuration import Expectaction,Latency
from cachet_url_monitor.configuration import HttpStatus, Regex from cachet_url_monitor.configuration import HttpStatus, Regex
from cachet_url_monitor.configuration import Latency
class LatencyTest(unittest.TestCase): class LatencyTest(unittest.TestCase):
@@ -16,6 +17,7 @@ class LatencyTest(unittest.TestCase):
def test_get_status_healthy(self): def test_get_status_healthy(self):
def total_seconds(): def total_seconds():
return 0.1 return 0.1
request = mock.Mock() request = mock.Mock()
elapsed = mock.Mock() elapsed = mock.Mock()
request.elapsed = elapsed request.elapsed = elapsed
@@ -26,6 +28,7 @@ class LatencyTest(unittest.TestCase):
def test_get_status_unhealthy(self): def test_get_status_unhealthy(self):
def total_seconds(): def total_seconds():
return 2 return 2
request = mock.Mock() request = mock.Mock()
elapsed = mock.Mock() elapsed = mock.Mock()
request.elapsed = elapsed request.elapsed = elapsed
@@ -36,6 +39,7 @@ class LatencyTest(unittest.TestCase):
def test_get_message(self): def test_get_message(self):
def total_seconds(): def total_seconds():
return 0.1 return 0.1
request = mock.Mock() request = mock.Mock()
elapsed = mock.Mock() elapsed = mock.Mock()
request.elapsed = elapsed request.elapsed = elapsed

View File

@@ -5,8 +5,6 @@ import unittest
import mock import mock
sys.modules['schedule'] = mock.Mock() sys.modules['schedule'] = mock.Mock()
# sys.modules['cachet_url_monitor.configuration.Configuration'] = mock.Mock()
sys.modules['requests'] = mock.Mock()
from cachet_url_monitor.scheduler import Agent, Scheduler from cachet_url_monitor.scheduler import Agent, Scheduler
@@ -36,9 +34,8 @@ class AgentTest(unittest.TestCase):
class SchedulerTest(unittest.TestCase): class SchedulerTest(unittest.TestCase):
@mock.patch('cachet_url_monitor.configuration.Configuration.__init__', mock.Mock(return_value=None)) @mock.patch('requests.get')
@mock.patch('cachet_url_monitor.configuration.Configuration.is_create_incident', mock.Mock(return_value=False)) def setUp(self, mock_requests):
def setUp(self):
def get(url, headers): def get(url, headers):
get_return = mock.Mock() get_return = mock.Mock()
get_return.ok = True get_return.ok = True
@@ -46,18 +43,15 @@ class SchedulerTest(unittest.TestCase):
get_return.json.return_value = {'data': {'status': 1}} get_return.json.return_value = {'data': {'status': 1}}
return get_return return get_return
sys.modules['requests'].get = get mock_requests.get = get
self.scheduler = Scheduler('config.yml') self.scheduler = Scheduler('config.yml')
def test_init(self): def test_init(self):
assert self.scheduler.stop == False assert self.scheduler.stop == False
@mock.patch('cachet_url_monitor.configuration.Configuration', create=True) def test_start(self):
def test_start(self, mock_configuration):
# TODO(mtakaki|2016-05-01): We need a better way of testing this method. # TODO(mtakaki|2016-05-01): We need a better way of testing this method.
# Leaving it as a placeholder. # Leaving it as a placeholder.
mock_configuration.data = {'frequency': 30}
self.scheduler.stop = True self.scheduler.stop = True
self.scheduler.start() self.scheduler.start()