mirror of
https://github.com/facebook/sapling.git
synced 2024-10-09 08:18:15 +03:00
1d44bd2bbb
This allows us to write doctests depending on a ui object, but not on global configs. ui.load() is a class method so we can do wsgiui.load(). All ui() calls but for doctests are replaced with ui.load(). Some of them could be changed to not load configs later.
204 lines
5.9 KiB
Python
204 lines
5.9 KiB
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
|
|
|
|
from __future__ import absolute_import, print_function
|
|
|
|
import os
|
|
from mercurial import (
|
|
error,
|
|
ui as uimod,
|
|
util,
|
|
)
|
|
|
|
hgrc = os.environ['HGRCPATH']
|
|
f = open(hgrc)
|
|
basehgrc = f.read()
|
|
f.close()
|
|
|
|
def testui(user='foo', group='bar', tusers=(), tgroups=(),
|
|
cuser='foo', cgroup='bar', debug=False, silent=False,
|
|
report=True):
|
|
# 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(basehgrc)
|
|
f.write('\n[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(st):
|
|
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))
|
|
|
|
u = uimod.ui.load()
|
|
u.setconfig('ui', 'debug', str(bool(debug)))
|
|
u.setconfig('ui', 'report_untrusted', str(bool(report)))
|
|
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('.', end=' ')
|
|
u.config('paths', name) # warning with debug=True
|
|
print('.', end=' ')
|
|
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.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("# report_untrusted enabled without debug hides warnings")
|
|
u = testui(user='abc', group='def', cuser='foo', report=False)
|
|
|
|
print("# report_untrusted enabled with debug shows warnings")
|
|
u = testui(user='abc', group='def', cuser='foo', debug=True, report=False)
|
|
|
|
print("# ui.readconfig sections")
|
|
filename = 'foobar'
|
|
f = open(filename, 'w')
|
|
f.write('[foobar]\n')
|
|
f.write('baz = quux\n')
|
|
f.close()
|
|
u.readconfig(filename, sections=['foobar'])
|
|
print(u.config('foobar', 'baz'))
|
|
|
|
print()
|
|
print("# read trusted, untrusted, new ui, trusted")
|
|
u = uimod.ui.load()
|
|
u.setconfig('ui', 'debug', 'on')
|
|
u.readconfig(filename)
|
|
u2 = u.copy()
|
|
def username(uid=None):
|
|
return 'foo'
|
|
util.username = username
|
|
u2.readconfig('.hg/hgrc')
|
|
print('trusted:')
|
|
print(u2.config('foobar', 'baz'))
|
|
print('untrusted:')
|
|
print(u2.config('foobar', 'baz', untrusted=True))
|
|
|
|
print()
|
|
print("# error handling")
|
|
|
|
def assertraises(f, exc=error.Abort):
|
|
try:
|
|
f()
|
|
except exc as 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')
|
|
f.close()
|
|
|
|
try:
|
|
testui(user='abc', group='def', silent=True)
|
|
except error.ParseError as inst:
|
|
print(inst)
|
|
|
|
try:
|
|
testui(debug=True, silent=True)
|
|
except error.ParseError as inst:
|
|
print(inst)
|