Test basic extension support

  $ cat > foobar.py <<EOF

  > import os

  > from mercurial import commands, exthelper, registrar

  > 

  > eh = exthelper.exthelper()

  > eh.configitem(b'tests', b'foo', default=b"Foo")

  > 

  > uisetup = eh.finaluisetup

  > uipopulate = eh.finaluipopulate

  > reposetup = eh.finalreposetup

  > cmdtable = eh.cmdtable

  > configtable = eh.configtable

  > 

  > @eh.uisetup

  > def _uisetup(ui):

  >     ui.debug(b"uisetup called [debug]\\n")

  >     ui.write(b"uisetup called\\n")

  >     ui.status(b"uisetup called [status]\\n")

  >     ui.flush()

  > @eh.uipopulate

  > def _uipopulate(ui):

  >     ui._populatecnt = getattr(ui, "_populatecnt", 0) + 1

  >     ui.write(b"uipopulate called (%d times)\n" % ui._populatecnt)

  > @eh.reposetup

  > def _reposetup(ui, repo):

  >     ui.write(b"reposetup called for %s\\n" % os.path.basename(repo.root))

  >     ui.write(b"ui %s= repo.ui\\n" % (ui == repo.ui and b"=" or b"!"))

  >     ui.flush()

  > @eh.command(b'foo', [], b'hg foo')

  > def foo(ui, *args, **kwargs):

  >     foo = ui.config(b'tests', b'foo')

  >     ui.write(foo)

  >     ui.write(b"\\n")

  > @eh.command(b'bar', [], b'hg bar', norepo=True)

  > def bar(ui, *args, **kwargs):

  >     ui.write(b"Bar\\n")

  > EOF

  $ abspath=`pwd`/foobar.py

  $ mkdir barfoo

  $ cp foobar.py barfoo/__init__.py

  $ barfoopath=`pwd`/barfoo

  $ hg init a

  $ cd a

  $ echo foo > file

  $ hg add file

  $ hg commit -m 'add file'

  $ echo '[extensions]' >> $HGRCPATH

  $ echo "foobar = $abspath" >> $HGRCPATH

  $ hg foo

  uisetup called

  uisetup called [status]

  uipopulate called (1 times)

  uipopulate called (1 times)

  uipopulate called (1 times)

  reposetup called for a

  ui == repo.ui

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  reposetup called for a (chg !)

  ui == repo.ui (chg !)

  Foo

  $ hg foo --quiet

  uisetup called (no-chg !)

  uipopulate called (1 times)

  uipopulate called (1 times)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times)

  reposetup called for a

  ui == repo.ui

  Foo

  $ hg foo --debug

  uisetup called [debug] (no-chg !)

  uisetup called (no-chg !)

  uisetup called [status] (no-chg !)

  uipopulate called (1 times)

  uipopulate called (1 times)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times)

  reposetup called for a

  ui == repo.ui

  Foo

  $ cd ..

  $ hg clone a b

  uisetup called (no-chg !)

  uisetup called [status] (no-chg !)

  uipopulate called (1 times)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times)

  reposetup called for a

  ui == repo.ui

  uipopulate called (1 times)

  uipopulate called (1 times)

  reposetup called for b

  ui == repo.ui

  updating to branch default

  1 files updated, 0 files merged, 0 files removed, 0 files unresolved

  $ hg bar

  uisetup called (no-chg !)

  uisetup called [status] (no-chg !)

  uipopulate called (1 times)

  uipopulate called (1 times) (chg !)

  Bar

  $ echo 'foobar = !' >> $HGRCPATH

module/__init__.py-style

  $ echo "barfoo = $barfoopath" >> $HGRCPATH

  $ cd a

  $ hg foo

  uisetup called

  uisetup called [status]

  uipopulate called (1 times)

  uipopulate called (1 times)

  uipopulate called (1 times)

  reposetup called for a

  ui == repo.ui

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  uipopulate called (1 times) (chg !)

  reposetup called for a (chg !)

  ui == repo.ui (chg !)

  Foo

  $ echo 'barfoo = !' >> $HGRCPATH

