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',
        'img': '',
    },
    'dir1/test1': {
        'title': 'An example sub-category',
        'thumbnail': '11.jpg',
        'img': ['11.jpg', 'archlinux-kiss-1024x640.png'],
    },
    'dir1/test2': {
        'title': 'Test2',
        'thumbnail': '21.jpg',
        'img': ['21.jpg', '22.jpg'],
    },
    'dir2': {
        'title': 'Another example gallery',
        'thumbnail': 'm57_the_ring_nebula-587px.jpg',
        'img': ['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',
        'img': [u'hélicoïde.jpg', 'superdupont_source_wikipedia_en.jpg']
    }
}


@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['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é'])

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

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


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_imglist(db):
    for p in REF.keys():
        assert set(db[p]['img']) == set(REF[p]['img'])


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é'])


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'))