##// END OF EJS Templates
revlog: subclass the new `repository.iverifyproblem` Protocol class...
revlog: subclass the new `repository.iverifyproblem` Protocol class This is the same transformation as 3a90a6fd710d did for dirstate, but the CamelCase naming was already cleaned up here. We shouldn't have to explicitly subclass, but I'm doing so to test the interplay of regular attributes and the `attrs` class. Also, PyCharm has a nifty feature that puts a jump point in the gutter to navigate back and forth between the base class and subclasses (and override functions and base class functions) when there's an explicit subclassing. Additionally, PyCharm will immediately flag signature mismatches without a 40m pytype run.

File last commit:

r52465:13c004b5 stable
r53365:4ef6dbc2 default
Show More
test-sshserver.py
64 lines | 1.7 KiB | text/x-python | PythonLexer
/ tests / test-sshserver.py
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 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)
Arseniy Alekseyev
wireprotoserver: ensure that output stream gets flushed on exception...
r52465 ui = server._ui
proto = wireprotoserver.sshv1protocolhandler(ui, ui.fin, ui.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)
Arseniy Alekseyev
wireprotoserver: ensure that output stream gets flushed on exception...
r52465 # note: this test unfortunately doesn't really test anything about
# `sshserver` class anymore: the entirety of logic of that class lives
# in `serveuntil`, and that function is not even called by this test.
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
Gregory Szorc
py3: use class X: instead of class X(object):...
r49801 class mockrepo:
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 def __init__(self, ui):
self.ui = ui
Augie Fackler
formatting: blacken the codebase...
r43346
Gregory Szorc
py3: use class X: instead of class X(object):...
r49801 class mockui:
Siddharth Agarwal
sshserver: add a couple of tests for argument parsing...
r35770 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__)