mirror of
https://github.com/mtan93/cachet-url-monitor.git
synced 2026-03-08 05:31:58 +00:00
117 lines
3.5 KiB
Python
117 lines
3.5 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
|
|
|
|
|
|
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) == 1
|
|
|
|
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) == 2
|
|
|
|
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) == 1
|
|
|
|
def test_get_status_unhealthy(self):
|
|
request = mock.Mock()
|
|
request.status_code = 400
|
|
|
|
assert self.expectation.get_status(request) == 3
|
|
|
|
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) == 1
|
|
|
|
def test_get_status_unhealthy(self):
|
|
request = mock.Mock()
|
|
request.text = 'We will not find it here'
|
|
|
|
assert self.expectation.get_status(request) == 3
|
|
|
|
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')
|