66 lines
2.2 KiB
Go
66 lines
2.2 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/havline/havline/internal/db"
|
||
|
|
"github.com/havline/havline/internal/terminal"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestListAuditsHandler：审计接口返回落库记录；未接存储时返回空数组而不是 null。
|
||
|
|
func TestListAuditsHandler(t *testing.T) {
|
||
|
|
conn, err := db.Open(filepath.Join(t.TempDir(), "audit.db"))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("打开数据库失败：%v", err)
|
||
|
|
}
|
||
|
|
defer conn.Close()
|
||
|
|
if err := db.Migrate(context.Background(), conn, filepath.Join("..", "..", "migrations")); err != nil {
|
||
|
|
t.Fatalf("迁移失败：%v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
store := terminal.NewAuditStore(conn)
|
||
|
|
now := time.Now().UTC()
|
||
|
|
entry := terminal.AuditEntry{
|
||
|
|
SessionID: "s1", ServerID: 3, Target: "东京 VPS · root@10.0.0.1",
|
||
|
|
User: "admin", ClientIP: "192.168.1.9",
|
||
|
|
StartedAt: now.Format(time.RFC3339), EndedAt: now.Add(time.Second).Format(time.RFC3339),
|
||
|
|
DurationMS: 1_000, Reason: "用户关闭",
|
||
|
|
}
|
||
|
|
if err := store.WriteAudit(context.Background(), entry); err != nil {
|
||
|
|
t.Fatalf("写入审计失败：%v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
handler := NewTerminalHandler(TerminalHandlerOptions{Audits: store})
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
handler.ListAudits(rec, httptest.NewRequest(http.MethodGet, "/api/terminal/audits", nil))
|
||
|
|
if rec.Code != http.StatusOK {
|
||
|
|
t.Fatalf("状态码不符：%d %s", rec.Code, rec.Body.String())
|
||
|
|
}
|
||
|
|
var body struct {
|
||
|
|
Entries []terminal.AuditEntry `json:"entries"`
|
||
|
|
}
|
||
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
||
|
|
t.Fatalf("解析响应失败：%v", err)
|
||
|
|
}
|
||
|
|
if len(body.Entries) != 1 {
|
||
|
|
t.Fatalf("期望 1 条审计，实际 %d 条", len(body.Entries))
|
||
|
|
}
|
||
|
|
if body.Entries[0].User != "admin" || body.Entries[0].Reason != "用户关闭" {
|
||
|
|
t.Fatalf("审计字段不符：%+v", body.Entries[0])
|
||
|
|
}
|
||
|
|
|
||
|
|
// 未接存储时也要给前端一个数组，避免前端拿到 null 崩掉
|
||
|
|
empty := NewTerminalHandler(TerminalHandlerOptions{})
|
||
|
|
emptyRec := httptest.NewRecorder()
|
||
|
|
empty.ListAudits(emptyRec, httptest.NewRequest(http.MethodGet, "/api/terminal/audits", nil))
|
||
|
|
if got := emptyRec.Body.String(); got != "{\"entries\":[]}\n" {
|
||
|
|
t.Fatalf("空审计响应不符：%s", got)
|
||
|
|
}
|
||
|
|
}
|