diff --git a/CHANGELOG b/CHANGELOG index ef9bc070..46aa68a5 100644 --- a/CHANGELOG +++ b/CHANGELOG @@ -4,6 +4,7 @@ X.X * Improved the english documentation * Fixed packaging using setuptools entrypoints * Added typogrify support +* Added a way to disable feed generation 2.8 diff --git a/tests/test_generators.py b/tests/test_generators.py index 94088c34..20929622 100644 --- a/tests/test_generators.py +++ b/tests/test_generators.py @@ -1,36 +1,28 @@ # -*- coding: utf-8 -*- -from __future__ import with_statement try: - from unittest2 import TestCase + import unittest2 as unittest except ImportError, e: - from unittest import TestCase + import unittest # NOQA from pelican.generators import ArticlesGenerator from pelican.settings import _DEFAULT_CONFIG -class TestArticlesGenerator(TestCase): +from mock import MagicMock + + +class TestArticlesGenerator(unittest.TestCase): def test_generate_feeds(self): - class FakeWriter(object): - def __init__(self): - self.called = False - - def write_feed(self, *args, **kwargs): - self.called = True - generator = ArticlesGenerator(None, {'FEED': _DEFAULT_CONFIG['FEED']}, None, _DEFAULT_CONFIG['THEME'], None, None) - writer = FakeWriter() + writer = MagicMock() generator.generate_feeds(writer) - assert writer.called, ("The feed should be written, " - "if settings['FEED'] is specified.") + writer.write_feed.assert_called_with([], None, 'feeds/all.atom.xml') generator = ArticlesGenerator(None, {'FEED': None}, None, _DEFAULT_CONFIG['THEME'], None, None) - writer = FakeWriter() + writer = MagicMock() generator.generate_feeds(writer) - assert not writer.called, ("If settings['FEED'] is None, " - "the feed should not be generated.") - + self.assertFalse(writer.write_feed.called)