74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
|
package frp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestAgentEndpointMissingDetection 锁定旧 Agent 的 404 / 405 兼容提示，且不把其它错误误判为版本过低。
|
||
|
|
func TestAgentEndpointMissingDetection(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
status int
|
||
|
|
minVersion string
|
||
|
|
route bool
|
||
|
|
wantMissing bool
|
||
|
|
}{
|
||
|
|
{"404 判定版本过低", http.StatusNotFound, agentMinVersionInsights, false, true},
|
||
|
|
{"405 判定版本过低", http.StatusMethodNotAllowed, agentMinVersionRouteVersions, true, true},
|
||
|
|
{"500 不判定版本过低", http.StatusInternalServerError, agentMinVersionInsights, false, false},
|
||
|
|
{"400 不判定版本过低", http.StatusBadRequest, agentMinVersionInsights, false, false},
|
||
|
|
}
|
||
|
|
for _, tc := range cases {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
||
|
|
w.WriteHeader(tc.status)
|
||
|
|
_, _ = w.Write([]byte(`{"error":"test"}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
client := NewAgentClient(server.URL, "token")
|
||
|
|
var err error
|
||
|
|
if tc.route {
|
||
|
|
_, err = client.RouteVersions(context.Background(), "example.com")
|
||
|
|
} else {
|
||
|
|
_, err = client.Metrics(context.Background())
|
||
|
|
}
|
||
|
|
if tc.wantMissing {
|
||
|
|
if !errors.Is(err, ErrAgentEndpointMissing) {
|
||
|
|
t.Fatalf("应判定为端点缺失：%v", err)
|
||
|
|
}
|
||
|
|
msg := err.Error()
|
||
|
|
if !strings.Contains(msg, tc.minVersion) || !strings.Contains(msg, "升级 Agent") {
|
||
|
|
t.Fatalf("错误信息应包含最低版本与升级提示：%q", msg)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if errors.Is(err, ErrAgentEndpointMissing) {
|
||
|
|
t.Fatalf("不应误判为版本过低：%v", err)
|
||
|
|
}
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("应返回错误")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAgentStatusVersionField(t *testing.T) {
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
||
|
|
_, _ = w.Write([]byte(`{"agent_version":"0.10.0","listen_addr":"127.0.0.1:7700"}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
client := NewAgentClient(server.URL, "token")
|
||
|
|
status, err := client.Status(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if status["agent_version"] != "0.10.0" {
|
||
|
|
t.Fatalf("agent_version 解析失败：%#v", status)
|
||
|
|
}
|
||
|
|
}
|