weeheavy/pelican-theme
1
0
Fork 0
forked from github/pelican
Code Pull requests Activity

added tests for custom save_as for direct templates

Browse source
This commit is contained in:
Stuart Colville 2012-04-29 13:45:14 +01:00
commit 5ab1933be7
1 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

40
tests/test_generators.py
View file

@ -9,6 +9,7 @@ from .support import unittest
CUR_DIR = os.path.dirname(__file__)
class TestArticlesGenerator(unittest.TestCase):
def test_generate_feeds(self):
@ -46,3 +47,42 @@ class TestArticlesGenerator(unittest.TestCase):
elif relfilepath == "article_without_category.rst":
self.assertEquals(article.category.name, 'Default')
def test_direct_templates_save_as_default(self):
settings = _DEFAULT_CONFIG.copy()
settings['DIRECT_TEMPLATES'] = ['archives']
generator = ArticlesGenerator(settings.copy(), settings, None,
_DEFAULT_CONFIG['THEME'], None,
_DEFAULT_CONFIG['MARKUP'])
writer = MagicMock()
generator.generate_direct_templates(writer)
writer.write_file.assert_called_with("archives.html",
generator.get_template("archives"), settings, relative_urls=True,
blog=True, paginated={}, page_name='archives')
def test_direct_templates_save_as_modified(self):
settings = _DEFAULT_CONFIG.copy()
settings['DIRECT_TEMPLATES'] = ['archives']
settings['ARCHIVES_SAVE_AS'] = 'archives/index.html'
generator = ArticlesGenerator(settings, settings, None,
_DEFAULT_CONFIG['THEME'], None,
_DEFAULT_CONFIG['MARKUP'])
writer = MagicMock()
generator.generate_direct_templates(writer)
writer.write_file.assert_called_with("archives/index.html",
generator.get_template("archives"), settings, relative_urls=True,
blog=True, paginated={}, page_name='archives')
def test_direct_templates_save_as_false(self):
settings = _DEFAULT_CONFIG.copy()
settings['DIRECT_TEMPLATES'] = ['archives']
settings['ARCHIVES_SAVE_AS'] = 'archives/index.html'
generator = ArticlesGenerator(settings, settings, None,
_DEFAULT_CONFIG['THEME'], None,
_DEFAULT_CONFIG['MARKUP'])
writer = MagicMock()
generator.generate_direct_templates(writer)
writer.write_file.assert_called_count == 0