Mercurial > piecrust2

comparison tests/tmpfs.py @ 673:d6403c21bdea

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
tests: Improve failure reporting.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 06 Mar 2016 23:41:35 -0800
parents bab91fcef741
children f987b29d6fab
comparison
equal deleted inserted replaced
672:783ff8bc4e03 673:d6403c21bdea
66 fp.write('\n'.join(traceback.format_stack(limit=10))) 66 fp.write('\n'.join(traceback.format_stack(limit=10)))
67 self._done = True 67 self._done = True
68 68
69 69
70 class TempDirScope(object): 70 class TempDirScope(object):
71 def __init__(self, fs, open_patches=None): 71 def __init__(self, fs, open_patches=None, keep=False):
72 self._fs = fs 72 self._fs = fs
73 self._open = open 73 self._open = open
74 self._keep = keep
74 75
75 @property 76 @property
76 def root(self): 77 def root(self):
77 return self._fs._root 78 return self._fs._root
78 79
79 def __enter__(self): 80 def __enter__(self):
80 return self 81 return self
81 82
82 def __exit__(self, type, value, traceback): 83 def __exit__(self, type, value, traceback):
83 shutil.rmtree(self.root) 84 if not self._keep:
85 shutil.rmtree(self.root)
84 86