def test_unlink(self): """ Test handling of an UNLINK token. """ srcPid = Pid(Atom("foo@bar"), 0, 0, 0) destPid = Pid(Atom("spam@egg"), 0, 0, 0) destPid.link(None, srcPid) # Sanity check self.assertNotEqual(destPid._links, set([])) ctrlMessage = (self.handler.CTRLMSGOP_UNLINK, srcPid, destPid) self.handler.passThroughMessage(None, ctrlMessage, None) self.assertEqual(destPid._links, set([]))
def test_exit2(self): """ Test handling of an EXIT2 token. """ srcPid = Pid(Atom("foo@bar"), 0, 0, 0) destPid = Pid(Atom("spam@egg"), 0, 0, 0) destPid.link(None, srcPid) called = [] destPid._handlers[srcPid] = [lambda *args: called.append(args)] ctrlMessage = (self.handler.CTRLMSGOP_EXIT2, srcPid, destPid, "reason") self.handler.passThroughMessage(None, ctrlMessage, None) self.assertEqual(called, [("reason",)]) self.assertEqual(destPid._links, set([]))