diff --git a/mercurial/exthelper.py b/mercurial/exthelper.py --- a/mercurial/exthelper.py +++ b/mercurial/exthelper.py @@ -40,12 +40,14 @@ class exthelper(object): self.command = registrar.command(self.cmdtable) self.configtable = {} self.configitem = registrar.configitem(self.configtable) + self.revsetpredicate = registrar.revsetpredicate() def merge(self, other): self._uicallables.extend(other._uicallables) self._uipopulatecallables.extend(other._uipopulatecallables) self._extcallables.extend(other._extcallables) self._repocallables.extend(other._repocallables) + self.revsetpredicate._table.update(other.revsetpredicate._table) self._commandwrappers.extend(other._commandwrappers) self._extcommandwrappers.extend(other._extcommandwrappers) self._functionwrappers.extend(other._functionwrappers) diff --git a/tests/test-extension.t b/tests/test-extension.t --- a/tests/test-extension.t +++ b/tests/test-extension.t @@ -151,25 +151,35 @@ Check that extensions are loaded in phas $ cat > foo.py < from __future__ import print_function > import os + > from mercurial import exthelper > name = os.path.basename(__file__).rsplit('.', 1)[0] > print("1) %s imported" % name, flush=True) - > def uisetup(ui): + > eh = exthelper.exthelper() + > @eh.uisetup + > def _uisetup(ui): > print("2) %s uisetup" % name, flush=True) - > def extsetup(): + > @eh.extsetup + > def _extsetup(ui): > print("3) %s extsetup" % name, flush=True) - > def uipopulate(ui): + > @eh.uipopulate + > def _uipopulate(ui): > print("4) %s uipopulate" % name, flush=True) - > def reposetup(ui, repo): + > @eh.reposetup + > def _reposetup(ui, repo): > print("5) %s reposetup" % name, flush=True) > + > extsetup = eh.finalextsetup + > reposetup = eh.finalreposetup + > uipopulate = eh.finaluipopulate + > uisetup = eh.finaluisetup + > revsetpredicate = eh.revsetpredicate + > > bytesname = name.encode('utf-8') > # custom predicate to check registration of functions at loading > from mercurial import ( - > registrar, > smartset, > ) - > revsetpredicate = registrar.revsetpredicate() - > @revsetpredicate(bytesname, safe=True) # safe=True for query via hgweb + > @eh.revsetpredicate(bytesname, safe=True) # safe=True for query via hgweb > def custompredicate(repo, subset, x): > return smartset.baseset([r for r in subset if r in {0}]) > EOF