68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
|
package frp
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 自定义 404 页面：填了「页面内容」就用 Havline 托管的那份（用户不用自己去 VPS 放文件），
|
|||
|
|
// 只有手填路径时才用路径，两者都没填就不输出 custom404Page。
|
|||
|
|
func TestBuildFRPSServerConfigCustom404Page(t *testing.T) {
|
|||
|
|
svc, ctx := newVhostPortTestService(t)
|
|||
|
|
|
|||
|
|
base := func() ServerOptions {
|
|||
|
|
return ServerOptions{AuthMethod: "token", LogLevel: "info", Protocol: "tcp"}
|
|||
|
|
}
|
|||
|
|
manualPath := base()
|
|||
|
|
manualPath.Custom404Page = "/var/www/404.html"
|
|||
|
|
managed := base()
|
|||
|
|
managed.Custom404HTML = "<html><body>404</body></html>"
|
|||
|
|
both := base()
|
|||
|
|
both.Custom404Page = "/var/www/404.html"
|
|||
|
|
both.Custom404HTML = "<html><body>404</body></html>"
|
|||
|
|
|
|||
|
|
cases := []struct {
|
|||
|
|
name string
|
|||
|
|
key string
|
|||
|
|
options ServerOptions
|
|||
|
|
want string
|
|||
|
|
notWant string
|
|||
|
|
}{
|
|||
|
|
{name: "都没填", key: "none", options: base(), notWant: "custom404Page"},
|
|||
|
|
{name: "只填路径", key: "path", options: manualPath, want: `custom404Page = "/var/www/404.html"`},
|
|||
|
|
{
|
|||
|
|
name: "填了页面内容", key: "html", options: managed,
|
|||
|
|
want: `custom404Page = "/etc/frp/havline-404.html"`,
|
|||
|
|
notWant: "/var/www/404.html",
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "两者都填时以内容为准", key: "both", options: both,
|
|||
|
|
want: `custom404Page = "/etc/frp/havline-404.html"`,
|
|||
|
|
notWant: `custom404Page = "/var/www/404.html"`,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tc := range cases {
|
|||
|
|
t.Run(tc.name, func(t *testing.T) {
|
|||
|
|
in := baseVhostServerInput()
|
|||
|
|
// 每个子用例用独立名字，避免同一份测试库里重名
|
|||
|
|
in.Name = "vps-" + tc.key
|
|||
|
|
in.Options = tc.options
|
|||
|
|
server, err := svc.CreateServer(ctx, in)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("创建服务端失败：%v", err)
|
|||
|
|
}
|
|||
|
|
content, err := svc.buildFRPSServerConfig(ctx, server)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("生成 frps.toml 失败：%v", err)
|
|||
|
|
}
|
|||
|
|
if tc.want != "" && !strings.Contains(content, tc.want) {
|
|||
|
|
t.Fatalf("期望包含 %q：\n%s", tc.want, content)
|
|||
|
|
}
|
|||
|
|
if tc.notWant != "" && strings.Contains(content, tc.notWant) {
|
|||
|
|
t.Fatalf("不应包含 %q：\n%s", tc.notWant, content)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|