spy: add support for cancel_read and cancel_write

1 files changed, 14 insertions, 1 deletions

diff --git a/serial/urlhandler/protocol_spy.py b/serial/urlhandler/protocol_spy.py
index 4921990..939d5aa 100644
--- a/serial/urlhandler/protocol_spy.py
+++ b/serial/urlhandler/protocol_spy.py
@@ -150,7 +150,10 @@ class FormatHexdump(object):
 
 
 class Serial(serial.Serial):
-    """Just inherit the native Serial port implementation and patch the port property."""
+    """\
+    Inherit the native Serial port implementation and wrap all the methods and
+    attributes.
+    """
     # pylint: disable=no-member
 
     def __init__(self, *args, **kwargs):
@@ -204,6 +207,16 @@ class Serial(serial.Serial):
             self.formatter.rx(rx)
         return rx
 
+    if hasattr(serial.Serial, 'cancel_read'):
+        def cancel_read(self):
+            self.formatter.control('Q-RX', 'cancel_read')
+            super(Serial, self).cancel_read()
+
+    if hasattr(serial.Serial, 'cancel_write'):
+        def cancel_write(self):
+            self.formatter.control('Q-TX', 'cancel_write')
+            super(Serial, self).cancel_write()
+
     @property
     def in_waiting(self):
         n = super(Serial, self).in_waiting