def test_signal_handler(self, stopAgent_mock, os_getpid_mock, os_exit_mock):
# testing exit of children
main.agentPid = 4444
os_getpid_mock.return_value = 5555
main.signal_handler("signum", "frame")
self.assertTrue(os_exit_mock.called)
os_exit_mock.reset_mock()
# testing exit of main process
os_getpid_mock.return_value = main.agentPid
main.signal_handler("signum", "frame")
self.assertFalse(os_exit_mock.called)
self.assertTrue(stopAgent_mock.called)
def test_signal_handler(self, stopAgent_mock, os_getpid_mock, os_exit_mock):
# testing exit of children
main.agentPid = 4444
os_getpid_mock.return_value = 5555
main.signal_handler("signum", "frame")
self.assertTrue(os_exit_mock.called)
os_exit_mock.reset_mock()
# testing exit of main process
os_getpid_mock.return_value = main.agentPid
main.signal_handler("signum", "frame")
self.assertFalse(os_exit_mock.called)
self.assertTrue(stopAgent_mock.called)