59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"codex-agent-manager/internal/app"
|
|
)
|
|
|
|
func TestAgentsEndpointReturnsAgentItems(t *testing.T) {
|
|
root := t.TempDir()
|
|
agentsDir := filepath.Join(root, "agents")
|
|
if err := os.MkdirAll(agentsDir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
content := `name = "代码员"
|
|
description = "负责实现"
|
|
`
|
|
if err := os.WriteFile(filepath.Join(agentsDir, "coder.toml"), []byte(content), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/api/agents", nil)
|
|
rec := httptest.NewRecorder()
|
|
New(app.Config{CodexHome: root, HTTPAddr: "127.0.0.1:0"}).ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, body = %s", rec.Code, rec.Body.String())
|
|
}
|
|
var body struct {
|
|
Items []struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
ParseStatus string `json:"parseStatus"`
|
|
} `json:"items"`
|
|
}
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
|
t.Fatalf("invalid json: %v", err)
|
|
}
|
|
if len(body.Items) != 1 || body.Items[0].Name != "代码员" || body.Items[0].ParseStatus != "valid" {
|
|
t.Fatalf("unexpected response: %#v", body)
|
|
}
|
|
}
|
|
|
|
func TestAgentsEndpointRejectsUnsupportedMethod(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodPost, "/api/agents", nil)
|
|
rec := httptest.NewRecorder()
|
|
New(app.Config{CodexHome: t.TempDir(), HTTPAddr: "127.0.0.1:0"}).ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusMethodNotAllowed {
|
|
t.Fatalf("status = %d, want %d", rec.Code, http.StatusMethodNotAllowed)
|
|
}
|
|
}
|