Check that extensions are loaded in phases:

  $ cat > foo.py <<EOF

  > import os

  > from mercurial import exthelper

  > from mercurial.utils import procutil

  > 

  > def write(msg):

  >     procutil.stdout.write(msg)

  >     procutil.stdout.flush()

  > 

  > name = os.path.basename(__file__).rsplit('.', 1)[0]

  > bytesname = name.encode('utf-8')

  > write(b"1) %s imported\n" % bytesname)

  > eh = exthelper.exthelper()

  > @eh.uisetup

  > def _uisetup(ui):

  >     write(b"2) %s uisetup\n" % bytesname)

  > @eh.extsetup

  > def _extsetup(ui):

  >     write(b"3) %s extsetup\n" % bytesname)

  > @eh.uipopulate

  > def _uipopulate(ui):

  >     write(b"4) %s uipopulate\n" % bytesname)

  > @eh.reposetup

  > def _reposetup(ui, repo):

  >     write(b"5) %s reposetup\n" % bytesname)

  > 

  > extsetup = eh.finalextsetup

  > reposetup = eh.finalreposetup

  > uipopulate = eh.finaluipopulate

  > uisetup = eh.finaluisetup

  > revsetpredicate = eh.revsetpredicate

  > 

  > # custom predicate to check registration of functions at loading

  > from mercurial import (

  >     smartset,

  > )

  > @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

  $ cp foo.py bar.py

  $ echo 'foo = foo.py' >> $HGRCPATH

  $ echo 'bar = bar.py' >> $HGRCPATH

Check normal command's load order of extensions and registration of functions

 On chg server, extension should be first set up by the server. Then

 object-level setup should follow in the worker process.

  $ hg log -r "foo() and bar()" -q

  1) foo imported

  1) bar imported

  2) foo uisetup

  2) bar uisetup

  3) foo extsetup

  3) bar extsetup

  4) foo uipopulate

  4) bar uipopulate

  4) foo uipopulate

  4) bar uipopulate

  4) foo uipopulate

  4) bar uipopulate

  5) foo reposetup

  5) bar reposetup

  4) foo uipopulate (chg !)

  4) bar uipopulate (chg !)

  4) foo uipopulate (chg !)

  4) bar uipopulate (chg !)

  4) foo uipopulate (chg !)

  4) bar uipopulate (chg !)

  4) foo uipopulate (chg !)

  4) bar uipopulate (chg !)

  4) foo uipopulate (chg !)

  4) bar uipopulate (chg !)

  5) foo reposetup (chg !)

  5) bar reposetup (chg !)

  0:c24b9ac61126

Check hgweb's load order of extensions and registration of functions

  $ cat > hgweb.cgi <<EOF

  > #!$PYTHON

  > from mercurial import demandimport; demandimport.enable()

  > from mercurial.hgweb import hgweb

  > from mercurial.hgweb import wsgicgi

  > application = hgweb(b'.', b'test repo')

  > wsgicgi.launch(application)

  > EOF

  $ . "$TESTDIR/cgienv"

  $ PATH_INFO='/' SCRIPT_NAME='' "$PYTHON" hgweb.cgi \

  >    | grep '^[0-9]) ' # ignores HTML output

  1) foo imported

  1) bar imported

  2) foo uisetup

  2) bar uisetup

  3) foo extsetup

  3) bar extsetup

  4) foo uipopulate

  4) bar uipopulate

  4) foo uipopulate

  4) bar uipopulate

  5) foo reposetup

  5) bar reposetup

(check that revset predicate foo() and bar() are available)

#if msys

  $ PATH_INFO='//shortlog'

#else

  $ PATH_INFO='/shortlog'

#endif

  $ export PATH_INFO

  $ SCRIPT_NAME='' QUERY_STRING='rev=foo() and bar()' "$PYTHON" hgweb.cgi \

  >     | grep '<a href="/rev/[0-9a-z]*">'

     <a href="/rev/c24b9ac61126">add file</a>

  $ echo 'foo = !' >> $HGRCPATH

  $ echo 'bar = !' >> $HGRCPATH

Check "from __future__ import absolute_import" support for external libraries

