70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
from datetime import datetime
|
|
|
|
import os
|
|
import time
|
|
import piexif
|
|
import pytest
|
|
import random
|
|
import shutil
|
|
|
|
from helpers import *
|
|
from fixtures import *
|
|
|
|
cmd = 'python3 updatescannedpics.py'
|
|
|
|
def test_diff_in_out(in_out_dirs):
|
|
in_dir, out_dir = in_out_dirs
|
|
|
|
year = random.randint(1900, 2000)
|
|
month = random.randint(1, 12)
|
|
day = random.randint(1, 28)
|
|
index = random.randint(0, 99)
|
|
|
|
file_name = f"{year}{month:02d}{day:02d}_{index:02d}.jpg"
|
|
in_file = os.path.join(in_dir, file_name)
|
|
out_file = os.path.join(out_dir, file_name)
|
|
|
|
shutil.copyfile("pytest/example_pics/u_001.jpg", in_file)
|
|
|
|
command = f"{cmd} -i {in_dir} -o {out_dir}"
|
|
print(command)
|
|
output = run_successfull_cmd(command)
|
|
|
|
exif_dict = piexif.load(out_file)
|
|
|
|
test_date = datetime(year, month, day, 0, 0, 0)
|
|
test_date_str = test_date.strftime("%Y:%m:%d %H:%M:%S")
|
|
|
|
assert test_date_str == exif_dict["0th"][piexif.ImageIFD.DateTime].decode('utf-8')
|
|
assert test_date_str == exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal].decode('utf-8')
|
|
assert test_date_str == exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal].decode('utf-8')
|
|
|
|
def test_same_in_out(in_out_dirs):
|
|
in_dir, out_dir = in_out_dirs
|
|
out_dir = in_dir
|
|
|
|
year = random.randint(1900, 2000)
|
|
month = random.randint(1, 12)
|
|
day = random.randint(1, 28)
|
|
index = random.randint(0, 99)
|
|
|
|
file_name = f"{year}{month:02d}{day:02d}_{index:02d}.jpg"
|
|
in_file = out_file = os.path.join(in_dir, file_name)
|
|
|
|
shutil.copyfile("pytest/example_pics/u_001.jpg", in_file)
|
|
|
|
command = f"{cmd} -i {in_dir} -o {out_dir}"
|
|
print(command)
|
|
output = run_successfull_cmd(command)
|
|
|
|
exif_dict = piexif.load(out_file)
|
|
|
|
test_date = datetime(year, month, day, 0, 0, 0)
|
|
test_date_str = test_date.strftime("%Y:%m:%d %H:%M:%S")
|
|
|
|
assert test_date_str == exif_dict["0th"][piexif.ImageIFD.DateTime].decode('utf-8')
|
|
assert test_date_str == exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal].decode('utf-8')
|
|
assert test_date_str == exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal].decode('utf-8')
|
|
|
|
|