view tests/test-sshserver.py @ 36743:d031609b3cb7 stable

changegroup: do not delta lfs revisions

There is no way to distinguish whether a delta base is LFS or non-LFS.

If the delta is against LFS rawtext, and the client trying to apply it has
the base revision stored as fulltext, the delta (aka. bundle) will fail to
apply.

This patch forbids using delta for LFS revisions in changegroup so bad
deltas won't be transmitted.

Note: this does not solve the problem entirely. It solves LFS delta applying
to non-LFS base. But the other direction: non-LFS delta applying to LFS base
is not solved yet.

Differential Revision: https://phab.mercurial-scm.org/D2067

from __future__ import absolute_import, print_function

import io
import unittest

import silenttestrunner

from mercurial import (
    sshserver,
    util,
    wireproto,
)

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__':
    # Don't call into msvcrt to set BytesIO to binary mode
    util.setbinary = lambda fp: True
    silenttestrunner.main(__name__)