from pathlib import Path from pilgrim import TravelDiary from pilgrim.service.photo_service import PhotoService import hashlib from unittest.mock import patch from pilgrim.models.photo import Photo from pilgrim.utils import DirectoryManager @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() def test_read_by_id_successfully(session_with_photos): session, photos = session_with_photos service = PhotoService(session) photo_to_find_id = photos[0].id found_photo = service.read_by_id(photo_to_find_id) assert found_photo is not None assert found_photo.id == photo_to_find_id assert found_photo.name == "Foto 1" def test_read_by_id_returns_none_for_invalid_id(db_session): service = PhotoService(db_session) result = service.read_by_id(999) assert result is None def test_read_all_returns_all_photos(session_with_photos): session, _ = session_with_photos service = PhotoService(session) all_photos = service.read_all() assert isinstance(all_photos, list) assert len(all_photos) == 2 assert all_photos[0].name == "Foto 1" assert all_photos[1].name == "Foto 2" def test_read_all_returns_empty_list_for_empty_db(db_session): service = PhotoService(db_session) all_photos = service.read_all() assert isinstance(all_photos, list) assert len(all_photos) == 0 def test_check_photo_by_hash_finds_existing_photo(session_with_photos): session, photos = session_with_photos service = PhotoService(session) existing_photo = photos[0] hash_to_find = existing_photo.photo_hash # "hash1" diary_id = existing_photo.fk_travel_diary_id # 1 found_photo = service.check_photo_by_hash(hash_to_find, diary_id) assert found_photo is not None assert found_photo.id == existing_photo.id assert found_photo.photo_hash == hash_to_find def test_check_photo_by_hash_returns_none_when_not_found(session_with_photos): session, photos = session_with_photos service = PhotoService(session) existing_hash = photos[0].photo_hash # "hash1" existing_diary_id = photos[0].fk_travel_diary_id # 1 result1 = service.check_photo_by_hash("hash_inexistente", existing_diary_id) assert result1 is None invalid_diary_id = 999 result2 = service.check_photo_by_hash(existing_hash, invalid_diary_id) assert result2 is None def test_update_photo_metadata_successfully(session_with_photos): session, photos = session_with_photos service = PhotoService(session) photo_to_update = photos[0] photo_with_new_data = Photo( filepath=photo_to_update.filepath, name="Novo Nome da Foto", caption="Nova legenda para a foto.", photo_hash=photo_to_update.photo_hash, # O hash não muda addition_date=photo_to_update.addition_date, fk_travel_diary_id=photo_to_update.fk_travel_diary_id ) updated_photo = service.update(photo_to_update, photo_with_new_data) assert updated_photo is not None assert updated_photo.name == "Novo Nome da Foto" assert updated_photo.caption == "Nova legenda para a foto." assert updated_photo.photo_hash == photo_to_update.photo_hash photo_in_db = session.query(Photo).get(photo_to_update.id) assert photo_in_db.name == "Novo Nome da Foto" @patch.object(PhotoService, 'hash_file') @patch('pathlib.Path.unlink') @patch('pathlib.Path.exists') @patch.object(PhotoService, '_copy_photo_to_diary') @patch.object(DirectoryManager, 'get_diaries_root', return_value="/fake/diaries_root") def test_update_photo_with_new_file_successfully( mock_get_root, mock_copy, mock_exists, mock_unlink, mock_hash, session_with_photos ): session, photos = session_with_photos service = PhotoService(session) photo_to_update = photos[0] new_source_path = Path("/path/para/nova_imagem.jpg") new_copied_path = Path(f"/fake/diaries_root/{photo_to_update.travel_diary.directory_name}/images/nova_imagem.jpg") mock_copy.return_value = new_copied_path mock_exists.return_value = True mock_hash.return_value = "novo_hash_calculado" photo_with_new_file = Photo( filepath=new_source_path, name=photo_to_update.name, photo_hash="hash_antigo", fk_travel_diary_id=photo_to_update.fk_travel_diary_id ) updated_photo = service.update(photo_to_update, photo_with_new_file) mock_copy.assert_called_once_with(new_source_path, photo_to_update.travel_diary) mock_unlink.assert_called_once() mock_hash.assert_called_once_with(new_copied_path) assert updated_photo.filepath == str(new_copied_path) assert updated_photo.photo_hash == "novo_hash_calculado" def test_update_photo_returns_none_if_photo_does_not_exist(db_session): service = PhotoService(db_session) non_existent_photo_src = Photo( filepath="/fake/path.jpg", name="dummy", photo_hash="dummy", fk_travel_diary_id=1 ) non_existent_photo_src.id = 999 photo_with_new_data = Photo( filepath="/fake/new.jpg", name="new dummy", photo_hash="new_dummy", fk_travel_diary_id=1 ) result = service.update(non_existent_photo_src, photo_with_new_data) assert result is None @patch.object(PhotoService, 'hash_file') @patch('pathlib.Path.unlink') @patch('pathlib.Path.exists') @patch.object(PhotoService, '_copy_photo_to_diary') @patch.object(DirectoryManager, 'get_diaries_root', return_value="/fake/diaries_root") def test_update_photo_with_new_file_successfully( mock_get_root, mock_copy, mock_exists, mock_unlink, mock_hash, session_with_photos ): session, photos = session_with_photos service = PhotoService(session) photo_to_update = photos[0] new_source_path = Path("/path/para/nova_imagem.jpg") new_copied_path = Path(f"/fake/diaries_root/{photo_to_update.travel_diary.directory_name}/images/nova_imagem.jpg") mock_copy.return_value = new_copied_path mock_exists.return_value = True mock_hash.return_value = "novo_hash_calculado" photo_with_new_file = Photo( filepath=new_source_path, name=photo_to_update.name, photo_hash="hash_antigo", fk_travel_diary_id=photo_to_update.fk_travel_diary_id ) updated_photo = service.update(photo_to_update, photo_with_new_file) mock_copy.assert_called_once_with(new_source_path, photo_to_update.travel_diary) mock_unlink.assert_called_once() mock_hash.assert_called_once_with(new_copied_path) assert updated_photo.filepath == str(new_copied_path) assert updated_photo.photo_hash == "novo_hash_calculado" @patch.object(DirectoryManager, 'get_diary_images_directory') def test_copy_photo_to_diary_handles_name_collision_with_patch(mock_get_images_dir, db_session, tmp_path: Path): images_dir = tmp_path / "images" images_dir.mkdir() mock_get_images_dir.return_value = images_dir source_file = tmp_path / "foto.jpg" source_file.touch() (images_dir / "foto.jpg").touch() service = PhotoService(db_session) fake_diary = TravelDiary(name="test",directory_name="fake_diary") copied_path = service._copy_photo_to_diary(source_file, fake_diary) assert copied_path.name == "foto_1.jpg" assert copied_path.exists() @patch('pathlib.Path.unlink') @patch('pathlib.Path.exists') @patch.object(DirectoryManager, 'get_diaries_root', return_value="/fake/diaries_root") def test_delete_photo_successfully(mock_get_root, mock_exists, mock_unlink, session_with_photos): session, photos = session_with_photos service = PhotoService(session) photo_to_delete = photos[0] photo_id = photo_to_delete.id mock_exists.return_value = True deleted_photo_data = service.delete(photo_to_delete) mock_unlink.assert_called_once() assert deleted_photo_data is not None assert deleted_photo_data.id == photo_id photo_in_db = service.read_by_id(photo_id) assert photo_in_db is None @patch('pathlib.Path.unlink') def test_delete_returns_none_for_non_existent_photo(mock_unlink, db_session): service = PhotoService(db_session) non_existent_photo = Photo( filepath="/fake/path.jpg", name="dummy", photo_hash="dummy_hash", fk_travel_diary_id=1 ) non_existent_photo.id = 999 result = service.delete(non_existent_photo) assert result is None mock_unlink.assert_not_called()