diff -r 1964d2d1f421 -r 65b86ee69383 mercurial/streamclone.py --- a/mercurial/streamclone.py Fri Apr 13 11:13:05 2018 -0700 +++ b/mercurial/streamclone.py Fri Apr 13 11:14:19 2018 -0700 @@ -126,11 +126,18 @@ # creation. rbranchmap = None if remote.capable('branchmap'): - rbranchmap = remote.branchmap() + with remote.commandexecutor() as e: + rbranchmap = e.callcommand('branchmap', {}).result() repo.ui.status(_('streaming all changes\n')) - fp = remote.stream_out() + with remote.commandexecutor() as e: + fp = e.callcommand('stream_out', {}).result() + + # TODO strictly speaking, this code should all be inside the context + # manager because the context manager is supposed to ensure all wire state + # is flushed when exiting. But the legacy peers don't do this, so it + # doesn't matter. l = fp.readline() try: resp = int(l)