Mercurial > public > mercurial-scm > hg-stable
diff tests/test-sshserver.py @ 43076:2372284d9457
formatting: blacken the codebase
This is using my patch to black
(https://github.com/psf/black/pull/826) so we don't un-wrap collection
literals.
Done with:
hg files 'set:**.py - mercurial/thirdparty/** - "contrib/python-zstandard/**"' | xargs black -S
# skip-blame mass-reformatting only
# no-check-commit reformats foo_bar functions
Differential Revision: https://phab.mercurial-scm.org/D6971
| author | Augie Fackler <augie@google.com> |
|---|---|
| date | Sun, 06 Oct 2019 09:45:02 -0400 |
| parents | cf8677cd7286 |
| children | 6000f5b25c9b |
line wrap: on
line diff
--- 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