877e4cf41d
将代码与测试注释改为中文以符合仓库规范；新增中文部署指南、 示例配置，并完善构建脚本对 VERSION 注入的说明。 Co-authored-by: Cursor <cursoragent@cursor.com>
118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"mfd-monitor-agent/internal/status"
|
|
)
|
|
|
|
// TestStatusReturnsCachedSnapshot 验证 /status 读取提供者缓存。
|
|
func TestStatusReturnsCachedSnapshot(t *testing.T) {
|
|
provider := StaticProvider{Snapshot: status.HostStatus{
|
|
Agent: status.AgentInfo{Version: "test", SnapshotAt: time.Now(), SnapshotStatus: "ok"},
|
|
Host: status.HostInfo{Hostname: "host-a"},
|
|
}, OK: true}
|
|
server := New(provider)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/status", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
var got status.HostStatus
|
|
if err := json.NewDecoder(rec.Body).Decode(&got); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got.Host.Hostname != "host-a" {
|
|
t.Fatalf("hostname = %q", got.Host.Hostname)
|
|
}
|
|
}
|
|
|
|
// TestHealthReportsSnapshotMetadata 验证 /health 返回快照元数据。
|
|
func TestHealthReportsSnapshotMetadata(t *testing.T) {
|
|
provider := StaticProvider{Snapshot: status.HostStatus{
|
|
Agent: status.AgentInfo{Version: "test", SnapshotAt: time.Unix(10, 0), SnapshotStatus: "partial"},
|
|
}, OK: true}
|
|
server := New(provider)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/health", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
var got map[string]any
|
|
if err := json.NewDecoder(rec.Body).Decode(&got); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got["snapshot_status"] != "partial" {
|
|
t.Fatalf("health = %+v", got)
|
|
}
|
|
}
|
|
|
|
// TestStatusReturnsServiceUnavailableBeforeSnapshot 验证 /status 不会在请求时实时采集。
|
|
func TestStatusReturnsServiceUnavailableBeforeSnapshot(t *testing.T) {
|
|
server := New(StaticProvider{OK: false})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/status", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusServiceUnavailable {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
}
|
|
|
|
// TestHealthReturnsStartingBeforeSnapshot 验证 /health 在启动阶段暴露 starting 状态。
|
|
func TestHealthReturnsStartingBeforeSnapshot(t *testing.T) {
|
|
server := New(StaticProvider{OK: false})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/health", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusServiceUnavailable {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
var got map[string]string
|
|
if err := json.NewDecoder(rec.Body).Decode(&got); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got["status"] != "starting" {
|
|
t.Fatalf("status = %q", got["status"])
|
|
}
|
|
}
|
|
|
|
// TestStatusRejectsNonGET 验证 /status 仅接受 GET 请求。
|
|
func TestStatusRejectsNonGET(t *testing.T) {
|
|
server := New(StaticProvider{OK: true})
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/status", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusMethodNotAllowed {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
}
|
|
|
|
// TestHealthRejectsNonGET 验证 /health 仅接受 GET 请求。
|
|
func TestHealthRejectsNonGET(t *testing.T) {
|
|
server := New(StaticProvider{OK: true})
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/health", nil)
|
|
rec := httptest.NewRecorder()
|
|
server.ServeHTTP(rec, req)
|
|
|
|
if rec.Code != http.StatusMethodNotAllowed {
|
|
t.Fatalf("status code = %d", rec.Code)
|
|
}
|
|
}
|