mirror of
https://github.com/drewkerrigan/nagios-http-json.git
synced 2024-11-22 02:13:47 +01:00
0aceabfe91
- Before we only had a boolean debug flag, good for debugging errors. The verbose flag can be used more precisely (`-v -vvv`) to specify when something should be printed. This is useful for adding more output whilst avoiding full debug output that contains secrets.
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
|
|
import unittest
|
|
import unittest.mock as mock
|
|
import sys
|
|
import os
|
|
|
|
sys.path.append('..')
|
|
|
|
from check_http_json import debugPrint
|
|
from check_http_json import verbosePrint
|
|
|
|
|
|
class CLITest(unittest.TestCase):
|
|
"""
|
|
Tests for CLI
|
|
"""
|
|
|
|
def setUp(self):
|
|
"""
|
|
Defining the exitcodes
|
|
"""
|
|
|
|
self.exit_0 = 0 << 8
|
|
self.exit_1 = 1 << 8
|
|
self.exit_2 = 2 << 8
|
|
self.exit_3 = 3 << 8
|
|
|
|
def test_debugprint(self):
|
|
with mock.patch('builtins.print') as mock_print:
|
|
debugPrint(True, 'debug')
|
|
mock_print.assert_called_once_with('debug')
|
|
|
|
def test_verbose(self):
|
|
with mock.patch('builtins.print') as mock_print:
|
|
verbosePrint(0, 3, 'verbose')
|
|
mock_print.assert_not_called()
|
|
|
|
verbosePrint(3, 3, 'verbose')
|
|
mock_print.assert_called_once_with('verbose')
|
|
|
|
def test_cli_without_params(self):
|
|
|
|
command = '/usr/bin/env python3 check_http_json.py > /dev/null 2>&1'
|
|
status = os.system(command)
|
|
|
|
self.assertEqual(status, self.exit_2)
|