view hgext/largefiles/wirestore.py @ 35503:bbcd2e478391

test-lfs: add tests covering http exchanges

This tries to test every combination of having the extension enabled/disabled
on each side, and then push/pull/clone/identify lfs and non-lfs content.  SSH is
ignored here, because there's enough going on as it is.

The root issue here is again that requirements are not exchanged and preserved
on push/pull/clone.  Doing so should eliminate the cryptic error messages when
using `hg serve`.  The 500 server error is triggered by "ValueError: no common
changegroup version", because the extension forces changegroup3.  Or, if
changegroup3 is enabled manually, it is triggered by "abort: missing processor
for flag '0x2000'!".

Sadly, run-tests.py doesn't support conditionalizing the exit code like it does
lines of output.  Therefore, a couple of tests blot out the exit code by
appending "|| true", since these failures will go away shortly anyway.

# Copyright 2010-2011 Fog Creek Software
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

'''largefile store working over Mercurial's wire protocol'''
from __future__ import absolute_import

from . import (
    lfutil,
    remotestore,
)

class wirestore(remotestore.remotestore):
    def __init__(self, ui, repo, remote):
        cap = remote.capable('largefiles')
        if not cap:
            raise lfutil.storeprotonotcapable([])
        storetypes = cap.split(',')
        if 'serve' not in storetypes:
            raise lfutil.storeprotonotcapable(storetypes)
        self.remote = remote
        super(wirestore, self).__init__(ui, repo, remote.url())

    def _put(self, hash, fd):
        return self.remote.putlfile(hash, fd)

    def _get(self, hash):
        return self.remote.getlfile(hash)

    def _stat(self, hashes):
        '''For each hash, return 0 if it is available, other values if not.
        It is usually 2 if the largefile is missing, but might be 1 the server
        has a corrupted copy.'''
        batch = self.remote.iterbatch()
        for hash in hashes:
            batch.statlfile(hash)
        batch.submit()
        return dict(zip(hashes, batch.results()))