test-sshserver: stabilize for Windows

tests/test-sshserver.py

@@ -1,44 +1,47
 from __future__ import absolute_import, print_function
 
 import io
 import unittest
 
 import silenttestrunner
 
 from mercurial import (
     sshserver,
+    util,
     wireproto,
 )
 
 class SSHServerGetArgsTests(unittest.TestCase):
     def testparseknown(self):
         tests = [
             ('* 0\nnodes 0\n', ['', {}]),
             ('* 0\nnodes 40\n1111111111111111111111111111111111111111\n',
              ['1111111111111111111111111111111111111111', {}]),
         ]
         for input, expected in tests:
             self.assertparse('known', input, expected)
 
     def assertparse(self, cmd, input, expected):
         server = mockserver(input)
         _func, spec = wireproto.commands[cmd]
         self.assertEqual(server.getargs(spec), expected)
 
 def mockserver(inbytes):
     ui = mockui(inbytes)
     repo = mockrepo(ui)
     return sshserver.sshserver(ui, repo)
 
 class mockrepo(object):
     def __init__(self, ui):
         self.ui = ui
 
 class mockui(object):
     def __init__(self, inbytes):
         self.fin = io.BytesIO(inbytes)
         self.fout = io.BytesIO()
         self.ferr = io.BytesIO()
 
 if __name__ == '__main__':
+    # Don't call into msvcrt to set BytesIO to binary mode
+    util.setbinary = lambda fp: True
     silenttestrunner.main(__name__)