changeset 49307:4554e2e965e2

tests: assume that `raw` attribute is present on original socket file object It seems like the original socket file object is always an io.BufferedIO instance. If not, the code will fail and we should try harder to get the socket object (e.g. if the original socket file object is unbuffered, we can get the `_sock` attribute directly from it).
author Manuel Jacob <me@manueljacob.de>
date Tue, 31 May 2022 02:47:22 +0200
parents 425ca3428d03
children 227124098e14
files tests/testlib/badserverext.py
diffstat 1 files changed, 1 insertions(+), 6 deletions(-) [+]
line wrap: on
line diff
--- a/tests/testlib/badserverext.py	Tue May 31 02:36:05 2022 +0200
+++ b/tests/testlib/badserverext.py	Tue May 31 02:47:22 2022 +0200
@@ -335,12 +335,7 @@
 
     def _close(self):
         # We wrap an io.BufferedIO instance.
-        orig = object.__getattribute__(self, '_orig')
-
-        if hasattr(orig, 'raw'):
-            orig.raw._sock.shutdown(socket.SHUT_RDWR)
-        else:
-            self.close()
+        self.raw._sock.shutdown(socket.SHUT_RDWR)
 
     def read(self, size=-1):
         cond = object.__getattribute__(self, '_cond')