diff options
author | Gokul Singaraju <gs244f@att.com> | 2018-05-21 15:32:52 -0400 |
---|---|---|
committer | Gokul Singaraju <gs244f@att.com> | 2018-05-22 22:40:18 -0400 |
commit | 678e65fa7938114bf7d66f212b16cef8633db214 (patch) | |
tree | b41462748c9101ea99ff3d3fa36296b159510a0c /tests/test_trapd_exit.py | |
parent | 4571d99c799412d22e140ea8396ba9ad105626a8 (diff) |
Miss HB CBS integration
Issue-ID: DCAEGEN2-279
Change-Id: I3f78e0870d4c620a304fb8d218f1d30187b4fbef
Signed-off-by: Gokul Singaraju <gs244f@att.com>
Diffstat (limited to 'tests/test_trapd_exit.py')
-rw-r--r-- | tests/test_trapd_exit.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/tests/test_trapd_exit.py b/tests/test_trapd_exit.py new file mode 100644 index 0000000..594624f --- /dev/null +++ b/tests/test_trapd_exit.py @@ -0,0 +1,37 @@ +import pytest +import unittest +import trapd_exit + +pid_file="/tmp/test_pid_file" +pid_file_dne="/tmp/test_pid_file_NOT" + +class test_cleanup_and_exit(unittest.TestCase): + """ + Test the cleanup_and_exit mod + """ + + def test_normal_exit(self): + """ + Test normal exit works as expected + """ + open(pid_file, 'w') + + with pytest.raises(SystemExit) as pytest_wrapped_sys_exit: + result = trapd_exit.cleanup_and_exit(0,pid_file) + assert pytest_wrapped_sys_exit.type == SystemExit + assert pytest_wrapped_sys_exit.value.code == 0 + + # compare = str(result).startswith("SystemExit: 0") + # self.assertEqual(compare, True) + + def test_abnormal_exit(self): + """ + Test exit with missing PID file exits non-zero + """ + with pytest.raises(SystemExit) as pytest_wrapped_sys_exit: + result = trapd_exit.cleanup_and_exit(0,pid_file_dne) + assert pytest_wrapped_sys_exit.type == SystemExit + assert pytest_wrapped_sys_exit.value.code == 1 + +if __name__ == '__main__': + unittest.main() |