mirror of https://github.com/saimn/sigal.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
212 lines
6.8 KiB
212 lines
6.8 KiB
# -*- coding:utf-8 -*- |
|
|
|
import locale |
|
import os |
|
import pytest |
|
|
|
from os.path import join |
|
from sigal.gallery import Album, Media, Image, Video, Gallery |
|
|
|
CURRENT_DIR = os.path.dirname(__file__) |
|
|
|
REF = { |
|
'dir1': { |
|
'title': 'An example gallery', |
|
'name': 'dir1', |
|
'thumbnail': 'dir1/test1/thumbnails/11.tn.jpg', |
|
'subdirs': ['test1', 'test2'], |
|
'medias': [], |
|
}, |
|
'dir1/test1': { |
|
'title': 'An example sub-category', |
|
'name': 'test1', |
|
'thumbnail': 'test1/thumbnails/11.tn.jpg', |
|
'subdirs': [], |
|
'medias': ['11.jpg', 'archlinux-kiss-1024x640.png', |
|
'flickr_jerquiaga_2394751088_cc-by-nc.jpg'], |
|
}, |
|
'dir1/test2': { |
|
'title': 'test2', |
|
'name': 'test2', |
|
'thumbnail': 'test2/thumbnails/21.tn.jpg', |
|
'subdirs': [], |
|
'medias': ['21.jpg', '22.jpg', 'archlinux-kiss-1024x640.png'], |
|
}, |
|
'dir2': { |
|
'title': 'Another example gallery with a very long name', |
|
'name': 'dir2', |
|
'thumbnail': 'dir2/thumbnails/m57_the_ring_nebula-587px.tn.jpg', |
|
'subdirs': [], |
|
'medias': ['exo20101028-b-full.jpg', |
|
'Hubble Interacting Galaxy NGC 5257.jpg', |
|
'Hubble ultra deep field.jpg', |
|
'm57_the_ring_nebula-587px.jpg'], |
|
}, |
|
u'accentué': { |
|
'title': u'accentué', |
|
'name': u'accentué', |
|
'thumbnail': u'accentué/thumbnails/hélicoïde.tn.jpg', |
|
'subdirs': [], |
|
'medias': [u'hélicoïde.jpg', 'superdupont_source_wikipedia_en.jpg'], |
|
}, |
|
'video': { |
|
'title': 'video', |
|
'name': 'video', |
|
'thumbnail': ('video/thumbnails/' |
|
'stallman software-freedom-day-low.tn.jpg'), |
|
'subdirs': [], |
|
'medias': ['stallman software-freedom-day-low.webm'] |
|
} |
|
} |
|
|
|
|
|
def test_media(settings): |
|
m = Media('11.jpg', 'dir1/test1', settings) |
|
path = join('dir1', 'test1') |
|
file_path = join(path, '11.jpg') |
|
thumb = join('thumbnails', '11.tn.jpg') |
|
|
|
assert m.filename == '11.jpg' |
|
assert m.src_path == join(settings['source'], file_path) |
|
assert m.dst_path == join(settings['destination'], file_path) |
|
assert m.thumb_name == thumb |
|
assert m.thumb_path == join(settings['destination'], path, thumb) |
|
assert m.title == "Foo Bar" |
|
assert m.description == "<p>This is a funny description of this image</p>" |
|
|
|
assert repr(m) == "<Media>('{}')".format(file_path) |
|
assert str(m) == file_path |
|
|
|
|
|
def test_media_orig(settings): |
|
settings['keep_orig'] = False |
|
m = Media('11.jpg', 'dir1/test1', settings) |
|
assert m.big is None |
|
|
|
settings['keep_orig'] = True |
|
m = Image('11.jpg', 'dir1/test1', settings) |
|
assert m.big == 'original/11.jpg' |
|
|
|
m = Video('file.ogv', 'video', settings) |
|
assert m.filename == 'file.webm' |
|
assert m.big == 'original/file.ogv' |
|
|
|
|
|
def test_image(settings, tmpdir): |
|
settings['destination'] = str(tmpdir) |
|
m = Image('11.jpg', 'dir1/test1', settings) |
|
assert m.exif['datetime'] == u'Sunday, 22. January 2006' |
|
|
|
os.makedirs(join(settings['destination'], 'dir1', 'test1', 'thumbnails')) |
|
assert m.thumbnail == join('thumbnails', '11.tn.jpg') |
|
assert os.path.isfile(m.thumb_path) |
|
|
|
|
|
def test_video(settings, tmpdir): |
|
settings['destination'] = str(tmpdir) |
|
m = Video('stallman software-freedom-day-low.ogv', 'video', settings) |
|
file_path = join('video', 'stallman software-freedom-day-low.webm') |
|
assert str(m) == file_path |
|
assert m.dst_path == join(settings['destination'], file_path) |
|
|
|
os.makedirs(join(settings['destination'], 'video', 'thumbnails')) |
|
assert m.thumbnail == join('thumbnails', |
|
'stallman software-freedom-day-low.tn.jpg') |
|
assert os.path.isfile(m.thumb_path) |
|
|
|
|
|
@pytest.mark.parametrize("path,album", REF.items()) |
|
def test_album(path, album, settings, tmpdir): |
|
# store current locale |
|
old_locale = locale.setlocale(locale.LC_ALL) |
|
|
|
# locale.setlocale(locale.LC_ALL, 'fr_FR') |
|
locale.setlocale(locale.LC_ALL, 'fr_FR.UTF-8') |
|
|
|
gal = Gallery(settings, ncpu=1) |
|
a = Album(path, settings, album['subdirs'], album['medias'], gal) |
|
|
|
assert a.title == album['title'] |
|
assert a.name == album['name'] |
|
assert a.subdirs == album['subdirs'] |
|
assert a.thumbnail == album['thumbnail'] |
|
assert [m.filename for m in a.medias] == album['medias'] |
|
assert len(a) == len(album['medias']) |
|
|
|
# restore locale back |
|
locale.setlocale(locale.LC_ALL, old_locale) |
|
|
|
|
|
def test_album_medias(settings): |
|
gal = Gallery(settings, ncpu=1) |
|
|
|
album = REF['dir1/test1'] |
|
a = Album('dir1/test1', settings, album['subdirs'], album['medias'], gal) |
|
assert list(im.filename for im in a.images) == album['medias'] |
|
assert list(a.videos) == [] |
|
|
|
album = REF['video'] |
|
a = Album('video', settings, album['subdirs'], album['medias'], gal) |
|
assert list(im.filename for im in a.videos) == album['medias'] |
|
assert list(a.images) == [] |
|
|
|
|
|
def test_albums_sort(settings): |
|
gal = Gallery(settings, ncpu=1) |
|
album = REF['dir1'] |
|
|
|
settings['albums_sort_reverse'] = True |
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal) |
|
assert [im.filename for im in a.images] == list(reversed(album['medias'])) |
|
|
|
|
|
def test_medias_sort(settings): |
|
gal = Gallery(settings, ncpu=1) |
|
album = REF['dir1/test2'] |
|
|
|
settings['medias_sort_reverse'] = True |
|
a = Album('dir1/test2', settings, album['subdirs'], album['medias'], gal) |
|
assert [im.filename for im in a.images] == list(reversed(album['medias'])) |
|
|
|
settings['medias_sort_attr'] = 'date' |
|
settings['medias_sort_reverse'] = False |
|
a = Album('dir1/test2', settings, album['subdirs'], album['medias'], gal) |
|
assert [im.filename for im in a.images] == ['22.jpg', '21.jpg', |
|
'archlinux-kiss-1024x640.png'] |
|
|
|
|
|
def test_gallery(settings, tmpdir): |
|
"Test the Gallery class." |
|
|
|
settings['destination'] = str(tmpdir) |
|
gal = Gallery(settings, ncpu=1) |
|
gal.build() |
|
|
|
out_html = os.path.join(settings['destination'], 'index.html') |
|
assert os.path.isfile(out_html) |
|
|
|
with open(out_html, 'r') as f: |
|
html = f.read() |
|
|
|
assert '<title>Sigal test gallery</title>' in html |
|
|
|
|
|
def test_empty_dirs(settings): |
|
gal = Gallery(settings, ncpu=1) |
|
assert 'empty' not in gal.albums |
|
assert 'dir1/empty' not in gal.albums |
|
|
|
|
|
def test_ignores(settings, tmpdir): |
|
tmp = str(tmpdir) |
|
settings['destination'] = tmp |
|
settings['ignore_directories'] = ['*test2'] |
|
settings['ignore_files'] = ['dir2/Hubble*', '*.png'] |
|
gal = Gallery(settings, ncpu=1) |
|
gal.build() |
|
|
|
assert 'test2' not in os.listdir(join(tmp, 'dir1')) |
|
assert 'archlinux-kiss-1024x640.png' not in os.listdir( |
|
join(tmp, 'dir1', 'test1')) |
|
assert 'Hubble Interacting Galaxy NGC 5257.jpg' not in os.listdir( |
|
join(tmp, 'dir2'))
|
|
|