tests/test-sshserver.py
changeset 43076 2372284d9457
parent 41285 cf8677cd7286
child 48875 6000f5b25c9b
--- 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