piecrust2: tests/mockutil.py comparison

comparison tests/mockutil.py @ 85:3471ffa059b2

Add a `BakeScheduler` to handle build dependencies. Add unit-tests.

author	Ludovic Chabant <ludovic@chabant.com>
date	Wed, 03 Sep 2014 17:27:50 -0700
parents	485682a6de50
children	133845647083

comparison

equal deleted inserted replaced

-:b3ce11b2cf36
+:3471ffa059b2
 app = mock.MagicMock(spec=PieCrust)
 app.config = PieCrustConfiguration()
 return app
-def with_mock_fs_app(f):
+class _MockFsEntry(object):
-@functools.wraps(f)
+def __init__(self, contents):
-def wrapper(app, *args, **kwargs):
+self.contents = contents
-with mock_fs_scope(app):
+self.metadata = {'mtime': time.time()}
-real_app = app.getApp()
-return f(real_app, *args, **kwargs)
-return wrapper
+class _MockFsEntryWriter(object):
+def __init__(self, entry):
+self._entry = entry
+if isinstance(entry.contents, str):
+self._stream = io.StringIO(entry.contents)
+elif isinstance(entry.contents, bytes):
+self._stream = io.BytesIO(entry.contents)
+else:
+raise Exception("Unexpected entry contents: %s" % type(entry.contents))
+def __getattr__(self, name):
+return getattr(self._stream, name)
+def __enter__(self):
+return self
+def __exit__(self, exc_type, exc_value, exc_tb):
+self._entry.contents = self._stream.getvalue()
+self._stream.close()
 class mock_fs(object):
 def __init__(self, default_spec=True):
 self._root = 'root_%d' % random.randrange(1000)
 p = p.replace('\\', '/')
 if p in ['/', '', None]:
 return '/%s' % self._root
 return '/%s/%s' % (self._root, p.lstrip('/'))
-def getApp(self):
+def getApp(self, cache=True):
 root_dir = self.path('/kitchen')
-return PieCrust(root_dir, cache=False)
+return PieCrust(root_dir, cache=cache, debug=True)
 def withDir(self, path):
 path = path.replace('\\', '/')
 path = path.lstrip('/')
 path = '/%s/%s' % (self._root, path)
 for k, v in root.items():
 self._getStructureRecursive(v, res, k)
 return res
 def _getStructureRecursive(self, src, target, name):
-if isinstance(src, tuple):
+if isinstance(src, _MockFsEntry):
-target[name] = src[0]
+target[name] = src.contents
 return
 e = {}
 for k, v in src.items():
 self._getStructureRecursive(v, e, k)
 return None
 return cur
 def _createDir(self, path):
 cur = self._fs
-bits = path.strip('/').split('/')
+path = path.replace('\\', '/').strip('/')
+bits = path.split('/')
 for b in bits:
 if b not in cur:
 cur[b] = {}
 cur = cur[b]
 return self
 def _createFile(self, path, contents):
 cur = self._fs
-bits = path.strip('/').split('/')
+path = path.replace('\\', '/').lstrip('/')
+bits = path.split('/')
 for b in bits[:-1]:
 if b not in cur:
 cur[b] = {}
 cur = cur[b]
-cur[bits[-1]] = (contents, {'mtime': time.time()})
+cur[bits[-1]] = _MockFsEntry(contents)
 return self
+def _deleteEntry(self, path):
+parent = self._getEntry(os.path.dirname(path))
+assert parent is not None
+name = os.path.basename(path)
+assert name in parent
+del parent[name]
 class mock_fs_scope(object):
 def __init__(self, fs):
 self._fs = fs
 def __exit__(self, type, value, traceback):
 self._endMock()
 def _startMock(self):
 self._createMock('__main__.open', open, self._open, create=True)
+# TODO: WTF, apparently the previous one doesn't really work?
+self._createMock('piecrust.records.open', open, self._open, create=True)
 self._createMock('codecs.open', codecs.open, self._codecsOpen)
 self._createMock('os.listdir', os.listdir, self._listdir)
 self._createMock('os.makedirs', os.makedirs, self._makedirs)
 self._createMock('os.path.isdir', os.path.isdir, self._isdir)
 self._createMock('os.path.isfile', os.path.isfile, self._isfile)
 self._createMock('os.path.islink', os.path.islink, self._islink)
 self._createMock('os.path.getmtime', os.path.getmtime, self._getmtime)
 self._createMock('shutil.copyfile', shutil.copyfile, self._copyfile)
+self._createMock('shutil.rmtree', shutil.rmtree, self._rmtree)
 for p in self._patchers:
 p.start()
 def _endMock(self):
 for p in self._patchers:
 def _createMock(self, name, orig, func, **kwargs):
 self._originals[name] = orig
 self._patchers.append(mock.patch(name, func, **kwargs))
