def test_list(self):
expected = [
os.path.join('repo1', 't1', 'template.j2'),
os.path.join('repo2', 't1', 'template.j2'),
os.path.join('repo2', 't2', 'template.j2'),
]
loader = FileSystemLoader(self.root_folder)
assert_items_equal(expected, loader.list_templates())
def test_get_source_removes_tags(self, exists_m, open_m, getmtime_m):
exists_m.return_value = True
open_m.return_value = MagicMock(spec=file)
file_handle = open_m.return_value.__enter__.return_value
file_handle.read.return_value = TEMPLATE_WITH_INVALID_TAGS
loader = FileSystemLoader(self.root_folder)
source, path, update = loader.get_source(Mock(), 't')
expected_path = os.path.join(self.root_folder, 't')
assert expected_path == path
expected_source = '''
<html>
<head>
</head>
<body>
<p>Hi <span>world</span></p>
</body>
</html>
'''
assert self.clean_string(expected_source) == self.clean_string(source)
def test_get_source_raises_error_if_no_template(self):
loader = FileSystemLoader(self.root_folder)
with pytest.raises(TemplateNotFound):
loader.get_source(Mock(), 't')