(import-checker.py reports issues for some of heredoc python code

fragments below, because import-checker.py does not know test specific

package hierarchy. NO_CHECK_* should be used as a limit mark of

heredoc, in order to make import-checker.py ignore them. For

simplicity, all python code fragments below are generated with such

limit mark, regardless of importing module or not.)

#if windows

  $ PATHSEP=";"

#else

  $ PATHSEP=":"

#endif

  $ export PATHSEP

  $ mkdir $TESTTMP/libroot

  $ echo "s = 'libroot/ambig.py'" > $TESTTMP/libroot/ambig.py

  $ mkdir $TESTTMP/libroot/mod

  $ touch $TESTTMP/libroot/mod/__init__.py

  $ echo "s = 'libroot/mod/ambig.py'" > $TESTTMP/libroot/mod/ambig.py

  $ cat > $TESTTMP/libroot/mod/ambigabs.py <<NO_CHECK_EOF

  > import ambig # should load "libroot/ambig.py"

  > s = ambig.s

  > NO_CHECK_EOF

  $ cat > loadabs.py <<NO_CHECK_EOF

  > import mod.ambigabs as ambigabs

  > def extsetup(ui):

  >     print('ambigabs.s=%s' % ambigabs.s, flush=True)

  > NO_CHECK_EOF

  $ (PYTHONPATH=${PYTHONPATH}${PATHSEP}${TESTTMP}/libroot; hg --config extensions.loadabs=loadabs.py root)

  ambigabs.s=libroot/ambig.py

  $TESTTMP/a

Check absolute/relative import of extension specific modules

  $ mkdir $TESTTMP/extroot

  $ cat > $TESTTMP/extroot/bar.py <<NO_CHECK_EOF

  > s = b'this is extroot.bar'

  > NO_CHECK_EOF

  $ mkdir $TESTTMP/extroot/sub1

  $ cat > $TESTTMP/extroot/sub1/__init__.py <<NO_CHECK_EOF

  > s = b'this is extroot.sub1.__init__'

  > NO_CHECK_EOF

  $ cat > $TESTTMP/extroot/sub1/baz.py <<NO_CHECK_EOF

  > s = b'this is extroot.sub1.baz'

  > NO_CHECK_EOF

  $ cat > $TESTTMP/extroot/__init__.py <<NO_CHECK_EOF

  > s = b'this is extroot.__init__'

  > from . import foo

  > def extsetup(ui):

  >     ui.write(b'(extroot) ', foo.func(), b'\n')

  >     ui.flush()

  > NO_CHECK_EOF

  $ cat > $TESTTMP/extroot/foo.py <<NO_CHECK_EOF

  > # test absolute import

  > buf = []

  > def func():

  >     # "not locals" case

  >     import extroot.bar

  >     buf.append(b'import extroot.bar in func(): %s' % extroot.bar.s)

  >     return b'\n(extroot) '.join(buf)

  > # b"fromlist == ('*',)" case

  > from extroot.bar import *

  > buf.append(b'from extroot.bar import *: %s' % s)

  > # "not fromlist" and "if '.' in name" case

  > import extroot.sub1.baz

  > buf.append(b'import extroot.sub1.baz: %s' % extroot.sub1.baz.s)

  > # "not fromlist" and NOT "if '.' in name" case

  > import extroot

  > buf.append(b'import extroot: %s' % extroot.s)

  > # NOT "not fromlist" and NOT "level != -1" case

  > from extroot.bar import s

  > buf.append(b'from extroot.bar import s: %s' % s)

  > NO_CHECK_EOF

  $ (PYTHONPATH=${PYTHONPATH}${PATHSEP}${TESTTMP}; hg --config extensions.extroot=$TESTTMP/extroot root)

  (extroot) from extroot.bar import *: this is extroot.bar

  (extroot) import extroot.sub1.baz: this is extroot.sub1.baz

  (extroot) import extroot: this is extroot.__init__

  (extroot) from extroot.bar import s: this is extroot.bar

  (extroot) import extroot.bar in func(): this is extroot.bar

  $TESTTMP/a

#if demandimport

Examine whether module loading is delayed until actual referring, even

though module is imported with "absolute_import" feature.

Files below in each packages are used for described purpose: