mirror of
https://github.com/mtan93/cachet-url-monitor.git
synced 2026-03-07 21:21:58 +00:00
5be0217c00
* #10 - Creating auxiliary client class to generate configuration class based on cachet's component list. * Updating the python version in codacy to reduce false positives * Moving some of the cachet operations to the client class to clean up the configuration class and making better constants * Refactoring status to have proper classes and adding more tests. Refactoring the requests tests to use requests-mock. * Removing unused imports from test_scheduler * Adding more tests and the ability to run the client from command line * Updating README and client arg parsing * Fixing broken unit tests
118 lines
3.7 KiB
Python
118 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
import re
|
|
import unittest
|
|
|
|
import mock
|
|
import pytest
|
|
|
|
from cachet_url_monitor.configuration import HttpStatus, Regex
|
|
from cachet_url_monitor.configuration import Latency
|
|
from cachet_url_monitor.status import ComponentStatus
|
|
|
|
|
|
class LatencyTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.expectation = Latency({'type': 'LATENCY', 'threshold': 1})
|
|
|
|
def test_init(self):
|
|
assert self.expectation.threshold == 1
|
|
|
|
def test_get_status_healthy(self):
|
|
def total_seconds():
|
|
return 0.1
|
|
|
|
request = mock.Mock()
|
|
elapsed = mock.Mock()
|
|
request.elapsed = elapsed
|
|
elapsed.total_seconds = total_seconds
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.OPERATIONAL
|
|
|
|
def test_get_status_unhealthy(self):
|
|
def total_seconds():
|
|
return 2
|
|
|
|
request = mock.Mock()
|
|
elapsed = mock.Mock()
|
|
request.elapsed = elapsed
|
|
elapsed.total_seconds = total_seconds
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.PERFORMANCE_ISSUES
|
|
|
|
def test_get_message(self):
|
|
def total_seconds():
|
|
return 0.1
|
|
|
|
request = mock.Mock()
|
|
elapsed = mock.Mock()
|
|
request.elapsed = elapsed
|
|
elapsed.total_seconds = total_seconds
|
|
|
|
assert self.expectation.get_message(request) == ('Latency above '
|
|
'threshold: 0.1000 seconds')
|
|
|
|
|
|
class HttpStatusTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.expectation = HttpStatus({'type': 'HTTP_STATUS', 'status_range': "200-300"})
|
|
|
|
def test_init(self):
|
|
assert self.expectation.status_range == (200, 300)
|
|
|
|
def test_init_with_one_status(self):
|
|
"""With only one value, we still expect a valid tuple"""
|
|
self.expectation = HttpStatus({'type': 'HTTP_STATUS', 'status_range': "200"})
|
|
|
|
assert self.expectation.status_range == (200, 201)
|
|
|
|
def test_init_with_invalid_number(self):
|
|
"""Invalid values should just fail with a ValueError, as we can't convert it to int."""
|
|
with pytest.raises(ValueError):
|
|
self.expectation = HttpStatus({'type': 'HTTP_STATUS', 'status_range': "foo"})
|
|
|
|
def test_get_status_healthy(self):
|
|
request = mock.Mock()
|
|
request.status_code = 200
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.OPERATIONAL
|
|
|
|
def test_get_status_unhealthy(self):
|
|
request = mock.Mock()
|
|
request.status_code = 400
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.PARTIAL_OUTAGE
|
|
|
|
def test_get_message(self):
|
|
request = mock.Mock()
|
|
request.status_code = 400
|
|
|
|
assert self.expectation.get_message(request) == ('Unexpected HTTP '
|
|
'status (400)')
|
|
|
|
|
|
class RegexTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.expectation = Regex({'type': 'REGEX', 'regex': '.*(find stuff).*'})
|
|
|
|
def test_init(self):
|
|
assert self.expectation.regex == re.compile('.*(find stuff).*', re.UNICODE + re.DOTALL)
|
|
|
|
def test_get_status_healthy(self):
|
|
request = mock.Mock()
|
|
request.text = 'We could find stuff\n in this body.'
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.OPERATIONAL
|
|
|
|
def test_get_status_unhealthy(self):
|
|
request = mock.Mock()
|
|
request.text = 'We will not find it here'
|
|
|
|
assert self.expectation.get_status(request) == ComponentStatus.PARTIAL_OUTAGE
|
|
|
|
def test_get_message(self):
|
|
request = mock.Mock()
|
|
request.text = 'We will not find it here'
|
|
|
|
assert self.expectation.get_message(request) == ('Regex did not match '
|
|
'anything in the body')
|