from __future__ import absolute_import

import os

from mercurial import (

    hg,

    ui as uimod,

)

from mercurial.hgweb import (

    hgwebdir_mod,

)

hgwebdir = hgwebdir_mod.hgwebdir

os.mkdir('webdir')

os.chdir('webdir')

webdir = os.path.realpath('.')

u = uimod.ui.load()

hg.repository(u, 'a', create=1)

hg.repository(u, 'b', create=1)

os.chdir('b')

hg.repository(u, 'd', create=1)

os.chdir('..')

hg.repository(u, 'c', create=1)

os.chdir('..')

paths = {'t/a/': '%s/a' % webdir,

         'b': '%s/b' % webdir,

         'coll': '%s/*' % webdir,

         'rcoll': '%s/**' % webdir}

config = os.path.join(webdir, 'hgwebdir.conf')

configfile = open(config, 'w')

configfile.write('[paths]\n')

for k, v in paths.items():

    configfile.write('%s = %s\n' % (k, v))

configfile.close()

confwd = hgwebdir(config)

dictwd = hgwebdir(paths)

assert len(confwd.repos) == len(dictwd.repos), 'different numbers'

assert len(confwd.repos) == 9, 'expected 9 repos, found %d' % len(confwd.repos)

found = dict(confwd.repos)

for key, path in dictwd.repos:

    assert key in found, 'repository %s was not found' % key

    assert found[key] == path, 'different paths for repo %s' % key