sapling/tests/test-bundle2-pushback-t.py
Jun Wu 0c36fd8b5e tests: translate some .t tests to Python
Summary:
Those tests were converted using:

  echo *.t | xargs -P20 -n1 python -m testutil.dott.translate --black --verify

They run 5x faster (via run-tests.py), and 10x faster (via python directly).

run-tests.py on old .t files, 652 CPU seconds:

  % time ./run-tests.py `cat list-t.txt`  --noprogress
  .................................................................................................................................
  # Ran 129 tests, 0 skipped, 0 failed.
  ./run-tests.py `cat list-t.txt` --noprogress  505.30s user 146.37s system 1451% cpu 44.899 total

run-tests.py on new py tests, 135 CPU seconds:

  % time ./run-tests.py `cat list-py.txt` --noprogress
  .............................................................................................................................
  # Ran 125 tests, 0 skipped, 0 failed.
  ./run-tests.py `cat list-py.txt` --noprogress  55.73s user 78.80s system 744% cpu 18.061 total

vanilla python on new tests, 59 CPU seconds:

  % time (for i in `cat list-py.txt`; do python $i; done;)
  ( for i in `cat list-py.txt`; do; python $i; done; )  41.61s user 17.47s system 90% cpu 1:05.31 total

The new tests also have auto fix ability. `python test-foo-t.py --fix` will
autofix the code.

Reviewed By: xavierd

Differential Revision: D16172902

fbshipit-source-id: dda53990a7dfff5ac214c1237e4206a4d67e8e48
2019-07-17 21:11:32 -07:00

109 lines
3.1 KiB
Python

# Copyright (c) Facebook, Inc. and its affiliates.
# Copyright (c) Mercurial Contributors.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
from __future__ import absolute_import
from testutil.dott import feature, sh, testtmp # noqa: F401
sh % "setconfig 'extensions.treemanifest=!'"
sh % "cat" << r'''
"""A small extension to test bundle2 pushback parts.
Current bundle2 implementation doesn't provide a way to generate those
parts, so they must be created by extensions.
"""
from __future__ import absolute_import
from edenscm.mercurial import bundle2, exchange, pushkey, util
def _newhandlechangegroup(op, inpart):
"""This function wraps the changegroup part handler for getbundle.
It issues an additional pushkey part to send a new
bookmark back to the client"""
result = bundle2.handlechangegroup(op, inpart)
if 'pushback' in op.reply.capabilities:
params = {'namespace': 'bookmarks',
'key': 'new-server-mark',
'old': '',
'new': 'tip'}
encodedparams = [(k, pushkey.encode(v)) for (k,v) in params.items()]
op.reply.newpart('pushkey', mandatoryparams=encodedparams)
else:
op.reply.newpart('output', data='pushback not enabled')
return result
_newhandlechangegroup.params = bundle2.handlechangegroup.params
bundle2.parthandlermapping['changegroup'] = _newhandlechangegroup
''' > "bundle2.py"
sh % "cat" << r"""
[ui]
ssh = $PYTHON "$TESTDIR/dummyssh"
username = nobody <no.reply@example.com>
""" >> "$HGRCPATH"
# Set up server repository
sh % "hg init server"
sh % "cd server"
sh % "echo c0" > "f0"
sh % "hg commit -Am 0" == "adding f0"
# Set up client repository
sh % "cd .."
sh % "hg clone 'ssh://user@dummy/server' client -q"
sh % "cd client"
# Enable extension
sh % "cat" << r"""
[extensions]
bundle2=$TESTTMP/bundle2.py
""" >> "$HGRCPATH"
# Without config
sh % "cd ../client"
sh % "echo c1" > "f1"
sh % "hg commit -Am 1" == "adding f1"
sh % "hg push" == r"""
pushing to ssh://user@dummy/server
searching for changes
remote: adding changesets
remote: adding manifests
remote: adding file changes
remote: added 1 changesets with 1 changes to 1 files
remote: pushback not enabled"""
sh % "hg bookmark" == "no bookmarks set"
sh % "cd ../server"
sh % "tglogp" == r"""
o 1: 2b9c7234e035 public '1'
|
@ 0: 6cee5c8f3e5b public '0'"""
# With config
sh % "cd ../client"
sh % "echo '[experimental]'" >> ".hg/hgrc"
sh % "echo 'bundle2.pushback = True'" >> ".hg/hgrc"
sh % "echo c2" > "f2"
sh % "hg commit -Am 2" == "adding f2"
sh % "hg push" == r"""
pushing to ssh://user@dummy/server
searching for changes
remote: adding changesets
remote: adding manifests
remote: adding file changes
remote: added 1 changesets with 1 changes to 1 files"""
sh % "hg bookmark" == " new-server-mark 2:0a76dfb2e179"
sh % "cd ../server"
sh % "tglogp" == r"""
o 2: 0a76dfb2e179 public '2'
|
o 1: 2b9c7234e035 public '1'
|
@ 0: 6cee5c8f3e5b public '0'"""