Passed
Push — master ( b2d36d...e1a621 )
by Jaisen
02:25
created

elodie/tests/media/base_test.py (2 issues)

1
# Project imports
2
import os
3
import sys
4
5
import hashlib
6
import random
7
import re
8
import shutil
9
import string
10
import tempfile
11
import time
12
13
sys.path.insert(0, os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))))))
14
sys.path.insert(0, os.path.abspath(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))))
15
16
import helper
17
from elodie.media.base import Base, get_all_subclasses
18
from elodie.media.media import Media
19
from elodie.media.audio import Audio
20
from elodie.media.text import Text
21
from elodie.media.photo import Photo
22
from elodie.media.video import Video
23
24
os.environ['TZ'] = 'GMT'
25
26
27
def test_get_all_subclasses():
28
    subclasses = get_all_subclasses(Base)
29
    expected = {Media, Base, Text, Photo, Video, Audio}
30
    assert subclasses == expected, subclasses
31
32
def test_get_class_by_file_without_extension():
33
    base_file = helper.get_file('withoutextension')
34
35
    cls = Base.get_class_by_file(base_file, [Audio, Text, Photo, Video])
36
    
37
    assert cls is None, cls
38
39 View Code Duplication
def test_get_original_name():
0 ignored issues
show
This code seems to be duplicated in your project.
Loading history...
40
    temporary_folder, folder = helper.create_working_folder()
41
42
    origin = '%s/%s' % (folder, 'with-original-name.jpg')
43
    file = helper.get_file('with-original-name.jpg')
44
    
45
    shutil.copyfile(file, origin)
46
47
    media = Media.get_class_by_file(origin, [Photo])
48
    original_name = media.get_original_name()
49
50
    assert original_name == 'originalfilename.jpg', original_name
51
52 View Code Duplication
def test_get_original_name_invalid_file():
0 ignored issues
show
This code seems to be duplicated in your project.
Loading history...
53
    temporary_folder, folder = helper.create_working_folder()
54
55
    origin = '%s/%s' % (folder, 'invalid.jpg')
56
    file = helper.get_file('invalid.jpg')
57
    
58
    shutil.copyfile(file, origin)
59
60
    media = Media.get_class_by_file(origin, [Photo])
61
    original_name = media.get_original_name()
62
63
    assert original_name is None, original_name
64
65
def test_set_album_from_folder_invalid_file():
66
    temporary_folder, folder = helper.create_working_folder()
67
68
    base_file = helper.get_file('invalid.jpg')
69
    origin = '%s/invalid.jpg' % folder
70
71
    shutil.copyfile(base_file, origin)
72
73
    base = Base(origin)
74
75
    status = base.set_album_from_folder()
76
77
    assert status == False, status
78
79
def test_set_album_from_folder():
80
    temporary_folder, folder = helper.create_working_folder()
81
82
    origin = '%s/photo.jpg' % folder
83
    shutil.copyfile(helper.get_file('plain.jpg'), origin)
84
85
    photo = Photo(origin)
86
    metadata = photo.get_metadata()
87
88
    assert metadata['album'] is None, metadata['album']
89
90
    new_album_name = os.path.split(folder)[1]
91
    status = photo.set_album_from_folder()
92
93
    assert status == True, status
94
95
    photo_new = Photo(origin)
96
    metadata_new = photo_new.get_metadata()
97
98
    shutil.rmtree(folder)
99
100
    assert metadata_new['album'] == new_album_name, metadata_new['album']
101
102
def test_set_metadata():
103
    temporary_folder, folder = helper.create_working_folder()
104
105
    origin = '%s/photo.jpg' % folder
106
    shutil.copyfile(helper.get_file('plain.jpg'), origin)
107
108
    photo = Photo(origin)
109
110
    metadata = photo.get_metadata()
111
112
    assert metadata['title'] == None, metadata['title']
113
114
    new_title = 'Some Title'
115
    photo.set_metadata(title = new_title)
116
117
    new_metadata = photo.get_metadata()
118
119
    assert new_metadata['title'] == new_title, new_metadata['title']
120
121
def test_set_metadata_basename():
122
    temporary_folder, folder = helper.create_working_folder()
123
124
    origin = '%s/photo.jpg' % folder
125
    shutil.copyfile(helper.get_file('plain.jpg'), origin)
126
127
    photo = Photo(origin)
128
129
    metadata = photo.get_metadata()
130
131
    assert metadata['base_name'] == 'photo', metadata['base_name']
132
133
    new_basename = 'Some Base Name'
134
    photo.set_metadata_basename(new_basename)
135
136
    new_metadata = photo.get_metadata()
137
138
    assert new_metadata['base_name'] == new_basename, new_metadata['base_name']
139