From 5d49281ede1b846be079bb89671acde652ec6f88 Mon Sep 17 00:00:00 2001 From: David Wilson Date: Sun, 10 Sep 2017 12:14:06 +0530 Subject: [PATCH] Crash if start_transmit/start_receive used on closed side. --- econtext/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/econtext/core.py b/econtext/core.py index d889439b..195dbe18 100644 --- a/econtext/core.py +++ b/econtext/core.py @@ -788,7 +788,6 @@ class Router(object): self._context_by_id[context.context_id] = context self.broker.start_receive(stream) - def add_handler(self, fn, handle=None, persist=True, respondent=None): """Invoke `fn(msg)` for each Message sent to `handle` from this context. Unregister after one invocation if `persist` is ``False``. If @@ -895,6 +894,7 @@ class Broker(object): file descriptor becomes ready for reading, :py:meth:`BasicStream.on_transmit` will be called.""" IOLOG.debug('%r.start_receive(%r)', self, stream) + assert stream.receive_side and stream.receive_side.fd is not None self.defer(self._readers.add, stream.receive_side) def stop_receive(self, stream): @@ -903,7 +903,7 @@ class Broker(object): def start_transmit(self, stream): IOLOG.debug('%r.start_transmit(%r)', self, stream) - assert stream.transmit_side + assert stream.transmit_side and stream.transmit_side.fd is not None self.defer(self._writers.add, stream.transmit_side) def stop_transmit(self, stream):