##// END OF EJS Templates
formatting: blacken the codebase...
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

File last commit:

r43346:2372284d default
r43346:2372284d default
Show More
test-sshserver.py
64 lines | 1.6 KiB | text/x-python | PythonLexer
/ tests / test-sshserver.py
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 from __future__ import absolute_import, print_function
import io
import unittest
import silenttestrunner
from mercurial import (
Gregory Szorc
wireprotoserver: move sshserver into module (API)...
r35877 wireprotoserver,
Gregory Szorc
wireproto: rename wireproto to wireprotov1server (API)...
r37803 wireprotov1server,
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 )
Augie Fackler
formatting: blacken the codebase...
r43346 from mercurial.utils import procutil
Yuya Nishihara
procutil: bulk-replace function calls to point to new module
r37138
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 class SSHServerGetArgsTests(unittest.TestCase):
def testparseknown(self):
tests = [
Gregory Szorc
py3: add b'' to test-sshserver.py...
r36230 (b'* 0\nnodes 0\n', [b'', {}]),
Augie Fackler
formatting: blacken the codebase...
r43346 (
b'* 0\nnodes 40\n1111111111111111111111111111111111111111\n',
[b'1111111111111111111111111111111111111111', {}],
),
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 ]
for input, expected in tests:
Gregory Szorc
py3: add b'' to test-sshserver.py...
r36230 self.assertparse(b'known', input, expected)
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770
def assertparse(self, cmd, input, expected):
server = mockserver(input)
Augie Fackler
formatting: blacken the codebase...
r43346 proto = wireprotoserver.sshv1protocolhandler(
server._ui, server._fin, server._fout
)
Gregory Szorc
wireproto: rename wireproto to wireprotov1server (API)...
r37803 _func, spec = wireprotov1server.commands[cmd]
Gregory Szorc
wireprotoserver: move SSH server operation to a standalone function...
r36232 self.assertEqual(proto.getargs(spec), expected)
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770
Augie Fackler
formatting: blacken the codebase...
r43346
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 def mockserver(inbytes):
ui = mockui(inbytes)
repo = mockrepo(ui)
Gregory Szorc
wireprotoserver: move sshserver into module (API)...
r35877 return wireprotoserver.sshserver(ui, repo)
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770
Augie Fackler
formatting: blacken the codebase...
r43346
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 class mockrepo(object):
def __init__(self, ui):
self.ui = ui
Augie Fackler
formatting: blacken the codebase...
r43346
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 class mockui(object):
def __init__(self, inbytes):
self.fin = io.BytesIO(inbytes)
self.fout = io.BytesIO()
self.ferr = io.BytesIO()
Yuya Nishihara
ui: proxy protect/restorestdio() calls to update internal flag...
r41321 def protectfinout(self):
return self.fin, self.fout
def restorefinout(self, fin, fout):
pass
Augie Fackler
formatting: blacken the codebase...
r43346
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 if __name__ == '__main__':
Matt Harbison
test-sshserver: stabilize for Windows
r35790 # Don't call into msvcrt to set BytesIO to binary mode
Yuya Nishihara
procutil: bulk-replace function calls to point to new module
r37138 procutil.setbinary = lambda fp: True
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 silenttestrunner.main(__name__)