test-filecache.py
184 lines
| 4.2 KiB
| text/x-python
|
PythonLexer
/ tests / test-filecache.py
|
r14928 | import sys, os, subprocess | ||
|
r16683 | if subprocess.call(['python', '%s/hghave' % os.environ['TESTDIR'], | ||
'cacheable']): | ||||
|
r14928 | sys.exit(80) | ||
|
r18313 | from mercurial import util, scmutil, extensions, hg, ui | ||
|
r14928 | |||
filecache = scmutil.filecache | ||||
class fakerepo(object): | ||||
def __init__(self): | ||||
self._filecache = {} | ||||
def join(self, p): | ||||
return p | ||||
def sjoin(self, p): | ||||
return p | ||||
|
r20045 | @filecache('x', 'y') | ||
|
r14928 | def cached(self): | ||
print 'creating' | ||||
|
r20040 | return 'string from function' | ||
|
r14928 | |||
def invalidate(self): | ||||
for k in self._filecache: | ||||
try: | ||||
delattr(self, k) | ||||
except AttributeError: | ||||
pass | ||||
def basic(repo): | ||||
|
r20045 | print "* neither file exists" | ||
|
r20041 | # calls function | ||
|
r14928 | repo.cached | ||
repo.invalidate() | ||||
|
r20045 | print "* neither file still exists" | ||
|
r20041 | # uses cache | ||
|
r14928 | repo.cached | ||
# create empty file | ||||
f = open('x', 'w') | ||||
f.close() | ||||
repo.invalidate() | ||||
|
r20041 | print "* empty file x created" | ||
|
r14928 | # should recreate the object | ||
repo.cached | ||||
f = open('x', 'w') | ||||
f.write('a') | ||||
f.close() | ||||
repo.invalidate() | ||||
|
r20041 | print "* file x changed size" | ||
|
r14928 | # should recreate the object | ||
repo.cached | ||||
repo.invalidate() | ||||
|
r20045 | print "* nothing changed with either file" | ||
|
r20041 | # stats file again, reuses object | ||
|
r14928 | repo.cached | ||
# atomic replace file, size doesn't change | ||||
# hopefully st_mtime doesn't change as well so this doesn't use the cache | ||||
# because of inode change | ||||
f = scmutil.opener('.')('x', 'w', atomictemp=True) | ||||
f.write('b') | ||||
|
r15057 | f.close() | ||
|
r14928 | |||
repo.invalidate() | ||||
|
r20041 | print "* file x changed inode" | ||
|
r14928 | repo.cached | ||
|
r20045 | # create empty file y | ||
f = open('y', 'w') | ||||
f.close() | ||||
repo.invalidate() | ||||
print "* empty file y created" | ||||
# should recreate the object | ||||
repo.cached | ||||
f = open('y', 'w') | ||||
f.write('A') | ||||
f.close() | ||||
repo.invalidate() | ||||
print "* file y changed size" | ||||
# should recreate the object | ||||
repo.cached | ||||
f = scmutil.opener('.')('y', 'w', atomictemp=True) | ||||
f.write('B') | ||||
f.close() | ||||
repo.invalidate() | ||||
print "* file y changed inode" | ||||
repo.cached | ||||
f = scmutil.opener('.')('x', 'w', atomictemp=True) | ||||
f.write('c') | ||||
f.close() | ||||
f = scmutil.opener('.')('y', 'w', atomictemp=True) | ||||
f.write('C') | ||||
f.close() | ||||
repo.invalidate() | ||||
print "* both files changed inode" | ||||
repo.cached | ||||
|
r14928 | def fakeuncacheable(): | ||
def wrapcacheable(orig, *args, **kwargs): | ||||
return False | ||||
def wrapinit(orig, *args, **kwargs): | ||||
pass | ||||
originit = extensions.wrapfunction(util.cachestat, '__init__', wrapinit) | ||||
|
r14937 | origcacheable = extensions.wrapfunction(util.cachestat, 'cacheable', | ||
wrapcacheable) | ||||
|
r14928 | |||
|
r20045 | for fn in ['x', 'y']: | ||
try: | ||||
os.remove(fn) | ||||
except OSError: | ||||
pass | ||||
|
r14928 | |||
basic(fakerepo()) | ||||
util.cachestat.cacheable = origcacheable | ||||
util.cachestat.__init__ = originit | ||||
|
r18313 | def test_filecache_synced(): | ||
# test old behaviour that caused filecached properties to go out of sync | ||||
os.system('hg init && echo a >> a && hg ci -qAm.') | ||||
repo = hg.repository(ui.ui()) | ||||
# first rollback clears the filecache, but changelog to stays in __dict__ | ||||
repo.rollback() | ||||
repo.commit('.') | ||||
# second rollback comes along and touches the changelog externally | ||||
# (file is moved) | ||||
repo.rollback() | ||||
# but since changelog isn't under the filecache control anymore, we don't | ||||
# see that it changed, and return the old changelog without reconstructing | ||||
# it | ||||
repo.commit('.') | ||||
|
r18316 | def setbeforeget(repo): | ||
os.remove('x') | ||||
|
r20045 | os.remove('y') | ||
|
r20040 | repo.cached = 'string set externally' | ||
|
r18316 | repo.invalidate() | ||
|
r20045 | print "* neither file exists" | ||
|
r18316 | print repo.cached | ||
repo.invalidate() | ||||
f = open('x', 'w') | ||||
f.write('a') | ||||
f.close() | ||||
|
r20041 | print "* file x created" | ||
|
r18316 | print repo.cached | ||
|
r20045 | repo.cached = 'string 2 set externally' | ||
repo.invalidate() | ||||
print "* string set externally again" | ||||
print repo.cached | ||||
repo.invalidate() | ||||
f = open('y', 'w') | ||||
f.write('b') | ||||
f.close() | ||||
print "* file y created" | ||||
print repo.cached | ||||
|
r14928 | print 'basic:' | ||
basic(fakerepo()) | ||||
print 'fakeuncacheable:' | ||||
fakeuncacheable() | ||||
|
r18313 | test_filecache_synced() | ||
|
r18316 | |||
print 'setbeforeget:' | ||||
setbeforeget(fakerepo()) | ||||