Add unittest for debugprint

This commit is contained in:
Markus Opolka 2020-03-15 10:53:00 +01:00
parent f567c1ca0c
commit 6fc41612c4

26
test/test_main.py Normal file
View File

@ -0,0 +1,26 @@
#!/usr/bin/env python3
import unittest
import unittest.mock as mock
import sys
sys.path.append('..')
from check_http_json import debugPrint
class MainTest(unittest.TestCase):
"""
Tests for main
"""
def test_debugprint(self):
with mock.patch('builtins.print') as mock_print:
debugPrint(True, 'debug')
mock_print.assert_called_once_with('debug')
def test_debugprint_pprint(self):
with mock.patch('check_http_json.pprint') as mock_pprint:
debugPrint(True, 'debug', True)
mock_pprint.assert_called_once_with('debug')