chore: import upstream snapshot with attribution
OSV-Scanner (Scheduled) / scan-scheduled (push) Failing after 0s
Create Release / test-gate (push) Has been cancelled
Create Release / release-gate (push) Has been cancelled
Create Release / ci-gate (push) Has been cancelled
Create Release / version-check (push) Has been cancelled
Create Release / e2e-test-gate (push) Has been cancelled
Create Release / responsive-test-gate (push) Has been cancelled
Create Release / compat-test-gate (push) Has been cancelled
Create Release / compose-integration-gate (push) Has been cancelled
Create Release / vulture-gate (push) Has been cancelled
Create Release / build (push) Has been cancelled
Create Release / provenance (push) Has been cancelled
Create Release / prerelease-docker (push) Has been cancelled
Create Release / publish-docker (push) Has been cancelled
Create Release / create-release (push) Has been cancelled
Create Release / cleanup-changelog (push) Has been cancelled
Create Release / trigger-pypi (push) Has been cancelled
Create Release / monitor-pypi (push) Has been cancelled
Create Release / Clean up orphan prerelease tags and signatures (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-form] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-metrics] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-workflow] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [settings-core] (push) Has been cancelled
CodeQL Advanced / Analyze (javascript-typescript) (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [history-news] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [library] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [link-analytics] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [chat-core] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [chat-lifecycle] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [error-benchmark] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [settings-pages] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) (push) Has been cancelled
Docker Tests (Consolidated) / Accessibility Tests (push) Has been cancelled
Docker Tests (Consolidated) / LLM Unit Tests (push) Has been cancelled
Docker Tests (Consolidated) / LLM Example Tests (push) Has been cancelled
Docker Tests (Consolidated) / Production Image Smoke Test (push) Has been cancelled
Docker Tests (Consolidated) / Infrastructure Tests (push) Has been cancelled
OSSF Scorecard / OSSF Security Scorecard Analysis (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [mobile] (push) Has been cancelled
Backwards Compatibility / Verify Encryption Constants (push) Has been cancelled
Backwards Compatibility / PyPI Version Compatibility (push) Has been cancelled
Backwards Compatibility / Database Migration Tests (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Docker Tests (Consolidated) / detect-changes (push) Has been cancelled
Docker Tests (Consolidated) / Build Test Image (push) Has been cancelled
Docker Tests (Consolidated) / All Pytest Tests + Coverage (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [accessibility] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [api-crud] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-login] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-pages] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-register] (push) Has been cancelled
OSV-Scanner (Scheduled) / scan-scheduled (push) Failing after 0s
Create Release / test-gate (push) Has been cancelled
Create Release / release-gate (push) Has been cancelled
Create Release / ci-gate (push) Has been cancelled
Create Release / version-check (push) Has been cancelled
Create Release / e2e-test-gate (push) Has been cancelled
Create Release / responsive-test-gate (push) Has been cancelled
Create Release / compat-test-gate (push) Has been cancelled
Create Release / compose-integration-gate (push) Has been cancelled
Create Release / vulture-gate (push) Has been cancelled
Create Release / build (push) Has been cancelled
Create Release / provenance (push) Has been cancelled
Create Release / prerelease-docker (push) Has been cancelled
Create Release / publish-docker (push) Has been cancelled
Create Release / create-release (push) Has been cancelled
Create Release / cleanup-changelog (push) Has been cancelled
Create Release / trigger-pypi (push) Has been cancelled
Create Release / monitor-pypi (push) Has been cancelled
Create Release / Clean up orphan prerelease tags and signatures (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-form] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-metrics] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [research-workflow] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [settings-core] (push) Has been cancelled
CodeQL Advanced / Analyze (javascript-typescript) (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [history-news] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [library] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [link-analytics] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [chat-core] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [chat-lifecycle] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [error-benchmark] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [settings-pages] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) (push) Has been cancelled
Docker Tests (Consolidated) / Accessibility Tests (push) Has been cancelled
Docker Tests (Consolidated) / LLM Unit Tests (push) Has been cancelled
Docker Tests (Consolidated) / LLM Example Tests (push) Has been cancelled
Docker Tests (Consolidated) / Production Image Smoke Test (push) Has been cancelled
Docker Tests (Consolidated) / Infrastructure Tests (push) Has been cancelled
OSSF Scorecard / OSSF Security Scorecard Analysis (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [mobile] (push) Has been cancelled
Backwards Compatibility / Verify Encryption Constants (push) Has been cancelled
Backwards Compatibility / PyPI Version Compatibility (push) Has been cancelled
Backwards Compatibility / Database Migration Tests (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Docker Tests (Consolidated) / detect-changes (push) Has been cancelled
Docker Tests (Consolidated) / Build Test Image (push) Has been cancelled
Docker Tests (Consolidated) / All Pytest Tests + Coverage (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [accessibility] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [api-crud] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-login] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-pages] (push) Has been cancelled
Docker Tests (Consolidated) / UI Tests (Puppeteer) [auth-register] (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,247 @@
|
||||
"""Tests for web/api.py module - REST API endpoints."""
|
||||
|
||||
from unittest.mock import patch, MagicMock
|
||||
import time
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
"""Create a test client for the API."""
|
||||
from flask import Flask
|
||||
from local_deep_research.web.api import api_blueprint
|
||||
|
||||
app = Flask(__name__)
|
||||
app.config["TESTING"] = True
|
||||
app.config["SECRET_KEY"] = "test-secret-key"
|
||||
|
||||
app.register_blueprint(api_blueprint)
|
||||
|
||||
return app.test_client()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def authenticated_client():
|
||||
"""Create a test client with authentication mocked."""
|
||||
from flask import Flask
|
||||
from local_deep_research.web.api import api_blueprint
|
||||
|
||||
app = Flask(__name__)
|
||||
app.config["TESTING"] = True
|
||||
app.config["SECRET_KEY"] = "test-secret-key"
|
||||
|
||||
app.register_blueprint(api_blueprint)
|
||||
|
||||
with app.test_client() as client:
|
||||
with client.session_transaction() as sess:
|
||||
sess["username"] = "testuser"
|
||||
yield client
|
||||
|
||||
|
||||
class TestHealthCheck:
|
||||
"""Tests for /api/v1/health endpoint."""
|
||||
|
||||
def test_returns_ok_status(self, client):
|
||||
"""Should return ok status."""
|
||||
response = client.get("/api/v1/health")
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data["status"] == "ok"
|
||||
assert data["message"] == "API is running"
|
||||
assert "timestamp" in data
|
||||
|
||||
def test_returns_timestamp(self, client):
|
||||
"""Should return a valid timestamp."""
|
||||
response = client.get("/api/v1/health")
|
||||
data = response.get_json()
|
||||
# Timestamp should be close to current time
|
||||
assert abs(data["timestamp"] - time.time()) < 5
|
||||
|
||||
|
||||
class TestApiDocumentation:
|
||||
"""Tests for /api/v1/ endpoint."""
|
||||
|
||||
def test_returns_api_docs(self, authenticated_client):
|
||||
"""Should return API documentation."""
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_user_db_session"
|
||||
) as mock_session_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_session_ctx.return_value.__enter__ = MagicMock(
|
||||
return_value=mock_session
|
||||
)
|
||||
mock_session_ctx.return_value.__exit__ = MagicMock(
|
||||
return_value=None
|
||||
)
|
||||
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_settings_manager"
|
||||
) as mock_settings:
|
||||
mock_manager = MagicMock()
|
||||
mock_manager.get_setting.side_effect = lambda key, default: {
|
||||
"app.enable_api": True,
|
||||
}.get(key, default)
|
||||
mock_settings.return_value = mock_manager
|
||||
|
||||
response = authenticated_client.get("/api/v1/")
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data["api_version"] == "v1"
|
||||
assert "endpoints" in data
|
||||
assert len(data["endpoints"]) >= 1
|
||||
|
||||
def test_lists_available_endpoints(self, authenticated_client):
|
||||
"""Should list all available endpoints."""
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_user_db_session"
|
||||
) as mock_session_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_session_ctx.return_value.__enter__ = MagicMock(
|
||||
return_value=mock_session
|
||||
)
|
||||
mock_session_ctx.return_value.__exit__ = MagicMock(
|
||||
return_value=None
|
||||
)
|
||||
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_settings_manager"
|
||||
) as mock_settings:
|
||||
mock_manager = MagicMock()
|
||||
mock_manager.get_setting.side_effect = lambda key, default: {
|
||||
"app.enable_api": True,
|
||||
}.get(key, default)
|
||||
mock_settings.return_value = mock_manager
|
||||
|
||||
response = authenticated_client.get("/api/v1/")
|
||||
data = response.get_json()
|
||||
|
||||
# Check that key endpoints are documented
|
||||
endpoints = data["endpoints"]
|
||||
paths = [ep["path"] for ep in endpoints]
|
||||
assert "/api/v1/quick_summary" in paths
|
||||
assert "/api/v1/generate_report" in paths
|
||||
assert "/api/v1/analyze_documents" in paths
|
||||
|
||||
|
||||
class TestApiAccessControl:
|
||||
"""Tests for API access control decorator."""
|
||||
|
||||
def test_returns_403_when_api_disabled(self, authenticated_client):
|
||||
"""Should return 403 when API is disabled."""
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_user_db_session"
|
||||
) as mock_session_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_session_ctx.return_value.__enter__ = MagicMock(
|
||||
return_value=mock_session
|
||||
)
|
||||
mock_session_ctx.return_value.__exit__ = MagicMock(
|
||||
return_value=None
|
||||
)
|
||||
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_settings_manager"
|
||||
) as mock_settings:
|
||||
mock_manager = MagicMock()
|
||||
mock_manager.get_setting.side_effect = lambda key, default: {
|
||||
"app.enable_api": False, # API disabled
|
||||
}.get(key, default)
|
||||
mock_settings.return_value = mock_manager
|
||||
|
||||
response = authenticated_client.get("/api/v1/")
|
||||
assert response.status_code == 403
|
||||
data = response.get_json()
|
||||
assert "disabled" in data["error"].lower()
|
||||
|
||||
|
||||
class TestRateLimiting:
|
||||
"""Tests for rate limiting functionality."""
|
||||
|
||||
def test_allows_requests_under_limit(self, authenticated_client):
|
||||
"""Should allow requests under the rate limit."""
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_user_db_session"
|
||||
) as mock_session_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_session_ctx.return_value.__enter__ = MagicMock(
|
||||
return_value=mock_session
|
||||
)
|
||||
mock_session_ctx.return_value.__exit__ = MagicMock(
|
||||
return_value=None
|
||||
)
|
||||
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_settings_manager"
|
||||
) as mock_settings:
|
||||
mock_manager = MagicMock()
|
||||
mock_manager.get_setting.side_effect = lambda key, default: {
|
||||
"app.enable_api": True,
|
||||
}.get(key, default)
|
||||
mock_settings.return_value = mock_manager
|
||||
|
||||
# Make a few requests
|
||||
for _ in range(3):
|
||||
response = authenticated_client.get("/api/v1/")
|
||||
assert response.status_code == 200
|
||||
|
||||
|
||||
class TestQuickSummaryQueryTypeValidation:
|
||||
"""Tests for query type validation in api_quick_summary (PR #2034).
|
||||
|
||||
PR #2034 adds isinstance(query, str) check to reject non-string
|
||||
query values with 400 status.
|
||||
"""
|
||||
|
||||
def _make_request(self, authenticated_client, json_data):
|
||||
"""Helper to make a quick_summary request with mocked settings."""
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_user_db_session"
|
||||
) as mock_session_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_session_ctx.return_value.__enter__ = MagicMock(
|
||||
return_value=mock_session
|
||||
)
|
||||
mock_session_ctx.return_value.__exit__ = MagicMock(
|
||||
return_value=None
|
||||
)
|
||||
|
||||
with patch(
|
||||
"local_deep_research.web.api.get_settings_manager"
|
||||
) as mock_settings:
|
||||
mock_manager = MagicMock()
|
||||
mock_manager.get_setting.side_effect = lambda key, default: {
|
||||
"app.enable_api": True,
|
||||
}.get(key, default)
|
||||
mock_settings.return_value = mock_manager
|
||||
|
||||
return authenticated_client.post(
|
||||
"/api/v1/quick_summary",
|
||||
json=json_data,
|
||||
)
|
||||
|
||||
def test_rejects_integer_query(self, authenticated_client):
|
||||
"""Integer query returns 400."""
|
||||
response = self._make_request(authenticated_client, {"query": 12345})
|
||||
assert response.status_code == 400
|
||||
data = response.get_json()
|
||||
assert "string" in data["error"].lower()
|
||||
|
||||
def test_rejects_list_query(self, authenticated_client):
|
||||
"""List query returns 400."""
|
||||
response = self._make_request(
|
||||
authenticated_client, {"query": ["a", "b"]}
|
||||
)
|
||||
assert response.status_code == 400
|
||||
|
||||
def test_rejects_dict_query(self, authenticated_client):
|
||||
"""Dict query returns 400."""
|
||||
response = self._make_request(
|
||||
authenticated_client, {"query": {"nested": "value"}}
|
||||
)
|
||||
assert response.status_code == 400
|
||||
|
||||
def test_rejects_null_query(self, authenticated_client):
|
||||
"""None/null query returns 400."""
|
||||
response = self._make_request(authenticated_client, {"query": None})
|
||||
assert response.status_code == 400
|
||||
Reference in New Issue
Block a user