mirror of https://github.com/gmbrax/Pilgrim.git
Add the tests for the hash methods and the create methods
This commit is contained in:
parent
d840299398
commit
2f81e4ff06
|
|
@ -0,0 +1,56 @@
|
|||
import pytest
|
||||
from pathlib import Path
|
||||
from pilgrim.service.photo_service import PhotoService
|
||||
import hashlib
|
||||
from unittest.mock import patch
|
||||
from pilgrim.models.photo import Photo
|
||||
|
||||
@patch.object(PhotoService, '_copy_photo_to_diary')
|
||||
@patch.object(PhotoService, 'hash_file', return_value="fake_hash_123")
|
||||
def test_create_photo_successfully(mock_hash, mock_copy, session_with_one_diary):
|
||||
session, diary = session_with_one_diary
|
||||
service = PhotoService(session)
|
||||
fake_source_path = Path("/path/original/imagem.jpg")
|
||||
fake_copied_path = Path(f"~/.pilgrim/diaries/{diary.directory_name}/images/imagem.jpg")
|
||||
mock_copy.return_value = fake_copied_path
|
||||
new_photo = service.create(
|
||||
filepath=fake_source_path,
|
||||
name="Foto da Praia",
|
||||
travel_diary_id=diary.id,
|
||||
caption="Pôr do sol")
|
||||
mock_hash.assert_called_once_with(fake_source_path)
|
||||
mock_copy.assert_called_once_with(fake_source_path, diary)
|
||||
assert new_photo is not None
|
||||
assert new_photo.name == "Foto da Praia"
|
||||
assert new_photo.photo_hash == "fake_hash_123"
|
||||
assert new_photo.filepath == str(fake_copied_path)
|
||||
|
||||
def test_hash_file_generates_correct_hash(tmp_path: Path):
|
||||
original_content_bytes = b"um conteudo de teste para o hash"
|
||||
file_on_disk = tmp_path / "test.jpg"
|
||||
file_on_disk.write_bytes(original_content_bytes)
|
||||
hash_from_file = PhotoService.hash_file(file_on_disk)
|
||||
expected_hash_func = hashlib.new('sha3_384')
|
||||
expected_hash_func.update(original_content_bytes)
|
||||
hash_from_memory = expected_hash_func.hexdigest()
|
||||
assert hash_from_file == hash_from_memory
|
||||
|
||||
@patch.object(PhotoService, '_copy_photo_to_diary')
|
||||
@patch.object(PhotoService, 'hash_file', return_value="hash_ja_existente")
|
||||
def test_create_photo_returns_none_if_hash_exists(mock_hash, mock_copy, session_with_one_diary):
|
||||
session, diary = session_with_one_diary
|
||||
existing_photo = Photo(
|
||||
filepath="/path/existente.jpg", name="Foto Antiga",
|
||||
photo_hash="hash_ja_existente", fk_travel_diary_id=diary.id
|
||||
)
|
||||
session.add(existing_photo)
|
||||
session.commit()
|
||||
|
||||
service = PhotoService(session)
|
||||
new_photo = service.create(
|
||||
filepath=Path("/path/novo/arquivo.jpg"),
|
||||
name="Foto Nova",
|
||||
travel_diary_id=diary.id
|
||||
)
|
||||
assert new_photo is None
|
||||
mock_copy.assert_not_called()
|
||||
Loading…
Reference in New Issue