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)