diff tests/test-sshserver.py @ 41285:cf8677cd7286

ui: proxy protect/restorestdio() calls to update internal flag It should be better to manage the redirection flag solely by the ui class.
author Yuya Nishihara <yuya@tcha.org>
date Wed, 26 Sep 2018 21:41:52 +0900
parents b4d85bc122bd
children 2372284d9457
line wrap: on
line diff
--- a/tests/test-sshserver.py	Wed Sep 26 21:29:13 2018 +0900
+++ b/tests/test-sshserver.py	Wed Sep 26 21:41:52 2018 +0900
@@ -47,6 +47,12 @@
         self.fout = io.BytesIO()
         self.ferr = io.BytesIO()
 
+    def protectfinout(self):
+        return self.fin, self.fout
+
+    def restorefinout(self, fin, fout):
+        pass
+
 if __name__ == '__main__':
     # Don't call into msvcrt to set BytesIO to binary mode
     procutil.setbinary = lambda fp: True