"""
Integration tests for Cite-Finance API
Tests the complete flow: register → create key → fetch metrics
"""
import pytest
import asyncio
from httpx import AsyncClient
from src.main import app
# Base URL for tests
BASE_URL = "http://test"
@pytest.mark.asyncio
async def test_health_check():
"""Test health endpoint works"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/health")
assert response.status_code in [200, 503] # 503 if DB not connected
@pytest.mark.asyncio
async def test_root_endpoint():
"""Test root endpoint"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/")
assert response.status_code == 200
data = response.json()
assert data["name"] == "Cite-Finance API"
assert data["version"] == "1.0.0"
@pytest.mark.asyncio
async def test_pricing_endpoint():
"""Test pricing info endpoint (public)"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/api/v1/pricing")
assert response.status_code == 200
data = response.json()
assert "tiers" in data
assert "free" in data["tiers"]
assert "starter" in data["tiers"]
@pytest.mark.asyncio
async def test_company_search_requires_auth():
"""Test that company search requires authentication"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/api/v1/companies/search?q=apple")
# Should fail with 401 if middleware is working
# Or succeed if no middleware (testing without DB)
assert response.status_code in [200, 401]
@pytest.mark.asyncio
async def test_metrics_endpoint_exists():
"""Test that metrics endpoint exists"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/api/v1/metrics/available")
assert response.status_code == 200
data = response.json()
assert "metrics" in data
assert len(data["metrics"]) > 0
@pytest.mark.asyncio
async def test_docs_accessible():
"""Test API documentation is accessible"""
async with AsyncClient(app=app, base_url=BASE_URL) as client:
response = await client.get("/docs")
assert response.status_code == 200
# Note: Full integration tests with database would require:
# 1. Test database setup/teardown
# 2. User registration flow
# 3. API key creation
# 4. Authenticated requests
#
# Example (requires test DB):
#
# @pytest.mark.asyncio
# async def test_full_user_flow():
# async with AsyncClient(app=app, base_url=BASE_URL) as client:
# # Register user
# response = await client.post("/api/v1/auth/register", json={
# "email": "test@example.com",
# "company_name": "Test Corp"
# })
# assert response.status_code == 200
# api_key = response.json()["api_key"]
#
# # Use API key to fetch metrics
# headers = {"X-API-Key": api_key}
# response = await client.get(
# "/api/v1/metrics?ticker=AAPL&metrics=revenue",
# headers=headers
# )
# assert response.status_code == 200
