66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
|
package agent
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestValidateRoute(t *testing.T) {
|
|||
|
|
cases := []struct {
|
|||
|
|
name string
|
|||
|
|
req routeRequest
|
|||
|
|
ok bool
|
|||
|
|
}{
|
|||
|
|
{"正常", routeRequest{Domain: "a.example.com", Upstream: "127.0.0.1:8080"}, true},
|
|||
|
|
{"非法域名", routeRequest{Domain: "bad domain", Upstream: "127.0.0.1:8080"}, false},
|
|||
|
|
{"空上游", routeRequest{Domain: "a.example.com"}, false},
|
|||
|
|
{"上游带分号", routeRequest{Domain: "a.example.com", Upstream: "127.0.0.1:80; rm -rf /"}, false},
|
|||
|
|
{"相对证书目录", routeRequest{Domain: "a.example.com", Upstream: "127.0.0.1:80", TLS: true, CertDir: "certs"}, false},
|
|||
|
|
}
|
|||
|
|
for _, c := range cases {
|
|||
|
|
err := validateRoute(c.req)
|
|||
|
|
if c.ok && err != nil {
|
|||
|
|
t.Errorf("%s：期望通过，实际 %v", c.name, err)
|
|||
|
|
}
|
|||
|
|
if !c.ok && err == nil {
|
|||
|
|
t.Errorf("%s：期望报错", c.name)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestNginxRouteRendersUpstream：vhost 模板的关键行（标记、server_name、proxy_pass）。
|
|||
|
|
// 不构造 Basic Auth 场景——那条路径会往 /var/lib/fonu-agent 写 htpasswd，不适合在单测里触发。
|
|||
|
|
func TestNginxRouteRendersUpstream(t *testing.T) {
|
|||
|
|
out := nginxRoute(routeRequest{Domain: "a.example.com", Upstream: "127.0.0.1:8080"}, "")
|
|||
|
|
if !strings.HasPrefix(out, "# Managed by fonu-agent.") {
|
|||
|
|
t.Errorf("缺少托管标记，其它工具无法识别该文件归属：\n%s", out)
|
|||
|
|
}
|
|||
|
|
if !strings.Contains(out, "server_name a.example.com;") {
|
|||
|
|
t.Errorf("缺少 server_name：\n%s", out)
|
|||
|
|
}
|
|||
|
|
if !strings.Contains(out, "proxy_pass http://127.0.0.1:8080;") {
|
|||
|
|
t.Errorf("缺少 proxy_pass 回源：\n%s", out)
|
|||
|
|
}
|
|||
|
|
if strings.Contains(out, "listen 443") {
|
|||
|
|
t.Errorf("未启用 TLS 时不应监听 443：\n%s", out)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestFileSHA256 覆盖 frps 安装包的校验路径。
|
|||
|
|
func TestFileSHA256(t *testing.T) {
|
|||
|
|
path := filepath.Join(t.TempDir(), "payload.txt")
|
|||
|
|
if err := os.WriteFile(path, []byte("hello"), 0o600); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
got, err := fileSHA256(path)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
const want = "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
|
|||
|
|
if got != want {
|
|||
|
|
t.Fatalf("sha256 不符：got %s", got)
|
|||
|
|
}
|
|||
|
|
}
|