sapling/eden/scm/tests/test-sshserver.py
Simon Farnsworth 5232dcf9d1 py3: make .py tests continue to run on python 2
Summary:
Use sed to convert testutil.ddot requires to hghave.

ignore-conflict-markers

for test-simplemerge.py

Reviewed By: simpkins

Differential Revision: D19658355

fbshipit-source-id: afae73eb1e43ead79514dfaf9f911f51ac25972e
2020-01-31 00:18:22 -08:00

52 lines
1.2 KiB
Python

from __future__ import absolute_import, print_function
import io
import unittest
import silenttestrunner
from edenscm.mercurial import sshserver, wireproto
from hghave import require
require(["py2"])
class SSHServerGetArgsTests(unittest.TestCase):
def testparseknown(self):
tests = [
("* 0\nnodes 0\n", ["", {}]),
(
"* 0\nnodes 40\n1111111111111111111111111111111111111111\n",
["1111111111111111111111111111111111111111", {}],
),
]
for input, expected in tests:
self.assertparse("known", input, expected)
def assertparse(self, cmd, input, expected):
server = mockserver(input)
_func, spec = wireproto.commands[cmd]
self.assertEqual(server.getargs(spec), expected)
def mockserver(inbytes):
ui = mockui(inbytes)
repo = mockrepo(ui)
return sshserver.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)
self.fout = io.BytesIO()
self.ferr = io.BytesIO()
if __name__ == "__main__":
silenttestrunner.main(__name__)