Jarian Cottingham
93e287f88b
- Add pytest test suite across all modules (config, database, file_processor, ai_processor, api, scheduler, monitoring, ftp_server) - 145 tests covering normal paths, error handling, edge cases - Mock external dependencies (AI endpoint, requests, pyftpdlib) - 95% code coverage with fail-under=90 threshold in CI - Update CI workflow to run pytest with coverage enforcement - Add pyproject.toml with pytest/coverage configuration
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
"""Shared fixtures for FactsDB tests."""
|
|
|
|
import os
|
|
import tempfile
|
|
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from factsdb.config import Config, DatabaseConfig, AIEndpointConfig, FTPServerConfig, SchedulerConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_db_path():
|
|
"""Provide a temporary database path that's cleaned up after the test."""
|
|
with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as f:
|
|
path = f.name
|
|
yield path
|
|
if os.path.exists(path):
|
|
os.unlink(path)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_env_vars():
|
|
"""Set required environment variables for Config initialization."""
|
|
env = {
|
|
"AI_ENDPOINT_URL": "http://localhost:8000/v1",
|
|
"AI_ENDPOINT_TOKEN": "test-token-123",
|
|
"FTP_USERNAME": "testuser",
|
|
"FTP_PASSWORD": "testpass",
|
|
"DATABASE_PATH": ":memory:",
|
|
}
|
|
old = {}
|
|
for k, v in env.items():
|
|
old[k] = os.environ.get(k)
|
|
os.environ[k] = v
|
|
yield env
|
|
for k, v in old.items():
|
|
if v is None:
|
|
os.environ.pop(k, None)
|
|
else:
|
|
os.environ[k] = v
|
|
|
|
|
|
@pytest.fixture
|
|
def config(mock_env_vars):
|
|
"""Provide a fully configured Config instance."""
|
|
return Config()
|
|
|
|
|
|
@pytest.fixture
|
|
def db_config(temp_db_path):
|
|
"""Provide a DatabaseConfig pointing to a temp file."""
|
|
return DatabaseConfig(path=temp_db_path)
|
|
|
|
|
|
@pytest.fixture
|
|
def ai_endpoint_config():
|
|
"""Provide an AIEndpointConfig for testing."""
|
|
return AIEndpointConfig(url="http://localhost:8000/v1", auth_token="test-token")
|