Add the tests for the delete photo service and also fixed the missing import

This commit is contained in:
Gustavo Henrique Santos Souza de Miranda 2025-07-21 16:48:09 -03:00
parent 65cf7b04f3
commit 32f363f15a
2 changed files with 31 additions and 1 deletions

View File

@ -4,6 +4,7 @@ from sqlalchemy.orm import sessionmaker
from pilgrim.database import Base from pilgrim.database import Base
from pilgrim.models.travel_diary import TravelDiary from pilgrim.models.travel_diary import TravelDiary
from pilgrim.models.photo import Photo
# Todos os imports necessários para as fixtures devem estar aqui # Todos os imports necessários para as fixtures devem estar aqui
# ... # ...

View File

@ -108,3 +108,32 @@ def test_check_photo_by_hash_returns_none_when_not_found(session_with_photos):
invalid_diary_id = 999 invalid_diary_id = 999
result2 = service.check_photo_by_hash(existing_hash, invalid_diary_id) result2 = service.check_photo_by_hash(existing_hash, invalid_diary_id)
assert result2 is None assert result2 is None
@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()