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
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
|
@ -5,6 +8,8 @@ from sqlalchemy.orm import sessionmaker
|
|||
from pilgrim.database import Base
|
||||
from pilgrim.models.travel_diary import TravelDiary
|
||||
from pilgrim.models.photo import Photo
|
||||
from pilgrim.utils import DirectoryManager
|
||||
|
||||
|
||||
# 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()
|
||||
|
||||
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