pelican/tests/test_generators.py
Andrea Crotti ff5921a469 - move the try/except dance in support.py
- use relative . imports in test files
- remove the "future with import", since python < 2.6 is not supported
2012-03-18 21:08:43 +00:00

25 lines
874 B
Python

# -*- coding: utf-8 -*-
from mock import MagicMock
from pelican.generators import ArticlesGenerator
from pelican.settings import _DEFAULT_CONFIG
from .support import unittest
class TestArticlesGenerator(unittest.TestCase):
def test_generate_feeds(self):
generator = ArticlesGenerator(None, {'FEED': _DEFAULT_CONFIG['FEED']},
None, _DEFAULT_CONFIG['THEME'], None,
None)
writer = MagicMock()
generator.generate_feeds(writer)
writer.write_feed.assert_called_with([], None, 'feeds/all.atom.xml')
generator = ArticlesGenerator(None, {'FEED': None}, None,
_DEFAULT_CONFIG['THEME'], None, None)
writer = MagicMock()
generator.generate_feeds(writer)
self.assertFalse(writer.write_feed.called)