diff -r 57875cf423c9 -r 2372284d9457 tests/test-sshserver.py --- a/tests/test-sshserver.py Sat Oct 05 10:29:34 2019 -0400 +++ b/tests/test-sshserver.py Sun Oct 06 09:45:02 2019 -0400 @@ -10,37 +10,41 @@ wireprotov1server, ) -from mercurial.utils import ( - procutil, -) +from mercurial.utils import procutil + class SSHServerGetArgsTests(unittest.TestCase): def testparseknown(self): tests = [ (b'* 0\nnodes 0\n', [b'', {}]), - (b'* 0\nnodes 40\n1111111111111111111111111111111111111111\n', - [b'1111111111111111111111111111111111111111', {}]), + ( + b'* 0\nnodes 40\n1111111111111111111111111111111111111111\n', + [b'1111111111111111111111111111111111111111', {}], + ), ] for input, expected in tests: self.assertparse(b'known', input, expected) def assertparse(self, cmd, input, expected): server = mockserver(input) - proto = wireprotoserver.sshv1protocolhandler(server._ui, - server._fin, - server._fout) + proto = wireprotoserver.sshv1protocolhandler( + server._ui, server._fin, server._fout + ) _func, spec = wireprotov1server.commands[cmd] self.assertEqual(proto.getargs(spec), expected) + def mockserver(inbytes): ui = mockui(inbytes) repo = mockrepo(ui) return wireprotoserver.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) @@ -53,6 +57,7 @@ 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