def test_restores_handler(self): handler_1 = lambda signum, frame: None handler_2 = lambda s, f: True with dbg._let_signal(signal.SIGALRM, handler_1): with dbg._let_signal(signal.SIGALRM, handler_2): self.assertEqual(signal.getsignal(signal.SIGALRM), handler_2) self.assertEqual(signal.getsignal(signal.SIGALRM), handler_1)
def test_saves_handler(self): handler = lambda signum, frame: None with patch_object(dbg.signal, 'signal') as sig: mgr = dbg._let_signal(signal.SIGALRM, handler) mgr.__enter__() self.assertTrue(sig.call_args[0] == (signal.SIGALRM, handler))