mirror of
https://github.com/elisiariocouto/leggen.git
synced 2025-12-13 12:32:18 +00:00
153 lines
4.1 KiB
Python
153 lines
4.1 KiB
Python
"""Pytest configuration and shared fixtures."""
|
|
|
|
import pytest
|
|
import tempfile
|
|
import json
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
from fastapi.testclient import TestClient
|
|
|
|
from leggend.main import create_app
|
|
from leggend.config import Config
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_config_dir():
|
|
"""Create a temporary config directory for testing."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
config_dir = Path(tmpdir) / ".config" / "leggen"
|
|
config_dir.mkdir(parents=True, exist_ok=True)
|
|
yield config_dir
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_db_path():
|
|
"""Create a temporary database file for testing."""
|
|
with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as tmp_file:
|
|
db_path = Path(tmp_file.name)
|
|
yield db_path
|
|
# Clean up the temporary database file after test
|
|
if db_path.exists():
|
|
db_path.unlink()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config(temp_config_dir, temp_db_path):
|
|
"""Mock configuration for testing."""
|
|
config_data = {
|
|
"gocardless": {
|
|
"key": "test-key",
|
|
"secret": "test-secret",
|
|
"url": "https://bankaccountdata.gocardless.com/api/v2",
|
|
},
|
|
"database": {"sqlite": True},
|
|
"scheduler": {"sync": {"enabled": True, "hour": 3, "minute": 0}},
|
|
}
|
|
|
|
config_file = temp_config_dir / "config.toml"
|
|
with open(config_file, "wb") as f:
|
|
import tomli_w
|
|
|
|
tomli_w.dump(config_data, f)
|
|
|
|
# Mock the config path
|
|
with patch.object(Config, "load_config") as mock_load:
|
|
mock_load.return_value = config_data
|
|
config = Config()
|
|
config._config = config_data
|
|
config._config_path = str(config_file)
|
|
yield config
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_auth_token(temp_config_dir):
|
|
"""Mock GoCardless authentication token."""
|
|
auth_data = {"access": "mock-access-token", "refresh": "mock-refresh-token"}
|
|
|
|
auth_file = temp_config_dir / "auth.json"
|
|
with open(auth_file, "w") as f:
|
|
json.dump(auth_data, f)
|
|
|
|
return auth_data
|
|
|
|
|
|
@pytest.fixture
|
|
def fastapi_app():
|
|
"""Create FastAPI test application."""
|
|
return create_app()
|
|
|
|
|
|
@pytest.fixture
|
|
def api_client(fastapi_app):
|
|
"""Create FastAPI test client."""
|
|
return TestClient(fastapi_app)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_db_path(temp_db_path):
|
|
"""Mock the database path to use temporary database for testing."""
|
|
from leggen.utils.paths import path_manager
|
|
|
|
# Set the path manager to use the temporary database
|
|
original_database_path = path_manager._database_path
|
|
path_manager.set_database_path(temp_db_path)
|
|
|
|
try:
|
|
yield temp_db_path
|
|
finally:
|
|
# Restore original path
|
|
path_manager._database_path = original_database_path
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_bank_data():
|
|
"""Sample bank/institution data for testing."""
|
|
return [
|
|
{
|
|
"id": "REVOLUT_REVOLT21",
|
|
"name": "Revolut",
|
|
"bic": "REVOLT21",
|
|
"transaction_total_days": 90,
|
|
"countries": ["GB", "LT"],
|
|
},
|
|
{
|
|
"id": "BANCOBPI_BBPIPTPL",
|
|
"name": "Banco BPI",
|
|
"bic": "BBPIPTPL",
|
|
"transaction_total_days": 90,
|
|
"countries": ["PT"],
|
|
},
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_account_data():
|
|
"""Sample account data for testing."""
|
|
return {
|
|
"id": "test-account-123",
|
|
"institution_id": "REVOLUT_REVOLT21",
|
|
"status": "READY",
|
|
"iban": "LT313250081177977789",
|
|
"created": "2024-02-13T23:56:00Z",
|
|
"last_accessed": "2025-09-01T09:30:00Z",
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_transaction_data():
|
|
"""Sample transaction data for testing."""
|
|
return {
|
|
"transactions": {
|
|
"booked": [
|
|
{
|
|
"internalTransactionId": "txn-123",
|
|
"bookingDate": "2025-09-01",
|
|
"valueDate": "2025-09-01",
|
|
"transactionAmount": {"amount": "-10.50", "currency": "EUR"},
|
|
"remittanceInformationUnstructured": "Coffee Shop Payment",
|
|
}
|
|
],
|
|
"pending": [],
|
|
}
|
|
}
|