diff --git a/tests/conftest.py b/tests/conftest.py index 91dff2a..b89d02d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,3 +1,6 @@ +from pathlib import Path +from unittest.mock import patch + import pytest from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker @@ -8,6 +11,8 @@ from pilgrim.database import Base from pilgrim.models.travel_diary import TravelDiary from pilgrim.models.entry import Entry from pilgrim.models.photo import Photo +from pilgrim.utils import DirectoryManager + # Todos os imports necessários para as fixtures devem estar aqui # ... @@ -66,6 +71,39 @@ def session_with_photos(session_with_one_diary): return session, [photo1, photo2] +@pytest.fixture +def backup_test_env_files_only(tmp_path): + fake_config_dir = tmp_path / "config" + fake_diaries_root = tmp_path / "diaries" + fake_db_path = fake_config_dir / "database.db" + fake_config_dir.mkdir() + fake_diaries_root.mkdir() + with patch.object(DirectoryManager, 'get_database_path', return_value=fake_db_path), \ + patch.object(DirectoryManager, 'get_config_directory', return_value=fake_config_dir), \ + patch.object(DirectoryManager, 'get_diaries_root', return_value=fake_diaries_root): + engine = create_engine(f"sqlite:///{fake_db_path}") + Base.metadata.create_all(engine) + Session = sessionmaker(bind=engine) + session = Session() + + diary = TravelDiary(name="Viagem de Teste", directory_name="viagem_de_teste") + session.add(diary) + session.commit() + photo_file_path_str = str(fake_diaries_root / "viagem_de_teste" / "images" / "foto1.jpg") + photo = Photo(filepath=photo_file_path_str, name="Foto 1", photo_hash="hash123", fk_travel_diary_id=diary.id) + session.add(photo) + session.commit() + photo_file_path = Path(photo_file_path_str) + photo_file_path.parent.mkdir(parents=True) + photo_file_path.touch() + yield { + "session": session, + "db_path": fake_db_path, + "config_dir": fake_config_dir, + "diaries_root": fake_diaries_root, + } + session.close() + @pytest.fixture def entry_with_photo_references(session_with_one_diary): session, diary = session_with_one_diary @@ -97,4 +135,5 @@ def session_with_multiple_entries(session_with_one_diary): session.add_all([entry1, entry2]) session.commit() - return session, diary \ No newline at end of file + return session, diary + diff --git a/tests/service/test_backup_service.py b/tests/service/test_backup_service.py new file mode 100644 index 0000000..3038194 --- /dev/null +++ b/tests/service/test_backup_service.py @@ -0,0 +1,39 @@ +import zipfile +from pathlib import Path +from unittest.mock import patch, MagicMock +from pilgrim.service.backup_service import BackupService +from pilgrim.utils.directory_manager import DirectoryManager +import pytest + +@patch.object(DirectoryManager, 'get_diaries_root') +@patch.object(DirectoryManager, 'get_config_directory') +@patch.object(DirectoryManager, 'get_database_path') +def test_create_backup_success(mock_get_db_path, mock_get_config_dir, mock_get_diaries_root, backup_test_env_files_only): + env = backup_test_env_files_only + session = env["session"] + mock_get_db_path.return_value = env["db_path"] + mock_get_config_dir.return_value = env["config_dir"] + mock_get_diaries_root.return_value = env["diaries_root"] + + service = BackupService(session) + backup_zip_path = env["config_dir"] / "backup.zip" + success, returned_path = service.create_backup() + assert success is True + assert returned_path == backup_zip_path + assert backup_zip_path.exists() + + with zipfile.ZipFile(backup_zip_path, 'r') as zf: + file_list = zf.namelist() + assert "database.sql" in file_list + assert "diaries/viagem_de_teste/images/foto1.jpg" in file_list + sql_dump = zf.read("database.sql").decode('utf-8') + assert "Viagem de Teste" in sql_dump + +@patch.object(DirectoryManager, 'get_database_path') +def test_create_backup_fails_if_db_not_found(mock_get_db_path, tmp_path: Path): + non_existent_db_path = tmp_path / "non_existent.db" + mock_get_db_path.return_value = non_existent_db_path + mock_session = MagicMock() + service = BackupService(mock_session) + with pytest.raises(FileNotFoundError, match="No Database Found"): + service.create_backup() \ No newline at end of file