2016-04-21 07:23:57 +02:00
|
|
|
# Project imports
|
|
|
|
|
from imp import load_source
|
|
|
|
|
import os
|
|
|
|
|
import sys
|
|
|
|
|
import shutil
|
|
|
|
|
|
2016-12-13 08:54:33 +01:00
|
|
|
from click.testing import CliRunner
|
2016-04-21 07:23:57 +02:00
|
|
|
from nose.plugins.skip import SkipTest
|
2016-12-13 08:54:33 +01:00
|
|
|
from nose.tools import assert_raises
|
2016-12-26 17:20:12 +01:00
|
|
|
from six import text_type, unichr as six_unichr
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))))
|
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))))))
|
|
|
|
|
|
|
|
|
|
import helper
|
|
|
|
|
elodie = load_source('elodie', os.path.abspath('{}/../../elodie.py'.format(os.path.dirname(os.path.realpath(__file__)))))
|
|
|
|
|
|
2016-12-13 08:54:33 +01:00
|
|
|
from elodie.localstorage import Db
|
2016-04-22 04:12:25 +02:00
|
|
|
from elodie.media.audio import Audio
|
|
|
|
|
from elodie.media.photo import Photo
|
|
|
|
|
from elodie.media.text import Text
|
|
|
|
|
from elodie.media.video import Video
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
os.environ['TZ'] = 'GMT'
|
|
|
|
|
|
|
|
|
|
def test_import_file_text():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
assert helper.path_tz_fix(os.path.join('2016-04-Apr','London','2016-04-07_11-15-26-valid-sample-title.txt')) in dest_path, dest_path
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
def test_import_file_audio():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/audio.m4a' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('audio.m4a'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2016-11-08 08:20:11 +01:00
|
|
|
assert helper.path_tz_fix(os.path.join('2016-01-Jan','Houston','2016-01-04_05-28-15-audio.m4a')) in dest_path, dest_path
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
def test_import_file_photo():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/plain.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('plain.jpg'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert helper.path_tz_fix(os.path.join('2015-12-Dec','Unknown Location','2015-12-05_00-59-26-plain.jpg')) in dest_path, dest_path
|
|
|
|
|
|
|
|
|
|
def test_import_file_video():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/video.mov' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('video.mov'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-21 07:23:57 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert helper.path_tz_fix(os.path.join('2015-01-Jan','California','2015-01-19_12-45-11-video.mov')) in dest_path, dest_path
|
|
|
|
|
|
2017-01-08 23:43:01 +01:00
|
|
|
def test_import_file_path_utf8_encoded_ascii():
|
2016-12-22 10:53:08 +01:00
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
2016-12-26 17:20:12 +01:00
|
|
|
origin = text_type(folder)+u'/unicode'+six_unichr(160)+u'filename.txt'
|
2017-01-08 23:43:01 +01:00
|
|
|
# encode the unicode string to ascii
|
2016-12-26 17:20:12 +01:00
|
|
|
origin = origin.encode('utf-8')
|
|
|
|
|
|
2016-12-22 10:53:08 +01:00
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-22 10:53:08 +01:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-22 10:53:08 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2017-01-08 23:43:01 +01:00
|
|
|
assert helper.path_tz_fix(os.path.join('2016-04-Apr','London',u'2016-04-07_11-15-26-unicode\xa0filename-sample-title.txt')) in dest_path, dest_path
|
|
|
|
|
|
|
|
|
|
def test_import_file_path_unicode():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = text_type(folder)+u'/unicode'+six_unichr(160)+u'filename.txt'
|
|
|
|
|
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
|
|
|
|
helper.reset_dbs()
|
|
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
|
|
|
|
helper.restore_dbs()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert helper.path_tz_fix(os.path.join('2016-04-Apr','London',u'2016-04-07_11-15-26-unicode\xa0filename-sample-title.txt')) in dest_path, dest_path
|
2016-12-22 10:53:08 +01:00
|
|
|
|
2016-09-08 06:08:13 +02:00
|
|
|
def test_import_file_allow_duplicate_false():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path1 = elodie.import_file(origin, folder_destination, False, False, False)
|
|
|
|
|
dest_path2 = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert dest_path1 is not None
|
|
|
|
|
assert dest_path2 is None
|
|
|
|
|
|
|
|
|
|
def test_import_file_allow_duplicate_true():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
dest_path1 = elodie.import_file(origin, folder_destination, False, False, True)
|
|
|
|
|
dest_path2 = elodie.import_file(origin, folder_destination, False, False, True)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-09-08 06:08:13 +02:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert dest_path1 is not None
|
|
|
|
|
assert dest_path2 is not None
|
|
|
|
|
assert dest_path1 == dest_path2
|
|
|
|
|
|
2016-12-23 06:35:15 +01:00
|
|
|
def test_import_file_send_to_trash_false():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-23 06:35:15 +01:00
|
|
|
dest_path1 = elodie.import_file(origin, folder_destination, False, False, False)
|
|
|
|
|
assert os.path.isfile(origin), origin
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-23 06:35:15 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert dest_path1 is not None
|
|
|
|
|
|
|
|
|
|
def test_import_file_send_to_trash_true():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-23 06:35:15 +01:00
|
|
|
dest_path1 = elodie.import_file(origin, folder_destination, False, True, False)
|
|
|
|
|
assert not os.path.isfile(origin), origin
|
2017-01-03 06:14:21 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-23 06:35:15 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert dest_path1 is not None
|
|
|
|
|
|
2016-11-08 08:09:23 +01:00
|
|
|
def test_import_destination_in_source():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
folder_destination = '{}/destination'.format(folder)
|
|
|
|
|
os.mkdir(folder_destination)
|
|
|
|
|
|
|
|
|
|
origin = '%s/video.mov' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('video.mov'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-11-08 08:09:23 +01:00
|
|
|
dest_path = elodie.import_file(origin, folder_destination, False, False, False)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-11-08 08:09:23 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
|
|
|
|
|
assert dest_path is None, dest_path
|
|
|
|
|
|
2017-01-22 05:52:32 +01:00
|
|
|
def test_import_invalid_file_exit_code():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
# use a good and bad
|
|
|
|
|
origin_invalid = '%s/invalid.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('invalid.jpg'), origin_invalid)
|
|
|
|
|
|
|
|
|
|
origin_valid = '%s/valid.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('plain.jpg'), origin_valid)
|
|
|
|
|
|
|
|
|
|
helper.reset_dbs()
|
|
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._import, ['--destination', folder_destination, origin_invalid, origin_valid])
|
|
|
|
|
helper.restore_dbs()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert result.exit_code == 1, result.exit_code
|
|
|
|
|
|
2016-04-22 04:12:25 +02:00
|
|
|
def test_update_location_on_audio():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/audio.m4a' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('audio.m4a'), origin)
|
|
|
|
|
|
|
|
|
|
audio = Audio(origin)
|
|
|
|
|
metadata = audio.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
status = elodie.update_location(audio, origin, 'Sunnyvale, CA')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
|
|
|
|
|
audio_processed = Audio(origin)
|
|
|
|
|
metadata_processed = audio_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2016-04-22 04:38:00 +02:00
|
|
|
assert status == True, status
|
2016-10-26 07:10:48 +02:00
|
|
|
assert metadata['latitude'] != metadata_processed['latitude'], metadata_processed['latitude']
|
2016-06-21 20:19:40 +02:00
|
|
|
assert helper.isclose(metadata_processed['latitude'], 37.36883), metadata_processed['latitude']
|
2016-04-22 04:12:25 +02:00
|
|
|
assert helper.isclose(metadata_processed['longitude'], -122.03635), metadata_processed['longitude']
|
|
|
|
|
|
|
|
|
|
def test_update_location_on_photo():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/plain.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('plain.jpg'), origin)
|
|
|
|
|
|
|
|
|
|
photo = Photo(origin)
|
|
|
|
|
metadata = photo.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
status = elodie.update_location(photo, origin, 'Sunnyvale, CA')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
|
|
|
|
|
photo_processed = Photo(origin)
|
|
|
|
|
metadata_processed = photo_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2016-04-22 04:38:00 +02:00
|
|
|
assert status == True, status
|
2016-04-22 04:12:25 +02:00
|
|
|
assert metadata['latitude'] != metadata_processed['latitude']
|
|
|
|
|
assert helper.isclose(metadata_processed['latitude'], 37.36883), metadata_processed['latitude']
|
|
|
|
|
assert helper.isclose(metadata_processed['longitude'], -122.03635), metadata_processed['longitude']
|
|
|
|
|
|
|
|
|
|
def test_update_location_on_text():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/text.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('text.txt'), origin)
|
|
|
|
|
|
|
|
|
|
text = Text(origin)
|
|
|
|
|
metadata = text.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
status = elodie.update_location(text, origin, 'Sunnyvale, CA')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
|
|
|
|
|
text_processed = Text(origin)
|
|
|
|
|
metadata_processed = text_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2016-04-22 04:38:00 +02:00
|
|
|
assert status == True, status
|
2016-04-22 04:12:25 +02:00
|
|
|
assert metadata['latitude'] != metadata_processed['latitude']
|
|
|
|
|
assert helper.isclose(metadata_processed['latitude'], 37.36883), metadata_processed['latitude']
|
|
|
|
|
assert helper.isclose(metadata_processed['longitude'], -122.03635), metadata_processed['longitude']
|
|
|
|
|
|
|
|
|
|
def test_update_location_on_video():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/video.mov' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('video.mov'), origin)
|
|
|
|
|
|
|
|
|
|
video = Video(origin)
|
|
|
|
|
metadata = video.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
status = elodie.update_location(video, origin, 'Sunnyvale, CA')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:12:25 +02:00
|
|
|
|
|
|
|
|
video_processed = Video(origin)
|
|
|
|
|
metadata_processed = video_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
2016-04-22 04:38:00 +02:00
|
|
|
assert status == True, status
|
2016-04-22 04:12:25 +02:00
|
|
|
assert metadata['latitude'] != metadata_processed['latitude']
|
2016-06-21 20:19:40 +02:00
|
|
|
assert helper.isclose(metadata_processed['latitude'], 37.36883), metadata_processed['latitude']
|
2016-04-22 04:12:25 +02:00
|
|
|
assert helper.isclose(metadata_processed['longitude'], -122.03635), metadata_processed['longitude']
|
|
|
|
|
|
2016-04-22 04:38:00 +02:00
|
|
|
def test_update_time_on_audio():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/audio.m4a' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('audio.m4a'), origin)
|
|
|
|
|
|
|
|
|
|
audio = Audio(origin)
|
|
|
|
|
metadata = audio.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
status = elodie.update_time(audio, origin, '2000-01-01 12:00:00')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
|
|
|
|
|
audio_processed = Audio(origin)
|
|
|
|
|
metadata_processed = audio_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert status == True, status
|
|
|
|
|
assert metadata['date_taken'] != metadata_processed['date_taken']
|
|
|
|
|
assert metadata_processed['date_taken'] == helper.time_convert((2000, 1, 1, 12, 0, 0, 5, 1, 0)), metadata_processed['date_taken']
|
|
|
|
|
|
|
|
|
|
def test_update_time_on_photo():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/plain.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('plain.jpg'), origin)
|
|
|
|
|
|
|
|
|
|
photo = Photo(origin)
|
|
|
|
|
metadata = photo.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
status = elodie.update_time(photo, origin, '2000-01-01 12:00:00')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
|
|
|
|
|
photo_processed = Photo(origin)
|
|
|
|
|
metadata_processed = photo_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert status == True, status
|
|
|
|
|
assert metadata['date_taken'] != metadata_processed['date_taken']
|
|
|
|
|
assert metadata_processed['date_taken'] == helper.time_convert((2000, 1, 1, 12, 0, 0, 5, 1, 0)), metadata_processed['date_taken']
|
|
|
|
|
|
|
|
|
|
def test_update_time_on_text():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/text.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('text.txt'), origin)
|
|
|
|
|
|
|
|
|
|
text = Text(origin)
|
|
|
|
|
metadata = text.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
status = elodie.update_time(text, origin, '2000-01-01 12:00:00')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
|
|
|
|
|
text_processed = Text(origin)
|
|
|
|
|
metadata_processed = text_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert status == True, status
|
|
|
|
|
assert metadata['date_taken'] != metadata_processed['date_taken']
|
|
|
|
|
assert metadata_processed['date_taken'] == helper.time_convert((2000, 1, 1, 12, 0, 0, 5, 1, 0)), metadata_processed['date_taken']
|
|
|
|
|
|
|
|
|
|
def test_update_time_on_video():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/video.mov' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('video.mov'), origin)
|
|
|
|
|
|
|
|
|
|
video = Video(origin)
|
|
|
|
|
metadata = video.get_metadata()
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
status = elodie.update_time(video, origin, '2000-01-01 12:00:00')
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-04-22 04:38:00 +02:00
|
|
|
|
|
|
|
|
video_processed = Video(origin)
|
|
|
|
|
metadata_processed = video_processed.get_metadata()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert status == True, status
|
|
|
|
|
assert metadata['date_taken'] != metadata_processed['date_taken']
|
|
|
|
|
assert metadata_processed['date_taken'] == helper.time_convert((2000, 1, 1, 12, 0, 0, 5, 1, 0)), metadata_processed['date_taken']
|
|
|
|
|
|
2017-01-08 06:49:00 +01:00
|
|
|
def test_update_with_directory_passed_in():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
|
|
|
|
helper.reset_dbs()
|
|
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._import, ['--destination', folder_destination, folder])
|
|
|
|
|
runner2 = CliRunner()
|
|
|
|
|
result = runner2.invoke(elodie._update, ['--album', 'test', folder_destination])
|
|
|
|
|
helper.restore_dbs()
|
|
|
|
|
|
|
|
|
|
updated_file_path = "{}/2016-04-Apr/test/2016-04-07_11-15-26-valid-sample-title.txt".format(folder_destination)
|
|
|
|
|
updated_file_exists = os.path.isfile(updated_file_path)
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert updated_file_exists, updated_file_path
|
|
|
|
|
|
2017-01-22 05:52:32 +01:00
|
|
|
def test_update_invalid_file_exit_code():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
temporary_folder_destination, folder_destination = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
# use a good and bad
|
|
|
|
|
origin_invalid = '%s/invalid.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('invalid.jpg'), origin_invalid)
|
|
|
|
|
|
|
|
|
|
origin_valid = '%s/valid.jpg' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('plain.jpg'), origin_valid)
|
|
|
|
|
|
|
|
|
|
helper.reset_dbs()
|
|
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._update, ['--album', 'test', origin_invalid, origin_valid])
|
|
|
|
|
helper.restore_dbs()
|
|
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
shutil.rmtree(folder_destination)
|
|
|
|
|
|
|
|
|
|
assert result.exit_code == 1, result.exit_code
|
|
|
|
|
|
2016-12-13 08:54:33 +01:00
|
|
|
def test_regenerate_db_invalid_source():
|
|
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._generate_db, ['--source', '/invalid/path'])
|
|
|
|
|
assert result.exit_code == 1, result.exit_code
|
|
|
|
|
|
|
|
|
|
def test_regenerate_valid_source():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-13 08:54:33 +01:00
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._generate_db, ['--source', folder])
|
|
|
|
|
db = Db()
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-13 08:54:33 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
|
|
|
|
|
assert result.exit_code == 0, result.exit_code
|
2017-01-03 05:58:52 +01:00
|
|
|
assert '3c19a5d751cf19e093b7447297731124d9cc987d3f91a9d1872c3b1c1b15639a' in db.hash_db, db.hash_db
|
2016-12-13 08:54:33 +01:00
|
|
|
|
|
|
|
|
def test_regenerate_valid_source_with_invalid_files():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin_valid = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin_valid)
|
|
|
|
|
origin_invalid = '%s/invalid.invalid' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('invalid.invalid'), origin_invalid)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-13 08:54:33 +01:00
|
|
|
runner = CliRunner()
|
|
|
|
|
result = runner.invoke(elodie._generate_db, ['--source', folder])
|
|
|
|
|
db = Db()
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-13 08:54:33 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
|
|
|
|
|
assert result.exit_code == 0, result.exit_code
|
2017-01-03 05:58:52 +01:00
|
|
|
assert '3c19a5d751cf19e093b7447297731124d9cc987d3f91a9d1872c3b1c1b15639a' in db.hash_db, db.hash_db
|
2016-12-13 08:54:33 +01:00
|
|
|
assert 'e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' not in db.hash_db, db.hash_db
|
|
|
|
|
|
2016-12-15 10:24:04 +01:00
|
|
|
def test_verify_ok():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-15 10:24:04 +01:00
|
|
|
runner = CliRunner()
|
|
|
|
|
runner.invoke(elodie._generate_db, ['--source', folder])
|
|
|
|
|
result = runner.invoke(elodie._verify)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-15 10:24:04 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
|
|
|
|
|
assert 'Success 1' in result.output, result.output
|
|
|
|
|
assert 'Error 0' in result.output, result.output
|
|
|
|
|
|
|
|
|
|
def test_verify_error():
|
|
|
|
|
temporary_folder, folder = helper.create_working_folder()
|
|
|
|
|
|
|
|
|
|
origin = '%s/valid.txt' % folder
|
|
|
|
|
shutil.copyfile(helper.get_file('valid.txt'), origin)
|
|
|
|
|
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.reset_dbs()
|
2016-12-15 10:24:04 +01:00
|
|
|
runner = CliRunner()
|
|
|
|
|
runner.invoke(elodie._generate_db, ['--source', folder])
|
|
|
|
|
with open(origin, 'w') as f:
|
|
|
|
|
f.write('changed text')
|
|
|
|
|
result = runner.invoke(elodie._verify)
|
2017-01-03 05:58:52 +01:00
|
|
|
helper.restore_dbs()
|
2016-12-15 10:24:04 +01:00
|
|
|
|
|
|
|
|
shutil.rmtree(folder)
|
|
|
|
|
|
|
|
|
|
assert origin in result.output, result.output
|
|
|
|
|
assert 'Error 1' in result.output, result.output
|
