52 lines
1.1 KiB
Python
52 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
|
|
import json
|
|
import unittest
|
|
from unittest.mock import patch
|
|
import sys
|
|
|
|
sys.path.append('..')
|
|
|
|
from check_http_json import *
|
|
|
|
|
|
class NagiosHelperTest(unittest.TestCase):
|
|
"""
|
|
Tests for the NagiosHelper
|
|
"""
|
|
|
|
def test_getcode_default(self):
|
|
|
|
helper = NagiosHelper()
|
|
self.assertEqual(0, helper.getCode())
|
|
|
|
def test_getcode_warning(self):
|
|
|
|
helper = NagiosHelper()
|
|
helper.warning_message = 'foobar'
|
|
self.assertEqual(1, helper.getCode())
|
|
|
|
def test_getcode_critical(self):
|
|
|
|
helper = NagiosHelper()
|
|
helper.critical_message = 'foobar'
|
|
self.assertEqual(2, helper.getCode())
|
|
|
|
def test_getcode_unknown(self):
|
|
|
|
helper = NagiosHelper()
|
|
helper.unknown_message = 'foobar'
|
|
self.assertEqual(3, helper.getCode())
|
|
|
|
def test_getmessage_default(self):
|
|
|
|
helper = NagiosHelper()
|
|
self.assertEqual('OK: Status OK.', helper.getMessage())
|
|
|
|
def test_getmessage_perfomance_data(self):
|
|
|
|
helper = NagiosHelper()
|
|
helper.performance_data = 'foobar'
|
|
self.assertEqual('OK: foobar Status OK. |foobar', helper.getMessage())
|