256 lines
7.1 KiB
Python
256 lines
7.1 KiB
Python
import shutil
|
|
from click.testing import CliRunner
|
|
from pathlib import Path
|
|
import pytest
|
|
import inquirer
|
|
|
|
from ordigi import cli
|
|
|
|
CONTENT = "content"
|
|
|
|
ORDIGI_PATH = Path(__file__).parent.parent
|
|
|
|
|
|
def get_arg_options_list(arg_options):
|
|
arg_options_list = []
|
|
for opt, arg in arg_options:
|
|
arg_options_list.append(opt)
|
|
arg_options_list.append(arg)
|
|
|
|
return arg_options_list
|
|
|
|
|
|
class TestOrdigi:
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_class(cls, sample_files_paths):
|
|
cls.runner = CliRunner()
|
|
cls.src_path, cls.file_paths = sample_files_paths
|
|
cls.logger_options = (('--verbose', 'DEBUG'),)
|
|
cls.filter_options = (
|
|
('--ignore-tags', 'CreateDate'),
|
|
('--ext', 'jpg'),
|
|
('--glob', '*'),
|
|
)
|
|
cls.sort_options = (
|
|
'--album-from-folder',
|
|
'--fill-date-original',
|
|
'--path-format',
|
|
'--remove-duplicates',
|
|
'--use-date-filename',
|
|
'--use-file-dates',
|
|
)
|
|
|
|
def assert_cli(self, command, attributes, state=0):
|
|
result = self.runner.invoke(command, [*attributes])
|
|
assert result.exit_code == state, (command, attributes)
|
|
|
|
def assert_options(self, command, bool_options, arg_options, paths):
|
|
for bool_option in bool_options:
|
|
self.assert_cli(command, [bool_option, *paths])
|
|
|
|
for opt, arg in arg_options:
|
|
self.assert_cli(command, [opt, arg, *paths])
|
|
|
|
def assert_all_options(self, command, bool_options, arg_options, paths):
|
|
arg_options_list = get_arg_options_list(arg_options)
|
|
self.assert_cli(command, [
|
|
*bool_options, *arg_options_list, *paths,
|
|
])
|
|
|
|
def test_commands(self):
|
|
# Check if fail if path not exist
|
|
commands = [
|
|
cli._check,
|
|
cli._clean,
|
|
cli._compare,
|
|
cli._edit,
|
|
cli._import,
|
|
cli._init,
|
|
cli._sort,
|
|
cli._update,
|
|
]
|
|
|
|
for command in commands:
|
|
if command.name == 'edit':
|
|
self.assert_cli(command, ['-k', 'date_original', 'not_exist'], state=1)
|
|
else:
|
|
self.assert_cli(command, ['not_exist'], state=1)
|
|
|
|
self.assert_cli(cli._clone, ['not_exist'], state=2)
|
|
|
|
def test_edit(self, monkeypatch):
|
|
|
|
bool_options = ()
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
*self.filter_options,
|
|
)
|
|
|
|
def mockreturn(prompt, theme):
|
|
return {'value': '03-12-2021 08:12:35'}
|
|
|
|
monkeypatch.setattr(inquirer, 'prompt', mockreturn)
|
|
|
|
args = (
|
|
'--key',
|
|
'date_original',
|
|
'--overwrite',
|
|
str(self.src_path.joinpath('test_exif/photo.png')),
|
|
str(self.src_path),
|
|
)
|
|
|
|
self.assert_cli(cli._edit, args)
|
|
|
|
# self.assert_options(cli._edit, bool_options, arg_options, args)
|
|
# self.assert_all_options(cli._edit, bool_options, arg_options, args)
|
|
|
|
def test_sort(self):
|
|
bool_options = (
|
|
# '--interactive',
|
|
'--dry-run',
|
|
'--album-from-folder',
|
|
'--remove-duplicates',
|
|
'--use-date-filename',
|
|
'--use-file-dates',
|
|
'--clean',
|
|
)
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
*self.filter_options,
|
|
('--path-format', '{%Y}/{folder}/{name}.{ext}'),
|
|
|
|
)
|
|
|
|
paths = (str(self.src_path),)
|
|
|
|
self.assert_cli(cli._sort, paths)
|
|
|
|
self.assert_options(cli._sort, bool_options, arg_options, paths)
|
|
self.assert_all_options(cli._sort, bool_options, arg_options, paths)
|
|
|
|
def test_clone(self, tmp_path):
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
|
|
)
|
|
|
|
paths = (str(self.src_path), str(tmp_path))
|
|
|
|
self.assert_cli(cli._init, [str(self.src_path)])
|
|
self.assert_cli(cli._clone, ['--dry-run', '--verbose', 'DEBUG', *paths])
|
|
self.assert_cli(cli._clone, paths)
|
|
|
|
|
|
def assert_init(self):
|
|
for opt, arg in self.logger_options:
|
|
self.assert_cli(cli._init, [opt, arg, str(self.src_path)])
|
|
|
|
def assert_update(self):
|
|
file_path = Path(ORDIGI_PATH, 'samples/test_exif/photo.cr2')
|
|
dest_path = self.src_path / 'photo_moved.cr2'
|
|
shutil.copyfile(file_path, dest_path)
|
|
for opt, arg in self.logger_options:
|
|
self.assert_cli(cli._update, [opt, arg, str(self.src_path)])
|
|
|
|
def assert_check(self):
|
|
for opt, arg in self.logger_options:
|
|
self.assert_cli(cli._check, [opt, arg, str(self.src_path)])
|
|
|
|
def assert_clean(self):
|
|
bool_options = (
|
|
# '--interactive',
|
|
'--dry-run',
|
|
'--delete-excluded',
|
|
'--folders',
|
|
'--path-string',
|
|
'--remove-duplicates',
|
|
)
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
*self.filter_options,
|
|
('--dedup-regex', r'\d{4}-\d{2}'),
|
|
)
|
|
|
|
paths = ('test_exif', str(self.src_path))
|
|
self.assert_cli(cli._clean, paths)
|
|
|
|
paths = (str(self.src_path),)
|
|
self.assert_cli(cli._clean, paths)
|
|
|
|
self.assert_options(cli._clean, bool_options, arg_options, paths)
|
|
self.assert_all_options(cli._clean, bool_options, arg_options, paths)
|
|
|
|
def test_init_update_check_clean(self):
|
|
self.assert_init()
|
|
self.assert_update()
|
|
self.assert_check()
|
|
self.assert_clean()
|
|
|
|
def test_import(self, tmp_path):
|
|
bool_options = (
|
|
# '--interactive',
|
|
'--dry-run',
|
|
'--album-from-folder',
|
|
'--remove-duplicates',
|
|
'--use-date-filename',
|
|
'--use-file-dates',
|
|
'--copy',
|
|
)
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
('--exclude', '.DS_Store'),
|
|
*self.filter_options,
|
|
('--path-format', '{%Y}/{folder}/{stem}.{ext}'),
|
|
|
|
)
|
|
|
|
paths = (str(self.src_path), str(tmp_path))
|
|
|
|
result = self.runner.invoke(cli._import, ['--copy', *paths])
|
|
assert result.exit_code == 0
|
|
|
|
self.assert_options(cli._import, bool_options, arg_options, paths)
|
|
self.assert_all_options(cli._import, bool_options, arg_options, paths)
|
|
|
|
def test_compare(self):
|
|
bool_options = (
|
|
# '--interactive',
|
|
'--dry-run',
|
|
'--find-duplicates',
|
|
'--remove-duplicates',
|
|
)
|
|
|
|
arg_options = (
|
|
*self.logger_options,
|
|
*self.filter_options,
|
|
# ('--similar-to', ''),
|
|
('--similarity', '65'),
|
|
)
|
|
|
|
paths = (str(self.src_path),)
|
|
|
|
# Workaround
|
|
self.assert_cli(cli._update, paths)
|
|
|
|
self.assert_cli(cli._compare, paths)
|
|
self.assert_options(cli._compare, bool_options, arg_options, paths)
|
|
|
|
|
|
def test_needsfiles(tmpdir):
|
|
assert tmpdir
|
|
|
|
|
|
def test_create_file(tmp_path):
|
|
directory = tmp_path / "sub"
|
|
directory.mkdir()
|
|
path = directory / "hello.txt"
|
|
path.write_text(CONTENT)
|
|
assert path.read_text() == CONTENT
|
|
assert len(list(tmp_path.iterdir())) == 1
|