45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
|
||
|
|
import pytest
|
||
|
|
from pathlib import Path
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
from dozo import constants
|
||
|
|
from dozo.media.media import Media
|
||
|
|
from dozo.media.audio import Audio
|
||
|
|
from dozo.media.photo import Photo
|
||
|
|
from dozo.media.video import Video
|
||
|
|
from dozo.exiftool import ExifToolCaching
|
||
|
|
|
||
|
|
DOZO_PATH = Path(__file__).parent.parent
|
||
|
|
|
||
|
|
class TestMetadata:
|
||
|
|
|
||
|
|
def setup_class(cls):
|
||
|
|
cls.SRCPATH = tempfile.mkdtemp(prefix='dozo-src')
|
||
|
|
filenames = ['invalid.jpg', 'photo.png', 'plain.jpg', 'text.txt', 'withoutextension']
|
||
|
|
cls.file_paths = set()
|
||
|
|
for filename in filenames:
|
||
|
|
source_path = Path(cls.SRCPATH, filename)
|
||
|
|
file_path = Path(DOZO_PATH, 'samples', filename)
|
||
|
|
shutil.copyfile(file_path, source_path)
|
||
|
|
cls.file_paths.add(source_path)
|
||
|
|
cls.path_format = constants.default_path + '/' + constants.default_name
|
||
|
|
|
||
|
|
def test_get_exiftool_attribute(self, tmp_path):
|
||
|
|
for file_path in self.file_paths:
|
||
|
|
exif_data = ExifToolCaching(str(file_path)).asdict()
|
||
|
|
ignore_tags = ('File:FileModifyDate', 'File:FileAccessDate')
|
||
|
|
exif_data_filtered = {}
|
||
|
|
for key in exif_data:
|
||
|
|
if key not in ignore_tags:
|
||
|
|
exif_data_filtered[key] = exif_data[key]
|
||
|
|
media = Media(str(file_path), ignore_tags)
|
||
|
|
exif = media.get_exiftool_attributes()
|
||
|
|
# Ensure returned value is a dictionary
|
||
|
|
assert isinstance(exif, dict)
|
||
|
|
for tag in ignore_tags:
|
||
|
|
assert tag not in exif
|
||
|
|
assert exif == exif_data_filtered
|
||
|
|
|