test-filecache.py
96 lines
| 2.0 KiB
| text/x-python
|
PythonLexer
/ tests / test-filecache.py
|
r14928 | import sys, os, subprocess | ||
try: | ||||
subprocess.check_call(['%s/hghave' % os.environ['TESTDIR'], 'cacheable']) | ||||
except subprocess.CalledProcessError: | ||||
sys.exit(80) | ||||
from mercurial import util, scmutil, extensions | ||||
filecache = scmutil.filecache | ||||
class fakerepo(object): | ||||
def __init__(self): | ||||
self._filecache = {} | ||||
def join(self, p): | ||||
return p | ||||
def sjoin(self, p): | ||||
return p | ||||
@filecache('x') | ||||
def cached(self): | ||||
print 'creating' | ||||
def invalidate(self): | ||||
for k in self._filecache: | ||||
try: | ||||
delattr(self, k) | ||||
except AttributeError: | ||||
pass | ||||
def basic(repo): | ||||
# file doesn't exist, calls function | ||||
repo.cached | ||||
repo.invalidate() | ||||
# file still doesn't exist, uses cache | ||||
repo.cached | ||||
# create empty file | ||||
f = open('x', 'w') | ||||
f.close() | ||||
repo.invalidate() | ||||
# should recreate the object | ||||
repo.cached | ||||
f = open('x', 'w') | ||||
f.write('a') | ||||
f.close() | ||||
repo.invalidate() | ||||
# should recreate the object | ||||
repo.cached | ||||
repo.invalidate() | ||||
# stats file again, nothing changed, reuses object | ||||
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') | ||||
f.rename() | ||||
repo.invalidate() | ||||
repo.cached | ||||
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 | |||
try: | ||||
os.remove('x') | ||||
except: | ||||
pass | ||||
basic(fakerepo()) | ||||
util.cachestat.cacheable = origcacheable | ||||
util.cachestat.__init__ = originit | ||||
print 'basic:' | ||||
basic(fakerepo()) | ||||
print 'fakeuncacheable:' | ||||
fakeuncacheable() | ||||