piecrust2: tests/mockutil.py annotate

annotate tests/mockutil.py @ 147:ab6e7e0e9d44

Pass date information to routing when building URLs. This is so that URLs with dates in them can be built even when the date information is not coming from the source metadata, but from the page's config.

author	Ludovic Chabant <ludovic@chabant.com>
date	Sat, 29 Nov 2014 21:00:44 -0800
parents	133845647083
children	d356f6178623

rev	line source
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	1 import io
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	2 import time
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	3 import random
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	4 import codecs
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	5 import shutil
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	6 import os.path
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	7 import functools
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 import mock
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	9 import yaml
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10 from piecrust.app import PieCrust, PieCrustConfiguration
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	13 resources_path = os.path.abspath(
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	14 os.path.join(
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	15 os.path.dirname(__file__),
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	16 '..', 'piecrust', 'resources'))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	17
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	18
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19 def get_mock_app(config=None):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 app = mock.MagicMock(spec=PieCrust)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 app.config = PieCrustConfiguration()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 return app
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	24
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	25 class _MockFsEntry(object):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	26 def __init__(self, contents):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	27 self.contents = contents
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	28 self.metadata = {'mtime': time.time()}
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	29
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	30
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	31 class _MockFsEntryWriter(object):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	32 def __init__(self, entry):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	33 self._entry = entry
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	34 if isinstance(entry.contents, str):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	35 self._stream = io.StringIO(entry.contents)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	36 elif isinstance(entry.contents, bytes):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	37 self._stream = io.BytesIO(entry.contents)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	38 else:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	39 raise Exception("Unexpected entry contents: %s" % type(entry.contents))
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	40
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	41 def __getattr__(self, name):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	42 return getattr(self._stream, name)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	43
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	44 def __enter__(self):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	45 return self
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	46
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	47 def __exit__(self, exc_type, exc_value, exc_tb):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	48 self._entry.contents = self._stream.getvalue()
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	49 self._entry.metadata['mtime'] = time.time()
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	50 self._stream.close()
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	51
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	52
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	53 class mock_fs(object):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	54 def __init__(self, default_spec=True):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	55 self._root = 'root_%d' % random.randrange(1000)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	56 self._fs = {self._root: {}}
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	57 if default_spec:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	58 self.withDir('counter')
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	59 self.withFile('kitchen/config.yml',
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	60 "site:\n title: Mock Website\n")
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	61
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	62 def path(self, p):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	63 p = p.replace('\\', '/')
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	64 if p in ['/', '', None]:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	65 return '/%s' % self._root
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	66 return '/%s/%s' % (self._root, p.lstrip('/'))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	67
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	68 def getApp(self, cache=True):
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	69 root_dir = self.path('/kitchen')
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	70 return PieCrust(root_dir, cache=cache, debug=True)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	71
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	72 def withDir(self, path):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	73 path = path.replace('\\', '/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	74 path = path.lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	75 path = '/%s/%s' % (self._root, path)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	76 self._createDir(path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	77 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	78
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	79 def withFile(self, path, contents):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	80 path = path.replace('\\', '/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	81 path = path.lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	82 path = '/%s/%s' % (self._root, path)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	83 self._createFile(path, contents)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	84 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	85
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	86 def withAsset(self, path, contents):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	87 return self.withFile('kitchen/' + path, contents)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	88
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	89 def withAssetDir(self, path):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	90 return self.withDir('kitchen/' + path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	91
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	92 def withConfig(self, config):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	93 return self.withFile(
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	94 'kitchen/config.yml',
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	95 yaml.dump(config))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	96
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	97 def withThemeConfig(self, config):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	98 return self.withFile(
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	99 'kitchen/theme/theme_config.yml',
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	100 yaml.dump(config))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	101
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	102 def withPage(self, url, config=None, contents=None):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	103 config = config or {}
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	104 contents = contents or "A test page."
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	105 text = "---\n"
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	106 text += yaml.dump(config)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	107 text += "---\n"
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	108 text += contents
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	109
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	110 name, ext = os.path.splitext(url)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	111 if not ext:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	112 url += '.md'
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	113 url = url.lstrip('/')
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	114 return self.withAsset(url, text)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	115
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	116 def withPageAsset(self, page_url, name, contents=None):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	117 contents = contents or "A test asset."
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	118 url_base, ext = os.path.splitext(page_url)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	119 dirname = url_base + '-assets'
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	120 return self.withAsset('%s/%s' % (dirname, name),
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	121 contents)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	122
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	123 def getStructure(self, path=None):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	124 root = self._fs[self._root]
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	125 if path:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	126 root = self._getEntry(self.path(path))
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	127
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	128 res = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	129 for k, v in root.items():
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	130 self._getStructureRecursive(v, res, k)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	131 return res
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	132
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	133 def _getStructureRecursive(self, src, target, name):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	134 if isinstance(src, _MockFsEntry):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	135 target[name] = src.contents
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	136 return
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	137
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	138 e = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	139 for k, v in src.items():
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	140 self._getStructureRecursive(v, e, k)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	141 target[name] = e
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	142
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	143 def _getEntry(self, path):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	144 cur = self._fs
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	145 path = path.replace('\\', '/').lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	146 bits = path.split('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	147 for p in bits:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	148 try:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	149 cur = cur[p]
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	150 except KeyError:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	151 return None
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	152 return cur
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	153
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	154 def _createDir(self, path):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	155 cur = self._fs
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	156 path = path.replace('\\', '/').strip('/')
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	157 bits = path.split('/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	158 for b in bits:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	159 if b not in cur:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	160 cur[b] = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	161 cur = cur[b]
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	162 return self
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	163
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	164 def _createFile(self, path, contents):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	165 cur = self._fs
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	166 path = path.replace('\\', '/').lstrip('/')
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	167 bits = path.split('/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	168 for b in bits[:-1]:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	169 if b not in cur:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	170 cur[b] = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	171 cur = cur[b]
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	172 cur[bits[-1]] = _MockFsEntry(contents)
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	173 return self
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	174
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	175 def _deleteEntry(self, path):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	176 parent = self._getEntry(os.path.dirname(path))
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	177 assert parent is not None
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	178 name = os.path.basename(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	179 assert name in parent
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	180 del parent[name]
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	181
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	182
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	183 class mock_fs_scope(object):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	184 def __init__(self, fs):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	185 self._fs = fs
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	186 self._patchers = []
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	187 self._originals = {}
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	188
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	189 @property
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	190 def root(self):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	191 return self._fs._root
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	192
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	193 def __enter__(self):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	194 self._startMock()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	195 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	196
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	197 def __exit__(self, type, value, traceback):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	198 self._endMock()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	199
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	200 def _startMock(self):
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	201 # TODO: sadly, there seems to be no way to replace `open` everywhere?
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	202 self._createMock('__main__.open', open, self._open, create=True)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	203 self._createMock('piecrust.records.open', open, self._open, create=True)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	204 self._createMock('codecs.open', codecs.open, self._codecsOpen)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	205 self._createMock('os.listdir', os.listdir, self._listdir)
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	206 self._createMock('os.makedirs', os.makedirs, self._makedirs)
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	207 self._createMock('os.remove', os.remove, self._remove)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	208 self._createMock('os.path.isdir', os.path.isdir, self._isdir)
24 644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	209 self._createMock('os.path.isfile', os.path.isfile, self._isfile)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	210 self._createMock('os.path.islink', os.path.islink, self._islink)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	211 self._createMock('os.path.getmtime', os.path.getmtime, self._getmtime)
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	212 self._createMock('shutil.copyfile', shutil.copyfile, self._copyfile)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	213 self._createMock('shutil.rmtree', shutil.rmtree, self._rmtree)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	214 for p in self._patchers:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	215 p.start()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	216
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	217 def _endMock(self):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	218 for p in self._patchers:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	219 p.stop()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	220
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	221 def _createMock(self, name, orig, func, **kwargs):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	222 self._originals[name] = orig
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	223 self._patchers.append(mock.patch(name, func, **kwargs))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	224
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	225 def _doOpen(self, orig_name, path, mode, args, *kwargs):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	226 path = os.path.normpath(path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	227 if path.startswith(resources_path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	228 return self._originals[orig_name](path, mode, args, *kwargs)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	229
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	230 if 'r' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	231 e = self._getFsEntry(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	232 elif 'w' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	233 e = self._getFsEntry(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	234 if e is None:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	235 contents = ''
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	236 if 'b' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	237 contents = bytes()
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	238 self._fs._createFile(path, contents)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	239 e = self._getFsEntry(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	240 assert e is not None
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	241 else:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	242 raise OSError("Unsupported open mode: %s" % mode)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	243
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	244 if e is None:
617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	245 raise OSError("No such file: %s" % path)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	246 if not isinstance(e, _MockFsEntry):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	247 raise OSError("'%s' is not a file %s" % (path, e))
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	248 if 'b' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	249 assert isinstance(e.contents, bytes)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	250 return _MockFsEntryWriter(e)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	251 assert isinstance(e.contents, str)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	252 return _MockFsEntryWriter(e)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	253
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	254 def _open(self, path, mode, args, *kwargs):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	255 return self._doOpen('__main__.open', path, mode, args, *kwargs)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	256
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	257 def _codecsOpen(self, path, mode, args, *kwargs):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	258 return self._doOpen('codecs.open', path, mode, args, *kwargs)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	259
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	260 def _listdir(self, path):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	261 path = os.path.normpath(path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	262 if path.startswith(resources_path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	263 return self._originals['os.listdir'](path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	264
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	265 e = self._getFsEntry(path)
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	266 if e is None:
617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	267 raise OSError("No such directory: %s" % path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	268 if not isinstance(e, dict):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	269 raise OSError("'%s' is not a directory." % path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	270 return list(e.keys())
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	271
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	272 def _makedirs(self, path, mode=0o777):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	273 if not path.replace('\\', '/').startswith('/' + self.root):
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	274 raise Exception("Shouldn't create directory: %s" % path)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	275 self._fs._createDir(path)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	276
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	277 def _remove(self, path):
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	278 path = os.path.normpath(path)
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	279 self._fs._deleteEntry(path)
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	280
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	281 def _isdir(self, path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	282 path = os.path.normpath(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	283 if path.startswith(resources_path):
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	284 return self._originals['os.path.isdir'](path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	285 e = self._getFsEntry(path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	286 return e is not None and isinstance(e, dict)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	287
24 644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	288 def _isfile(self, path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	289 path = os.path.normpath(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	290 if path.startswith(resources_path):
24 644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	291 return self._originals['os.path.isfile'](path)
644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	292 e = self._getFsEntry(path)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	293 return e is not None and isinstance(e, _MockFsEntry)
24 644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	294
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	295 def _islink(self, path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	296 path = os.path.normpath(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	297 if path.startswith(resources_path):
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	298 return self._originals['os.path.islink'](path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	299 return False
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	300
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	301 def _getmtime(self, path):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	302 path = os.path.normpath(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	303 if path.startswith(resources_path):
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	304 return self._originals['os.path.getmtime'](path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	305 e = self._getFsEntry(path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	306 if e is None:
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	307 raise OSError("No such file: %s" % path)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	308 return e.metadata['mtime']
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	309
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	310 def _copyfile(self, src, dst):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	311 src = os.path.normpath(src)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	312 if src.startswith(resources_path):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	313 with self._originals['__main__.open'](src, 'r') as fp:
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	314 src_text = fp.read()
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	315 else:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	316 e = self._getFsEntry(src)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	317 src_text = e.contents
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	318 if not dst.replace('\\', '/').startswith('/' + self.root):
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	319 raise Exception("Shouldn't copy to: %s" % dst)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	320 self._fs._createFile(dst, src_text)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	321
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	322 def _rmtree(self, path):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	323 if not path.replace('\\', '/').startswith('/' + self.root):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	324 raise Exception("Shouldn't delete trees from: %s" % path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	325 e = self._fs._getEntry(os.path.dirname(path))
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	326 del e[os.path.basename(path)]
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	327
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	328 def _getFsEntry(self, path):
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	329 return self._fs._getEntry(path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	330

Mercurial > piecrust2

annotate tests/mockutil.py @ 147:ab6e7e0e9d44