pelican-plugins/photos/test_photos.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import os
from pelican.generators import ArticlesGenerator
from pelican.tests.support import unittest, get_settings
from tempfile import mkdtemp
from shutil import rmtree
import photos

CUR_DIR = os.path.dirname(__file__)


class TestPhotos(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.temp_path = mkdtemp(prefix='pelicantests.')
        cls.settings = get_settings(filenames={})
        cls.settings['PATH'] = os.path.join(CUR_DIR, 'test_data')
        cls.settings['PHOTO_LIBRARY'] = os.path.join(CUR_DIR, 'test_data')
        cls.settings['DEFAULT_DATE'] = (1970, 1, 1)
        cls.settings['FILENAME_METADATA'] = '(?P<slug>[^.]+)'
        cls.settings['PLUGINS'] = [photos]
        cls.settings['CACHE_CONTENT'] = False
        cls.settings['OUTPUT_PATH'] = cls.temp_path
        photos.initialized(cls)
        cls.generator = ArticlesGenerator(
            context=cls.settings.copy(), settings=cls.settings,
            path=cls.settings['PATH'], theme=cls.settings['THEME'],
            output_path=cls.settings['OUTPUT_PATH'])
        photos.register()
        cls.generator.generate_context()
        photos.detect_gallery(cls.generator)
        photos.detect_image(cls.generator)

    @classmethod
    def tearDownClass(cls):
        rmtree(cls.temp_path)

    def test_image(self):
        for a in self.generator.articles:
            if 'image' in a.metadata:
                self.assertTrue(
                    hasattr(a, 'photo_image'),
                    msg="{} not recognized.".format(a.metadata['image']))

    def test_gallery(self):
        for a in self.generator.articles:
            if 'gallety' in a.metadata:
                self.assertTrue(
                    hasattr(a, 'photo_gallery'),
                    msg="{} not recognized.".format(a.metadata['gallery']))

    def get_article(self, slug):
        for a in self.generator.articles:
            if slug == a.slug:
                return a
        return None

    def test_photo_article_image(self):
        self.assertEqual(self.get_article('photo').photo_image,
                         ('best.jpg',
                          'photos/agallery/besta.jpg',
                          'photos/agallery/bestt.jpg'))

    def test_photo_article_gallery(self):
        self.assertEqual(self.get_article('photo').photo_gallery[0],
                         ('best.jpg',
                          'photos/agallery/best.jpg',
                          'photos/agallery/bestt.jpg',
                          'EXIF-best', 'Caption-best'))
        self.assertEqual(self.get_article('photo').photo_gallery[1],
                         ('night.png',
                          'photos/agallery/night.jpg',
                          'photos/agallery/nightt.jpg',
                          'EXIF-night', ''))

    def test_photo_article_body(self):
        expected = ('<p>Here is my best photo, again.</p>\n'
                    '<p><img alt="" src="/photos/agallery/besta.jpg" />.</p>')
        self.assertEqual(expected, self.get_article('photo').content)

    def test_filename_article_image(self):
        self.assertEqual(
            ('best.jpg', 'agallery/best.jpg', 'photos/agallery/bestt.jpg'),
            self.get_article('filename').photo_image)

    def test_filename_article_gallery(self):
        self.assertEqual(self.get_article('filename').photo_gallery[0],
                         ('best.jpg',
                          'agallery/best.jpg',
                          'photos/agallery/bestt.jpg',
                          'EXIF-best', 'Caption-best'))
        self.assertEqual(self.get_article('filename').photo_gallery[1],
                         ('night.png',
                          'agallery/night.png',
                          'photos/agallery/nightt.jpg',
                          'EXIF-night', ''))

    def test_filename_article_body(self):
        expected = ('<p>Here is my best photo, again.</p>\n'
                    '<p><img alt="" src="{filename}agallery/best.jpg" />.</p>')
        self.assertEqual(expected, self.get_article('filename').content)

    def test_queue_resize(self):
        expected = [
            ('photos/agallery/best.jpg',
                ('test_data/agallery/best.jpg', (1024, 768, 80))),
            ('photos/agallery/besta.jpg',
                ('test_data/agallery/best.jpg', (760, 506, 80))),
            ('photos/agallery/bestt.jpg',
                ('test_data/agallery/best.jpg', (192, 144, 60))),
            ('photos/agallery/night.jpg',
                ('test_data/agallery/night.png', (1024, 768, 80))),
            ('photos/agallery/nightt.jpg',
                ('test_data/agallery/night.png', (192, 144, 60)))]
        self.assertEqual(sorted(expected), sorted(photos.queue_resize.items()))

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