mirror of https://github.com/gmbrax/Pilgrim.git
				
				
				
			Add fixture and tests for the backup service
This commit is contained in:
		
							parent
							
								
									7faab93dfe
								
							
						
					
					
						commit
						53e8b36a42
					
				|  | @ -1,3 +1,6 @@ | ||||||
|  | from pathlib import Path | ||||||
|  | from unittest.mock import patch | ||||||
|  | 
 | ||||||
| import pytest | import pytest | ||||||
| from sqlalchemy import create_engine | from sqlalchemy import create_engine | ||||||
| from sqlalchemy.orm import sessionmaker | from sqlalchemy.orm import sessionmaker | ||||||
|  | @ -5,6 +8,8 @@ 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 | from pilgrim.models.photo import Photo | ||||||
|  | from pilgrim.utils import DirectoryManager | ||||||
|  | 
 | ||||||
| 
 | 
 | ||||||
| # Todos os imports necessários para as fixtures devem estar aqui | # Todos os imports necessários para as fixtures devem estar aqui | ||||||
| # ... | # ... | ||||||
|  | @ -62,3 +67,38 @@ def session_with_photos(session_with_one_diary): | ||||||
|     session.commit() |     session.commit() | ||||||
| 
 | 
 | ||||||
|     return session, [photo1, photo2] |     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() | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  |  | ||||||
|  | @ -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() | ||||||
		Loading…
	
		Reference in New Issue