sigal/tests/test_gallery.py

# -*- coding:utf-8 -*-

import os
import pytest

from sigal.gallery import PathsDb, get_metadata
from sigal.settings import read_settings

CURRENT_DIR = os.path.dirname(__file__)
SAMPLE_DIR = os.path.join(CURRENT_DIR, 'sample')

REF = {
    'dir1': {
        'title': 'An example gallery',
        'thumbnail': 'test1/11.jpg',
        'medias': [],
    },
    'dir1/test1': {
        'title': 'An example sub-category',
        'thumbnail': '11.jpg',
        'medias': ['11.jpg', 'archlinux-kiss-1024x640.png'],
    },
    'dir1/test2': {
        'title': 'Test2',
        'thumbnail': '21.jpg',
        'medias': ['21.jpg', '22.jpg'],
    },
    'dir2': {
        'title': 'Another example gallery',
        'thumbnail': 'm57_the_ring_nebula-587px.jpg',
        'medias': ['exo20101028-b-full.jpg',
                'm57_the_ring_nebula-587px.jpg',
                'Hubble ultra deep field.jpg',
                'Hubble Interacting Galaxy NGC 5257.jpg'],
    },
    u'accentué': {
        'title': u'Accentué',
        'thumbnail': u'hélicoïde.jpg',
        'medias': [u'hélicoïde.jpg', 'superdupont_source_wikipedia_en.jpg'],
    },
    'video': {
        'title': 'Video',
        'thumbnail': 'stallman-software-freedom-day-low.ogv',
        'medias': ['stallman-software-freedom-day-low.ogv']
    }
}


@pytest.fixture(scope='module')
def paths():
    """Read the sample config file and build the PathsDb object."""

    default_conf = os.path.join(SAMPLE_DIR, 'sigal.conf.py')
    settings = read_settings(default_conf)
    return PathsDb(os.path.join(SAMPLE_DIR, 'pictures'),
            settings['img_ext_list'], settings['vid_ext_list'])


@pytest.fixture(scope='module')
def db(paths):
    paths.build()
    return paths.db


def test_filelist(db):
    assert set(db.keys()) == set(['paths_list', 'skipped_dir', '.',
        'dir1', 'dir2', 'dir1/test1', 'dir1/test2', u'accentué', 'video'])

    assert set(db['paths_list']) == set(['.', 'dir1', 'dir1/test1',
        'dir1/test2', 'dir2', u'accentué', 'video'])

    assert set(db['skipped_dir']) == set(['empty', 'dir1/empty'])
    assert db['.']['medias'] == []
    assert set(db['.']['subdir']) == set([u'accentué', 'dir1', 'dir2',
        'video'])


def test_title(db):
    for p in REF.keys():
        assert db[p]['title'] == REF[p]['title']


def test_thumbnail(db):
    for p in REF.keys():
        assert db[p]['thumbnail'] == REF[p]['thumbnail']


def test_medialist(db):
    for p in REF.keys():
        assert set(db[p]['medias']) == set(REF[p]['medias'])


def test_get_subdir(paths):
    assert set(paths.get_subdirs('dir1')) == set(['dir1/test1', 'dir1/test2'])
    assert set(paths.get_subdirs('.')) == set(['dir1', 'dir2', 'dir1/test1',
                                               'dir1/test2', u'accentué',
                                               'video'])


def test_get_metadata():
    "Test the get_metadata function."

    m = get_metadata(os.path.join(SAMPLE_DIR, 'pictures', 'dir1'))
    assert m['title'] == REF['dir1']['title']
    assert m['thumbnail'] == ''

    m = get_metadata(os.path.join(SAMPLE_DIR, 'pictures', 'dir2'))
    assert m['title'] == REF['dir2']['title']
    assert m['thumbnail'] == REF['dir2']['thumbnail']


# class TestGallery(unittest.TestCase):
#     "Test the Gallery class."

#     @classmethod
#     def setUp(cls):
#         """Read the sample config file."""

#         default_conf = os.path.join(CURRENT_DIR, 'sample', 'sigal.conf.py')
#         settings = read_settings(default_conf)
#         cls.gal = Gallery(settings, os.path.join(CURRENT_DIR, 'sample'),
#                           os.path.join(CURRENT_DIR, 'output'))