feat: init commint
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.core.management import call_command
|
||||
|
||||
from apps.agents.models import AgentRun
|
||||
from apps.agents.models import AgentDefinition, AgentRun
|
||||
from apps.agents.services import build_config_snapshot
|
||||
from apps.modelhub.services import CompletionResult
|
||||
|
||||
@@ -18,9 +18,15 @@ def login_admin(client):
|
||||
def test_admin_publishes_and_runs_agent(client, db, monkeypatch):
|
||||
login_admin(client)
|
||||
definition, version = create_agent()
|
||||
outputs = iter(
|
||||
[
|
||||
CompletionResult('{"summary":"ok"}', 3, 2),
|
||||
CompletionResult('{"reply":"ok"}', 3, 2),
|
||||
]
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"apps.modelhub.services._request_completion",
|
||||
lambda *args, **kwargs: CompletionResult('{"summary":"ok","reply":"ok"}', 3, 2),
|
||||
lambda *args, **kwargs: next(outputs),
|
||||
)
|
||||
|
||||
published = client.post(f"/api/agents/{definition.id}/publish")
|
||||
@@ -57,6 +63,18 @@ def test_publishing_rejects_draft_skill(client, db):
|
||||
assert "已发布 Skill" in response.json()["detail"]
|
||||
|
||||
|
||||
def test_publish_returns_structured_error_code_for_unverified_model(client, db):
|
||||
login_admin(client)
|
||||
definition, version = create_agent()
|
||||
version.model_deployment.structured_output_test_status = "untested"
|
||||
version.model_deployment.save(update_fields=["structured_output_test_status"])
|
||||
|
||||
response = client.post(f"/api/agents/{definition.id}/publish")
|
||||
|
||||
assert response.status_code == 400
|
||||
assert response.json()["detail"].startswith("structured_output_not_verified:")
|
||||
|
||||
|
||||
def test_draft_steps_can_override_model_and_clone_binding(client, db):
|
||||
login_admin(client)
|
||||
definition, version = create_agent()
|
||||
@@ -152,6 +170,48 @@ def test_agent_run_creator_can_view_own_detail_without_global_run_permission(cli
|
||||
assert outsider_response.status_code == 403
|
||||
|
||||
|
||||
def test_agent_run_list_is_paginated_after_filtering(client, db):
|
||||
login_admin(client)
|
||||
definition, version = create_agent(published=True)
|
||||
runs = [
|
||||
AgentRun.objects.create(
|
||||
definition=definition,
|
||||
agent_version=version,
|
||||
status="failed" if index < 2 else "queued",
|
||||
input_payload={"sequence": index},
|
||||
)
|
||||
for index in range(25)
|
||||
]
|
||||
|
||||
response = client.get("/api/agent-runs?status=queued&page=2&page_size=10")
|
||||
|
||||
assert response.status_code == 200
|
||||
payload = response.json()
|
||||
assert payload["page"] == 2
|
||||
assert payload["page_size"] == 10
|
||||
assert payload["total"] == 23
|
||||
assert payload["total_pages"] == 3
|
||||
assert len(payload["items"]) == 10
|
||||
assert all(item["status"] == "queued" for item in payload["items"])
|
||||
assert payload["items"][0]["public_id"] == str(runs[14].public_id)
|
||||
|
||||
|
||||
def test_agent_run_list_clamps_invalid_pagination_values(client, db):
|
||||
login_admin(client)
|
||||
definition, version = create_agent(published=True)
|
||||
AgentRun.objects.create(
|
||||
definition=definition,
|
||||
agent_version=version,
|
||||
input_payload={"problem": "无法登录"},
|
||||
)
|
||||
|
||||
response = client.get("/api/agent-runs?page=0&page_size=500")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.json()["page"] == 1
|
||||
assert response.json()["page_size"] == 100
|
||||
|
||||
|
||||
def test_queued_run_can_be_cancelled(client, db, monkeypatch):
|
||||
login_admin(client)
|
||||
definition, version = create_agent(published=True)
|
||||
@@ -180,10 +240,15 @@ def test_queued_run_can_be_cancelled(client, db, monkeypatch):
|
||||
|
||||
def test_seed_command_creates_support_agent(client, db):
|
||||
deployment = create_deployment()
|
||||
business_agent = AgentDefinition.objects.create(
|
||||
code="support-analysis",
|
||||
name="支持分析 Agent",
|
||||
)
|
||||
|
||||
call_command("seed_ai_demo", deployment_id=deployment.id)
|
||||
|
||||
from apps.agents.models import AgentDefinition
|
||||
|
||||
agent = AgentDefinition.objects.get(code="support-analysis")
|
||||
assert agent.versions.get(status="draft").steps.count() == 3
|
||||
demo_agent = AgentDefinition.objects.get(code="support-analysis-demo")
|
||||
assert demo_agent.versions.get(status="draft").steps.count() == 3
|
||||
business_agent.refresh_from_db()
|
||||
assert business_agent.versions.count() == 0
|
||||
assert business_agent.active_version is None
|
||||
|
||||
Reference in New Issue
Block a user