piecrust2: tests/mockutil.py annotate

annotate tests/mockutil.py @ 212:701591ebfcba

data: Improve the Linker and RecursiveLinker features. Add tests.

author	Ludovic Chabant <ludovic@chabant.com>
date	Sat, 31 Jan 2015 11:50:50 -0800
parents	1d44d8bd93e2
children	7544c03b6bab

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
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import mock
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	8 import yaml
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 from piecrust.app import PieCrust, PieCrustConfiguration
181 d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	10 from piecrust.page import Page
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	11 from piecrust.rendering import PageRenderingContext, render_page
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	14 resources_path = os.path.abspath(
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	15 os.path.join(
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	16 os.path.dirname(__file__),
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	17 '..', 'piecrust', 'resources'))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	18
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	19
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 def get_mock_app(config=None):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 app = mock.MagicMock(spec=PieCrust)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 app.config = PieCrustConfiguration()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23 return app
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	25
181 d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	26 def get_simple_page(app, rel_path):
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	27 source = app.getSource('pages')
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	28 metadata = {'path': os.path.splitext(rel_path)[0]}
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	29 return Page(source, metadata, rel_path)
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	30
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	31
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	32 def render_simple_page(page, uri):
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	33 ctx = PageRenderingContext(page, uri)
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	34 rp = render_page(ctx)
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	35 return rp.content
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	36
d356f6178623 tests: Add help functions to get and render a simple page. Ludovic Chabant <ludovic@chabant.com> parents: 120 diff changeset	37
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	38 class _MockFsEntry(object):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	39 def __init__(self, contents):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	40 self.contents = contents
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	41 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	42
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 class _MockFsEntryWriter(object):
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	45 def __init__(self, entry, mode='rt'):
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	46 self._entry = entry
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	47 self._mode = mode
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	48
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	49 if 'b' in mode:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	50 data = entry.contents
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	51 if isinstance(data, str):
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	52 data = data.encode('utf8')
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	53 self._stream = io.BytesIO(data)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	54 else:
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	55 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	56
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	57 def __getattr__(self, name):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	58 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	59
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	60 def __enter__(self):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	61 return self
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	62
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	63 def __exit__(self, exc_type, exc_value, exc_tb):
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	64 if 'w' in self._mode:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	65 if 'a' in self._mode:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	66 self._entry.contents += self._stream.getvalue()
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	67 else:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	68 self._entry.contents = self._stream.getvalue()
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	69 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	70 self._stream.close()
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
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	73 class mock_fs(object):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	74 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	75 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	76 self._fs = {self._root: {}}
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	77 if default_spec:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	78 self.withDir('counter')
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	79 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	80 "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	81
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	82 def path(self, p):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	83 p = p.replace('\\', '/')
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	84 if p in ['/', '', None]:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	85 return '/%s' % self._root
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	86 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	87
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	88 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	89 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	90 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	91
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	92 def withDir(self, path):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	93 path = path.replace('\\', '/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	94 path = path.lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	95 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	96 self._createDir(path)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	97 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	98
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	99 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	100 path = path.replace('\\', '/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	101 path = path.lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	102 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	103 self._createFile(path, contents)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	104 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	105
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	106 def withAsset(self, path, contents):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	107 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	108
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	109 def withAssetDir(self, path):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	110 return self.withDir('kitchen/' + path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	111
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	112 def withConfig(self, config):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	113 return self.withFile(
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	114 'kitchen/config.yml',
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	115 yaml.dump(config))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	116
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	117 def withThemeConfig(self, config):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	118 return self.withFile(
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	119 '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	120 yaml.dump(config))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	121
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	122 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	123 config = config or {}
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	124 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	125 text = "---\n"
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	126 text += yaml.dump(config)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	127 text += "---\n"
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	128 text += contents
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	129
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	130 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	131 if not ext:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	132 url += '.md'
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	133 url = url.lstrip('/')
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	134 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	135
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	136 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	137 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	138 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	139 dirname = url_base + '-assets'
36 485682a6de50 New site layout support. Ludovic Chabant <ludovic@chabant.com> parents: 35 diff changeset	140 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	141 contents)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	142
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	143 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	144 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	145 if path:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	146 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	147
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	148 res = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	149 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	150 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	151 return res
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	152
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	153 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	154 if isinstance(src, _MockFsEntry):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	155 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	156 return
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	157
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	158 e = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	159 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	160 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	161 target[name] = e
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	162
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	163 def _getEntry(self, path):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	164 cur = self._fs
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	165 path = path.replace('\\', '/').lstrip('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	166 bits = path.split('/')
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	167 for p in bits:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	168 try:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	169 cur = cur[p]
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	170 except KeyError:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	171 return None
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	172 return cur
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	173
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	174 def _createDir(self, path):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	175 cur = self._fs
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	176 path = path.replace('\\', '/').strip('/')
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	177 bits = path.split('/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	178 for b in bits:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	179 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	180 cur[b] = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	181 cur = cur[b]
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	182 return self
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	183
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	184 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	185 cur = self._fs
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	186 path = path.replace('\\', '/').lstrip('/')
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	187 bits = path.split('/')
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	188 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	189 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	190 cur[b] = {}
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	191 cur = cur[b]
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	192 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	193 return self
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	194
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	195 def _deleteEntry(self, path):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	196 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	197 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	198 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	199 assert name in parent
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	200 del parent[name]
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	201
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	202
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	203 class mock_fs_scope(object):
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	204 def __init__(self, fs, open_patches=None):
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	205 self.open_patches = open_patches or []
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	206 self._fs = fs
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	207 self._patchers = []
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	208 self._originals = {}
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	209
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	210 @property
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	211 def root(self):
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	212 return self._fs._root
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	213
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	214 def __enter__(self):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	215 self._startMock()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	216 return self
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	217
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	218 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	219 self._endMock()
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 _startMock(self):
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	222 # TODO: sadly, there seems to be no way to replace `open` everywhere?
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	223 modules = self.open_patches + ['__main__', 'piecrust.records']
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	224 for m in modules:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	225 self._createMock('%s.open' % m, open, self._open, create=True)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	226
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	227 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	228 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	229 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	230 self._createMock('os.remove', os.remove, self._remove)
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	231 self._createMock('os.path.exists', os.path.exists, self._exists)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	232 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	233 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	234 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	235 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	236 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	237 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	238 for p in self._patchers:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	239 p.start()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	240
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	241 def _endMock(self):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	242 for p in self._patchers:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	243 p.stop()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	244
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	245 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	246 self._originals[name] = orig
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	247 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	248
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	249 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	250 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	251 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	252 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	253
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	254 if 'r' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	255 e = self._getFsEntry(path)
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	256 elif 'w' in mode or 'x' in mode or 'a' in mode:
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	257 e = self._getFsEntry(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	258 if e is None:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	259 contents = ''
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	260 if 'b' in mode:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	261 contents = bytes()
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	262 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	263 e = self._getFsEntry(path)
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	264 assert e is not None
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	265 elif 'x' in mode:
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	266 raise OSError("File '%s' already exists" % path)
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	267 else:
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	268 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	269
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	270 if e is None:
617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	271 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	272 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	273 raise OSError("'%s' is not a file %s" % (path, e))
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	274
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	275 return _MockFsEntryWriter(e, mode)
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	276
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	277 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	278 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	279
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	280 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	281 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	282
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	283 def _listdir(self, path):
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	284 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	285 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	286 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	287
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	288 e = self._getFsEntry(path)
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	289 if e is None:
617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	290 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	291 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	292 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	293 return list(e.keys())
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	294
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	295 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	296 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	297 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	298 self._fs._createDir(path)
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	299
120 133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	300 def _remove(self, path):
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	301 path = os.path.normpath(path)
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	302 self._fs._deleteEntry(path)
133845647083 Better error management and removal support in baking/processing. Ludovic Chabant <ludovic@chabant.com> parents: 85 diff changeset	303
182 a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	304 def _exists(self, path):
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	305 print("Checking for %s" % path)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	306 path = os.path.normpath(path)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	307 if path.startswith(resources_path):
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	308 return self._originals['os.path.isdir'](path)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	309 e = self._getFsEntry(path)
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	310 return e is not None
a54d3c0b5f4a tests: Patch `os.path.exists` and improve patching for `open`. Ludovic Chabant <ludovic@chabant.com> parents: 181 diff changeset	311
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	312 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	313 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	314 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	315 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	316 e = self._getFsEntry(path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	317 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	318
24 644869022b6e Mock `os.path.isfile`, and fix a few other test utilities. Ludovic Chabant <ludovic@chabant.com> parents: 11 diff changeset	319 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	320 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	321 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	322 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	323 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	324 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	325
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	326 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	327 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	328 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	329 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	330 return False
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	331
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	332 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	333 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	334 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	335 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	336 e = self._getFsEntry(path)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	337 if e is None:
11 617191dec18e Fixes for Windows, make `findPagePath` return a ref path. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	338 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	339 return e.metadata['mtime']
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	340
35 e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	341 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	342 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	343 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	344 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	345 src_text = fp.read()
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	346 else:
e4c345dcf33c More unit tests, fix a bug with the skip patterns. Ludovic Chabant <ludovic@chabant.com> parents: 32 diff changeset	347 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	348 src_text = e.contents
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	349 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	350 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	351 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	352
85 3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	353 def _rmtree(self, path):
3471ffa059b2 Add a `BakeScheduler` to handle build dependencies. Add unit-tests. Ludovic Chabant <ludovic@chabant.com> parents: 36 diff changeset	354 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	355 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	356 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	357 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	358
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	359 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	360 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	361

Mercurial > piecrust2

annotate tests/mockutil.py @ 212:701591ebfcba