-def _open(self, path, *args, **kwargs):
+def _doOpen(self, orig_name, path, mode, *args, **kwargs):
 path = os.path.normpath(path)
 if path.startswith(resources_path):
-return self._originals['__main__.open'](path, **kwargs)
+return self._originals[orig_name](path, mode, *args, **kwargs)
-e = self._getFsEntry(path)
+if 'r' in mode:
+e = self._getFsEntry(path)
+elif 'w' in mode:
+e = self._getFsEntry(path)
+if e is None:
+contents = ''
+if 'b' in mode:
+contents = bytes()
+self._fs._createFile(path, contents)
+e = self._getFsEntry(path)
+assert e is not None
+else:
+raise OSError("Unsupported open mode: %s" % mode)
 if e is None:
 raise OSError("No such file: %s" % path)
-if not isinstance(e, tuple):
+if not isinstance(e, _MockFsEntry):
-raise OSError("'%s' is not a file" % path)
+raise OSError("'%s' is not a file %s" % (path, e))
-return io.StringIO(e[0])
+if 'b' in mode:
+assert isinstance(e.contents, bytes)
-def _codecsOpen(self, path, *args, **kwargs):
+return _MockFsEntryWriter(e)
-path = os.path.normpath(path)
+assert isinstance(e.contents, str)
-if path.startswith(resources_path):
+return _MockFsEntryWriter(e)
-return self._originals['codecs.open'](path, *args, **kwargs)
-e = self._getFsEntry(path)
+def _open(self, path, mode, *args, **kwargs):
-if e is None:
+return self._doOpen('__main__.open', path, mode, *args, **kwargs)
-raise OSError("No such file: %s" % path)
-if not isinstance(e, tuple):
+def _codecsOpen(self, path, mode, *args, **kwargs):
-raise OSError("'%s' is not a file" % path)
+return self._doOpen('codecs.open', path, mode, *args, **kwargs)
-return io.StringIO(e[0])
 def _listdir(self, path):
-if not path.startswith('/' + self.root):
+path = os.path.normpath(path)
+if path.startswith(resources_path):
 return self._originals['os.listdir'](path)
 e = self._getFsEntry(path)
 if e is None:
 raise OSError("No such directory: %s" % path)
 if not isinstance(e, dict):
 raise OSError("'%s' is not a directory." % path)
 return list(e.keys())
-def _makedirs(self, path, mode):
+def _makedirs(self, path, mode=0o777):
-if not path.startswith('/' + self.root):
+if not path.replace('\\', '/').startswith('/' + self.root):
 raise Exception("Shouldn't create directory: %s" % path)
 self._fs._createDir(path)
 def _isdir(self, path):
-if not path.startswith('/' + self.root):
+path = os.path.normpath(path)
+if path.startswith(resources_path):
 return self._originals['os.path.isdir'](path)
 e = self._getFsEntry(path)
 return e is not None and isinstance(e, dict)
 def _isfile(self, path):
-if not path.startswith('/' + self.root):
+path = os.path.normpath(path)
+if path.startswith(resources_path):
 return self._originals['os.path.isfile'](path)
 e = self._getFsEntry(path)
-return e is not None and isinstance(e, tuple)
+return e is not None and isinstance(e, _MockFsEntry)
 def _islink(self, path):
-if not path.startswith('/' + self.root):
+path = os.path.normpath(path)
+if path.startswith(resources_path):
 return self._originals['os.path.islink'](path)
 return False
 def _getmtime(self, path):
-if not path.startswith('/' + self.root):
+path = os.path.normpath(path)
+if path.startswith(resources_path):
 return self._originals['os.path.getmtime'](path)
 e = self._getFsEntry(path)
 if e is None:
 raise OSError("No such file: %s" % path)
-return e[1]['mtime']
+return e.metadata['mtime']
 def _copyfile(self, src, dst):
-if not src.startswith('/' + self.root):
+src = os.path.normpath(src)
-with open(src, 'r') as fp:
+if src.startswith(resources_path):
+with self._originals['__main__.open'](src, 'r') as fp:
 src_text = fp.read()
 else:
 e = self._getFsEntry(src)
-src_text = e[0]
+src_text = e.contents
-if not dst.startswith('/' + self.root):
+if not dst.replace('\\', '/').startswith('/' + self.root):
 raise Exception("Shouldn't copy to: %s" % dst)
 self._fs._createFile(dst, src_text)
+def _rmtree(self, path):
+if not path.replace('\\', '/').startswith('/' + self.root):
+raise Exception("Shouldn't delete trees from: %s" % path)
+e = self._fs._getEntry(os.path.dirname(path))
+del e[os.path.basename(path)]
 def _getFsEntry(self, path):
 return self._fs._getEntry(path)

Mercurial > piecrust2

comparison tests/mockutil.py @ 85:3471ffa059b2