mirror of
https://github.com/elisiariocouto/leggen.git
synced 2025-12-11 17:22:18 +00:00
22ec0e36b1
- Fix critical S3 client configuration bug for path-style addressing - Add toast notifications for better user feedback on S3 config operations - Set up Toaster component in root layout for app-wide notifications - Clean up unused imports in test files 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
227 lines
7.9 KiB
Python
227 lines
7.9 KiB
Python
"""Tests for backup service functionality."""
|
|
|
|
import tempfile
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from botocore.exceptions import ClientError, NoCredentialsError
|
|
|
|
from leggen.models.config import S3BackupConfig
|
|
from leggen.services.backup_service import BackupService
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestBackupService:
|
|
"""Test backup service functionality."""
|
|
|
|
def test_backup_service_initialization(self):
|
|
"""Test backup service can be initialized."""
|
|
service = BackupService()
|
|
assert service.s3_config is None
|
|
assert service._s3_client is None
|
|
|
|
def test_backup_service_with_config(self):
|
|
"""Test backup service initialization with config."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
service = BackupService(s3_config)
|
|
assert service.s3_config == s3_config
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_test_connection_success(self):
|
|
"""Test successful S3 connection test."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
|
|
service = BackupService()
|
|
|
|
# Mock S3 client
|
|
with patch("boto3.Session") as mock_session:
|
|
mock_client = MagicMock()
|
|
mock_session.return_value.client.return_value = mock_client
|
|
|
|
# Mock successful list_objects_v2 call
|
|
mock_client.list_objects_v2.return_value = {"Contents": []}
|
|
|
|
result = await service.test_connection(s3_config)
|
|
assert result is True
|
|
|
|
# Verify the client was called correctly
|
|
mock_client.list_objects_v2.assert_called_once_with(
|
|
Bucket="test-bucket", MaxKeys=1
|
|
)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_test_connection_no_credentials(self):
|
|
"""Test S3 connection test with no credentials."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
|
|
service = BackupService()
|
|
|
|
# Mock S3 client to raise NoCredentialsError
|
|
with patch("boto3.Session") as mock_session:
|
|
mock_client = MagicMock()
|
|
mock_session.return_value.client.return_value = mock_client
|
|
mock_client.list_objects_v2.side_effect = NoCredentialsError()
|
|
|
|
result = await service.test_connection(s3_config)
|
|
assert result is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_test_connection_client_error(self):
|
|
"""Test S3 connection test with client error."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
|
|
service = BackupService()
|
|
|
|
# Mock S3 client to raise ClientError
|
|
with patch("boto3.Session") as mock_session:
|
|
mock_client = MagicMock()
|
|
mock_session.return_value.client.return_value = mock_client
|
|
error_response = {
|
|
"Error": {"Code": "NoSuchBucket", "Message": "Bucket not found"}
|
|
}
|
|
mock_client.list_objects_v2.side_effect = ClientError(
|
|
error_response, "ListObjectsV2"
|
|
)
|
|
|
|
result = await service.test_connection(s3_config)
|
|
assert result is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_backup_database_no_config(self):
|
|
"""Test backup database with no configuration."""
|
|
service = BackupService()
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db_path.write_text("test database content")
|
|
|
|
result = await service.backup_database(db_path)
|
|
assert result is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_backup_database_disabled(self):
|
|
"""Test backup database with disabled configuration."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
enabled=False,
|
|
)
|
|
service = BackupService(s3_config)
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db_path.write_text("test database content")
|
|
|
|
result = await service.backup_database(db_path)
|
|
assert result is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_backup_database_file_not_found(self):
|
|
"""Test backup database with non-existent file."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
service = BackupService(s3_config)
|
|
|
|
non_existent_path = Path("/non/existent/path.db")
|
|
result = await service.backup_database(non_existent_path)
|
|
assert result is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_backup_database_success(self):
|
|
"""Test successful database backup."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
service = BackupService(s3_config)
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db_path.write_text("test database content")
|
|
|
|
# Mock S3 client
|
|
with patch("boto3.Session") as mock_session:
|
|
mock_client = MagicMock()
|
|
mock_session.return_value.client.return_value = mock_client
|
|
|
|
result = await service.backup_database(db_path)
|
|
assert result is True
|
|
|
|
# Verify upload_file was called
|
|
mock_client.upload_file.assert_called_once()
|
|
args = mock_client.upload_file.call_args[0]
|
|
assert args[0] == str(db_path) # source file
|
|
assert args[1] == "test-bucket" # bucket name
|
|
assert args[2].startswith("leggen_backups/database_backup_") # key
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_backups_success(self):
|
|
"""Test successful backup listing."""
|
|
s3_config = S3BackupConfig(
|
|
access_key_id="test-key",
|
|
secret_access_key="test-secret",
|
|
bucket_name="test-bucket",
|
|
region="us-east-1",
|
|
)
|
|
service = BackupService(s3_config)
|
|
|
|
# Mock S3 client response
|
|
with patch("boto3.Session") as mock_session:
|
|
mock_client = MagicMock()
|
|
mock_session.return_value.client.return_value = mock_client
|
|
|
|
from datetime import datetime
|
|
|
|
mock_response = {
|
|
"Contents": [
|
|
{
|
|
"Key": "leggen_backups/database_backup_20250101_120000.db",
|
|
"LastModified": datetime(2025, 1, 1, 12, 0, 0),
|
|
"Size": 1024,
|
|
},
|
|
{
|
|
"Key": "leggen_backups/database_backup_20250101_130000.db",
|
|
"LastModified": datetime(2025, 1, 1, 13, 0, 0),
|
|
"Size": 2048,
|
|
},
|
|
]
|
|
}
|
|
mock_client.list_objects_v2.return_value = mock_response
|
|
|
|
backups = await service.list_backups()
|
|
assert len(backups) == 2
|
|
|
|
# Check that backups are sorted by last modified (newest first)
|
|
assert backups[0]["last_modified"] > backups[1]["last_modified"]
|
|
assert backups[0]["size"] == 2048
|
|
assert backups[1]["size"] == 1024
|