b52318eeae
- Add intelligent-router.sh hook for automatic agent routing - Add AUTO-TRIGGER-SUMMARY.md documentation - Add FINAL-INTEGRATION-SUMMARY.md documentation - Complete Prometheus integration (6 commands + 4 tools) - Complete Dexto integration (12 commands + 5 tools) - Enhanced Ralph with access to all agents - Fix /clawd command (removed disable-model-invocation) - Update hooks.json to v5 with intelligent routing - 291 total skills now available - All 21 commands with automatic routing 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from unittest import mock
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.testclient import TestClient
|
|
|
|
from prometheus.app.api.routes import user
|
|
from prometheus.app.entity.user import User
|
|
from prometheus.app.exception_handler import register_exception_handlers
|
|
|
|
app = FastAPI()
|
|
register_exception_handlers(app)
|
|
app.include_router(user.router, prefix="/user", tags=["user"])
|
|
|
|
|
|
@app.middleware("mock_jwt_middleware")
|
|
async def add_user_id(request: Request, call_next):
|
|
request.state.user_id = 1 # Set user_id to 1 for testing purposes
|
|
response = await call_next(request)
|
|
return response
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_service():
|
|
service = mock.MagicMock()
|
|
app.state.service = service
|
|
yield service
|
|
|
|
|
|
def test_list(mock_service):
|
|
# Mock user as admin and return a list of users
|
|
mock_service["user_service"].list_users = AsyncMock(
|
|
return_value=[
|
|
User(
|
|
id=1,
|
|
username="testuser",
|
|
email="test@gmail.com",
|
|
password_hash="hashedpassword",
|
|
github_token="ghp_1234567890abcdef1234567890abcdef1234",
|
|
issue_credit=10,
|
|
is_superuser=False,
|
|
)
|
|
]
|
|
)
|
|
mock_service["user_service"].is_admin = AsyncMock(return_value=True)
|
|
|
|
# Test the list endpoint
|
|
response = client.get("user/list/")
|
|
assert response.status_code == 200
|
|
assert response.json() == {
|
|
"code": 200,
|
|
"message": "success",
|
|
"data": [
|
|
{
|
|
"id": 1,
|
|
"username": "testuser",
|
|
"email": "test@gmail.com",
|
|
"issue_credit": 10,
|
|
"is_superuser": False,
|
|
}
|
|
],
|
|
}
|
|
|
|
|
|
def test_set_github_token(mock_service):
|
|
# Mock user as admin and return a list of users
|
|
mock_service["user_service"].set_github_token = AsyncMock(return_value=None)
|
|
|
|
# Test the list endpoint
|
|
response = client.put(
|
|
"user/set-github-token/", json={"github_token": "ghp_1234567890abcdef1234567890abcdef1234"}
|
|
)
|
|
assert response.status_code == 200
|
|
assert response.json() == {
|
|
"code": 200,
|
|
"message": "success",
|
|
"data": None,
|
|
}
|