def test_load_targets(target):
stream = StringIO()
yaml.dump({
'targets': {
'foo': {
'template_name': 'foo.txt.tpl',
'filename': 'foo.txt',
'scopes': {
'path': 'path',
},
},
},
}, stream)
stream.seek(0)
index = Index.load('', stream)
target.assert_called_once_with(
filename='foo.txt',
template_name='foo.txt.tpl',
scopes={
'path': Scope(['path']),
},
)
assert index.targets == {
'foo': target(),
}
def test_load_targets(target):
stream = StringIO()
yaml.dump(
{
'targets': {
'foo': {
'template_name': 'foo.txt.tpl',
'filename': 'foo.txt',
'scopes': {
'path': 'path',
},
},
},
}, stream)
stream.seek(0)
index = Index.load('', stream)
target.assert_called_once_with(
filename='foo.txt',
template_name='foo.txt.tpl',
scopes={
'path': Scope(['path']),
},
)
assert index.targets == {
'foo': target(),
}
def test_load_no_template_paths():
stream = StringIO()
yaml.dump({}, stream)
stream.seek(0)
index = Index.load('', stream)
assert index.targets == {}
assert index.environment.keep_trailing_newline
assert isinstance(index.environment.loader, FileSystemLoader)
def test_load_no_template_paths():
stream = StringIO()
yaml.dump({}, stream)
stream.seek(0)
index = Index.load('', stream)
assert index.targets == {}
assert index.environment.keep_trailing_newline
assert isinstance(index.environment.loader, FileSystemLoader)
def test_generate():
targets = {
'foo': MagicMock(spec=Target),
'bar': MagicMock(spec=Target),
}
targets['foo'].generate.return_value = (
('foo1', 'myfoo1'),
('foo2', 'myfoo2'),
)
targets['bar'].generate.return_value = (('bar1', 'mybar1'), )
context = {}
index = Index(environment=MagicMock(spec=Environment), targets=targets)
result = list(index.generate(context=context))
assert result == [
('bar', 'bar1', 'mybar1'),
('foo', 'foo1', 'myfoo1'),
('foo', 'foo2', 'myfoo2'),
]
def test_generate():
targets = {
'foo': MagicMock(spec=Target),
'bar': MagicMock(spec=Target),
}
targets['foo'].generate.return_value = (
('foo1', 'myfoo1'),
('foo2', 'myfoo2'),
)
targets['bar'].generate.return_value = (
('bar1', 'mybar1'),
)
context = {}
index = Index(environment=MagicMock(spec=Environment), targets=targets)
result = list(index.generate(context=context))
assert result == [
('bar', 'bar1', 'mybar1'),
('foo', 'foo1', 'myfoo1'),
('foo', 'foo2', 'myfoo2'),
]
def test_load_template_paths():
stream = StringIO()
yaml.dump({
'template_paths': {
'foo': 'foo',
'bar': 'bar',
},
}, stream)
stream.seek(0)
index = Index.load('', stream)
assert index.targets == {}
assert index.environment.keep_trailing_newline
assert isinstance(index.environment.loader, PrefixLoader)
def test_load_template_paths():
stream = StringIO()
yaml.dump({
'template_paths': {
'foo': 'foo',
'bar': 'bar',
},
}, stream)
stream.seek(0)
index = Index.load('', stream)
assert index.targets == {}
assert index.environment.keep_trailing_newline
assert isinstance(index.environment.loader, PrefixLoader)
