pelican-plugins/tag_cloud/test_tag_cloud.py

import unittest, os, sys
import six
import tag_cloud

from pelican.generators import ArticlesGenerator
from pelican.tests.support import get_settings
from pelican.urlwrappers import Tag

CUR_DIR = os.path.dirname(__file__)
CONTENT_DIR = os.path.join(CUR_DIR, 'test_data')

class TestTagCloudGeneration(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls._settings = get_settings(filenames={})
        cls._settings['DEFAULT_CATEGORY'] = 'Default'
        cls._settings['DEFAULT_DATE'] = (1970, 1, 1)
        cls._settings['READERS'] = {'asc': None}
        cls._settings['CACHE_CONTENT'] = False
        tag_cloud.set_default_settings(cls._settings)

        cls.generator = ArticlesGenerator(
            context=cls._settings.copy(), settings=cls._settings,
            path=CONTENT_DIR, theme=cls._settings['THEME'], output_path=None)
        cls.generator.generate_context()


    def test_tag_cloud_random(self):
        tag_cloud.generate_tag_cloud(self.generator)
        expected = [
                (Tag('plugins', self._settings), 1),
                (Tag('fun', self._settings), 4),
                (Tag('python', self._settings), 4),
                (Tag('pelican', self._settings), 1)
            ]
        six.assertCountEqual(self, self.generator.tag_cloud, expected)

    def test_tag_cloud_alphabetical(self):
        self.generator.settings['TAG_CLOUD_SORTING'] = 'alphabetically'
        tag_cloud.generate_tag_cloud(self.generator)
        expected = [
                (Tag('fun', self._settings), 4),
                (Tag('pelican', self._settings), 1),
                (Tag('plugins', self._settings), 1),
                (Tag('python', self._settings), 4)
            ]
        self.assertEqual(self.generator.tag_cloud, expected)
    
    def test_tag_cloud_alphabetical_rev(self):
        self.generator.settings['TAG_CLOUD_SORTING'] = 'alphabetically-rev'
        tag_cloud.generate_tag_cloud(self.generator)
        expected = [
                (Tag('python', self._settings), 4),
                (Tag('plugins', self._settings), 1),
                (Tag('pelican', self._settings), 1),
                (Tag('fun', self._settings), 4)
            ]
        self.assertEqual(self.generator.tag_cloud, expected)

    def test_tag_cloud_size(self):
        self.generator.settings['TAG_CLOUD_SORTING'] = 'size'
        tag_cloud.generate_tag_cloud(self.generator)
        expected = [
                (Tag('pelican', self._settings), 1),
                (Tag('plugins', self._settings), 1),
                (Tag('fun', self._settings), 4),
                (Tag('python', self._settings), 4)
            ]
        self.assertEqual(self.generator.tag_cloud, expected)

    def test_tag_cloud_size_rev(self):
        self.generator.settings['TAG_CLOUD_SORTING'] = 'size-rev'
        tag_cloud.generate_tag_cloud(self.generator)
        expected = [
                (Tag('fun', self._settings), 4),
                (Tag('python', self._settings), 4),
                (Tag('pelican', self._settings), 1),
                (Tag('plugins', self._settings), 1)
            ]
        self.assertEqual(self.generator.tag_cloud, expected)

if __name__ == "__main__":
    unittest.main()