def test_get_hash(image):
hash_functions = cli.HASH_FUNCTIONS
for key in hash_functions:
hash_function = hash_functions[key]
hash = cli.get_hash(image, key)
assert type(hash) == str
assert hash == str(hash_function(image))
try:
hash = cli.get_hash(image, 'FOOBAR')
assert False, 'Unknown hash type should raise an exception'
except click.UsageError:
pass
def test_cli_file(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
with runner.isolated_filesystem():
image.save(filename, format='JPEG')
result = runner.invoke(cli.main, [filename])
assert result.exit_code == 0
assert result.output == imghash
os.remove(filename)
def test_cli_file_rename_dry_run(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
new_name = cli.get_new_name(filename, imghash)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main, [filename, '--rename', '--dry-run'])
assert result.exit_code == 0
assert not os.path.exists(new_name)
def test_cli_file_rename_bad_filename(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
template = '/tmp/test.jpg/not_a_directory'
new_name = cli.get_new_name(filename, imghash, template)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main,
[filename, '--rename', '--template', template])
assert result.exception
assert not os.path.exists(new_name)
def test_cli_file_rename_template(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
template = '{path}/FOOBAR-{name}-{hash}{ext}'
new_name = cli.get_new_name(filename, imghash, template)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main,
[filename, '--rename', '--template', template])
assert result.exit_code == 0
assert os.path.exists(new_name)
# clean up the new file
os.remove(new_name)
def test_cli_multiple(runner, image):
num_images = 5
filenames = ['/tmp/test-%d.jpg' % (n) for n in range(num_images)]
imghash = cli.get_hash(image)
with runner.isolated_filesystem():
for path in filenames:
if os.path.exists(path):
os.remove(path)
image.save(path, format='JPEG')
result = runner.invoke(cli.main, filenames)
assert result.exit_code == 0
expected_output = os.linesep.join(
['%s %s' % (path, imghash) for path in filenames])
assert result.output == expected_output
# clean up
for path in filenames:
os.remove(path)
def test_cli_multiple_rename(runner, image):
num_images = 5
filenames = ['/tmp/test-%d.jpg' % (n) for n in range(num_images)]
imghash = cli.get_hash(image)
with runner.isolated_filesystem():
for path in filenames:
if os.path.exists(path):
os.remove(path)
image.save(path, format='JPEG')
result = runner.invoke(cli.main, filenames + ['--rename'])
assert result.exit_code == 0
for path in filenames:
new_name = cli.get_new_name(path, imghash)
assert os.path.exists(new_name)
assert not os.path.exists(path)
# clean up
for path in filenames:
new_name = cli.get_new_name(path, imghash)
# hashes might be the same
if os.path.exists(new_name):
os.remove(new_name)
