mirror of
https://github.com/gmbrax/Pilgrim.git
synced 2026-11-10 15:20:18 +00:00
Compare commits
2
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
736c0f160f | ||
|
|
de27485c98 |
+2
-1
@@ -20,7 +20,8 @@ dependencies = [
|
|||||||
"sqlalchemy",
|
"sqlalchemy",
|
||||||
"textual",
|
"textual",
|
||||||
"tomli",
|
"tomli",
|
||||||
"tomli_w"
|
"tomli_w",
|
||||||
|
"unidecode"
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.urls]
|
[project.urls]
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from pilgrim.utils import DirectoryManager
|
|||||||
from sqlalchemy.exc import IntegrityError
|
from sqlalchemy.exc import IntegrityError
|
||||||
|
|
||||||
from pilgrim.models.travel_diary import TravelDiary
|
from pilgrim.models.travel_diary import TravelDiary
|
||||||
|
from unidecode import unidecode
|
||||||
|
|
||||||
class TravelDiaryService:
|
class TravelDiaryService:
|
||||||
def __init__(self, session):
|
def __init__(self, session):
|
||||||
@@ -20,8 +20,10 @@ class TravelDiaryService:
|
|||||||
- Replaces spaces with underscores
|
- Replaces spaces with underscores
|
||||||
- Ensures name is unique by adding a suffix if needed
|
- Ensures name is unique by adding a suffix if needed
|
||||||
"""
|
"""
|
||||||
|
transliterated_name = unidecode(name)
|
||||||
|
|
||||||
# Remove special characters and replace spaces
|
# Remove special characters and replace spaces
|
||||||
safe_name = re.sub(r'[^\w\s-]', '', name)
|
safe_name = re.sub(r'[^\w\s-]', '', transliterated_name)
|
||||||
safe_name = safe_name.strip().replace(' ', '_').lower()
|
safe_name = safe_name.strip().replace(' ', '_').lower()
|
||||||
|
|
||||||
# Ensure we have a valid name
|
# Ensure we have a valid name
|
||||||
|
|||||||
@@ -26,11 +26,3 @@ def populated_db_session(db_session):
|
|||||||
db_session.add(travel_diary)
|
db_session.add(travel_diary)
|
||||||
db_session.commit()
|
db_session.commit()
|
||||||
return db_session
|
return db_session
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def session_with_one_diary(db_session):
|
|
||||||
diary = TravelDiary(name="Diário de Teste", directory_name="diario_de_teste")
|
|
||||||
db_session.add(diary)
|
|
||||||
db_session.commit()
|
|
||||||
db_session.refresh(diary)
|
|
||||||
return db_session, diary
|
|
||||||
|
|||||||
@@ -1,88 +0,0 @@
|
|||||||
from unittest.mock import patch
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
|
|
||||||
from pilgrim import TravelDiary
|
|
||||||
from pilgrim.service.travel_diary_service import TravelDiaryService
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
@pytest.mark.asyncio # Marca o teste para rodar código assíncrono
|
|
||||||
async def test_create_diary_successfully(mock_ensure_dir, db_session):
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
new_diary = await service.async_create("Viagem para a Serra")
|
|
||||||
assert new_diary is not None
|
|
||||||
assert new_diary.id is not None
|
|
||||||
assert new_diary.name == "Viagem para a Serra"
|
|
||||||
assert new_diary.directory_name == "viagem_para_a_serra"
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
@patch.object(TravelDiaryService, '_sanitize_directory_name', return_value="nome_existente")
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
async def test_create_diary_handles_integrity_error(mock_sanitize, mock_ensure_dir, db_session):
|
|
||||||
existing_diary = TravelDiary(name="Diário Antigo", directory_name="nome_existente")
|
|
||||||
db_session.add(existing_diary)
|
|
||||||
db_session.commit()
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
with pytest.raises(ValueError, match="Could not create diary"):
|
|
||||||
await service.async_create("Qualquer Nome Novo")
|
|
||||||
mock_ensure_dir.assert_not_called()
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
def test_read_by_id_successfully(mock_ensure_dir, session_with_one_diary):
|
|
||||||
session, diary_to_find = session_with_one_diary
|
|
||||||
service = TravelDiaryService(session)
|
|
||||||
found_diary = service.read_by_id(diary_to_find.id)
|
|
||||||
assert found_diary is not None
|
|
||||||
assert found_diary.id == diary_to_find.id
|
|
||||||
assert found_diary.name == "Diário de Teste"
|
|
||||||
mock_ensure_dir.assert_called_once_with(found_diary)
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
def test_read_by_id_returns_none_for_invalid_id(mock_ensure_dir, db_session):
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
result = service.read_by_id(999)
|
|
||||||
assert result is None
|
|
||||||
mock_ensure_dir.assert_not_called()
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
def test_read_all_returns_all_diaries(mock_ensure_dir, db_session):
|
|
||||||
d1 = TravelDiary(name="Diário 1", directory_name="d1")
|
|
||||||
d2 = TravelDiary(name="Diário 2", directory_name="d2")
|
|
||||||
db_session.add_all([d1, d2])
|
|
||||||
db_session.commit()
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
diaries = service.read_all()
|
|
||||||
assert isinstance(diaries, list)
|
|
||||||
assert len(diaries) == 2
|
|
||||||
assert mock_ensure_dir.call_count == 2
|
|
||||||
|
|
||||||
@patch.object(TravelDiaryService, '_ensure_diary_directory')
|
|
||||||
def test_read_all_returns_empty_list_for_empty_db(mock_ensure_dir, db_session):
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
diaries = service.read_all()
|
|
||||||
assert isinstance(diaries, list)
|
|
||||||
assert len(diaries) == 0
|
|
||||||
mock_ensure_dir.assert_not_called()
|
|
||||||
|
|
||||||
def test_sanitize_directory_name_formats_string_correctly(db_session):
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
name1 = "Minha Primeira Viagem"
|
|
||||||
assert service._sanitize_directory_name(name1) == "minha_primeira_viagem"
|
|
||||||
name2 = "Viagem para o #Rio de Janeiro! @2025"
|
|
||||||
assert service._sanitize_directory_name(name2) == "viagem_para_o_rio_de_janeiro_2025"
|
|
||||||
name3 = " Mochilão na Europa "
|
|
||||||
assert service._sanitize_directory_name(name3) == "mochilão_na_europa"
|
|
||||||
|
|
||||||
def test_sanitize_directory_name_handles_uniqueness(db_session):
|
|
||||||
existing_diary = TravelDiary(name="Viagem para a Praia", directory_name="viagem_para_a_praia")
|
|
||||||
db_session.add(existing_diary)
|
|
||||||
db_session.commit()
|
|
||||||
service = TravelDiaryService(db_session)
|
|
||||||
new_sanitized_name = service._sanitize_directory_name("Viagem para a Praia")
|
|
||||||
assert new_sanitized_name == "viagem_para_a_praia_1"
|
|
||||||
another_diary = TravelDiary(name="Outra Viagem", directory_name="viagem_para_a_praia_1")
|
|
||||||
db_session.add(another_diary)
|
|
||||||
db_session.commit()
|
|
||||||
|
|
||||||
third_sanitized_name = service._sanitize_directory_name("Viagem para a Praia")
|
|
||||||
assert third_sanitized_name == "viagem_para_a_praia_2"
|
|
||||||
Reference in New Issue
Block a user