Mercurial > hg
view tests/test-sshserver.py @ 35924:d857cad588e4
lfs: prefetch lfs blobs during revert
The revert command oddly prints out what it will do before requesting the files
to be prefetched. But the 'need to transfer' line indicates the blobs are being
grouped.
author | Matt Harbison <matt_harbison@yahoo.com> |
---|---|
date | Sun, 04 Feb 2018 00:33:28 -0500 |
parents | 1bf5263fe5cc |
children | bf676267f64f |
line wrap: on
line source
from __future__ import absolute_import, print_function import io import unittest import silenttestrunner from mercurial import ( util, wireproto, wireprotoserver, ) 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 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) self.fout = io.BytesIO() self.ferr = io.BytesIO() if __name__ == '__main__': # Don't call into msvcrt to set BytesIO to binary mode util.setbinary = lambda fp: True silenttestrunner.main(__name__)