--- 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