#!/usr/bin/env python

# Since it's not easy to write a test that portably deals

# with files from different users/groups, we cheat a bit by

# monkey-patching some functions in the util module

import os

from mercurial import ui, util

hgrc = os.environ['HGRCPATH']

def testui(user='foo', group='bar', tusers=(), tgroups=(),

           cuser='foo', cgroup='bar', debug=False, silent=False):

    # user, group => owners of the file

    # tusers, tgroups => trusted users/groups

    # cuser, cgroup => user/group of the current process

    # write a global hgrc with the list of trusted users/groups and

    # some setting so that we can be sure it was read

    f = open(hgrc, 'w')

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

    f.write('global = /some/path\n\n')

    if tusers or tgroups:

        f.write('[trusted]\n')

        if tusers:

            f.write('users = %s\n' % ', '.join(tusers))

        if tgroups:

            f.write('groups = %s\n' % ', '.join(tgroups))

    f.close()

    # override the functions that give names to uids and gids

    def username(uid=None):

        if uid is None:

            return cuser

        return user

    util.username = username

    def groupname(gid=None):

        if gid is None:

            return 'bar'

        return group

    util.groupname = groupname

    def isowner(fp, st=None):

        return user == cuser

    util.isowner = isowner

    # try to read everything

    #print '# File belongs to user %s, group %s' % (user, group)

    #print '# trusted users = %s; trusted groups = %s' % (tusers, tgroups)

    kind = ('different', 'same')

    who = ('', 'user', 'group', 'user and the group')

    trusted = who[(user in tusers) + 2*(group in tgroups)]

    if trusted:

        trusted = ', but we trust the ' + trusted

    print '# %s user, %s group%s' % (kind[user == cuser], kind[group == cgroup],

                                     trusted)

    parentui = ui.ui()

    parentui.updateopts(debug=debug)

    u = ui.ui(parentui=parentui)

    u.readconfig('.hg/hgrc')

    if silent:

        return u

    print 'trusted'

    for name, path in u.configitems('paths'):

        print '   ', name, '=', path

    print 'untrusted'

    for name, path in u.configitems('paths', untrusted=True):

        print '.',

        u.config('paths', name) # warning with debug=True

        print '.',

        u.config('paths', name, untrusted=True) # no warnings

        print name, '=', path

    print

    return u

os.mkdir('repo')

os.chdir('repo')

os.mkdir('.hg')

f = open('.hg/hgrc', 'w')

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

f.write('local = /another/path\n\n')

f.write('interpolated = %(global)s%(local)s\n\n')

f.close()

#print '# Everything is run by user foo, group bar\n'

# same user, same group

testui()

# same user, different group

testui(group='def')

# different user, same group

testui(user='abc')

# ... but we trust the group

testui(user='abc', tgroups=['bar'])

# different user, different group

testui(user='abc', group='def')

# ... but we trust the user

testui(user='abc', group='def', tusers=['abc'])

# ... but we trust the group

testui(user='abc', group='def', tgroups=['def'])

# ... but we trust the user and the group

testui(user='abc', group='def', tusers=['abc'], tgroups=['def'])

# ... but we trust all users

print '# we trust all users'

testui(user='abc', group='def', tusers=['*'])

# ... but we trust all groups

print '# we trust all groups'

testui(user='abc', group='def', tgroups=['*'])

# ... but we trust the whole universe

print '# we trust all users and groups'

testui(user='abc', group='def', tusers=['*'], tgroups=['*'])

# ... check that users and groups are in different namespaces

print "# we don't get confused by users and groups with the same name"

testui(user='abc', group='def', tusers=['def'], tgroups=['abc'])

# ... lists of user names work

print "# list of user names"

testui(user='abc', group='def', tusers=['foo', 'xyz', 'abc', 'bleh'],

       tgroups=['bar', 'baz', 'qux'])

# ... lists of group names work

print "# list of group names"

testui(user='abc', group='def', tusers=['foo', 'xyz', 'bleh'],

       tgroups=['bar', 'def', 'baz', 'qux'])

print "# Can't figure out the name of the user running this process"

testui(user='abc', group='def', cuser=None)

print "# prints debug warnings"

u = testui(user='abc', group='def', cuser='foo', debug=True)

print "# ui.readsections"

filename = 'foobar'

f = open(filename, 'w')

f.write('[foobar]\n')

f.write('baz = quux\n')

f.close()

u.readsections(filename, 'foobar')

print u.config('foobar', 'baz')

print

print "# read trusted, untrusted, new ui, trusted"

u = ui.ui()

u.updateopts(debug=True)

u.readconfig(filename)

u2 = ui.ui(parentui=u)

def username(uid=None):

    return 'foo'

util.username = username

u2.readconfig('.hg/hgrc')

print 'trusted:'

print u2.config('foobar', 'baz')

print u2.config('paths', 'interpolated')

print 'untrusted:'

print u2.config('foobar', 'baz', untrusted=True)

print u2.config('paths', 'interpolated', untrusted=True)

print 

print "# error handling"

def assertraises(f, exc=util.Abort):

    try:

        f()

    except exc, inst:

        print 'raised', inst.__class__.__name__

    else:

        print 'no exception?!'

print "# file doesn't exist"

os.unlink('.hg/hgrc')

assert not os.path.exists('.hg/hgrc')

testui(debug=True, silent=True)

testui(user='abc', group='def', debug=True, silent=True)

print

print "# parse error"

f = open('.hg/hgrc', 'w')

f.write('foo = bar')

f.close()

testui(user='abc', group='def', silent=True)

assertraises(lambda: testui(debug=True, silent=True))

print

print "# interpolation error"

f = open('.hg/hgrc', 'w')

f.write('[foo]\n')

f.write('bar = %(')

f.close()

u = testui(debug=True, silent=True)

print '# regular config:'

print '  trusted',

assertraises(lambda: u.config('foo', 'bar'))

print 'untrusted',

assertraises(lambda: u.config('foo', 'bar', untrusted=True))

u = testui(user='abc', group='def', debug=True, silent=True)

print '  trusted ',

print u.config('foo', 'bar')

print 'untrusted',

assertraises(lambda: u.config('foo', 'bar', untrusted=True))

print '# configitems:'

print '  trusted ',

print u.configitems('foo')

print 'untrusted',

assertraises(lambda: u.configitems('foo', untrusted=True))