52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
|
package cloudflared
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestManualRoutesFromConfig(t *testing.T) {
|
|||
|
|
path := filepath.Join(t.TempDir(), "config.yml")
|
|||
|
|
content := `tunnel: test
|
|||
|
|
ingress:
|
|||
|
|
- hostname: a.example.com
|
|||
|
|
service: http://127.0.0.1:8080
|
|||
|
|
- hostname: b.example.com
|
|||
|
|
service: http://127.0.0.1:9090
|
|||
|
|
- service: http_status:404
|
|||
|
|
`
|
|||
|
|
if err := os.WriteFile(path, []byte(content), 0o640); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
result, err := manualRoutes(path)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if len(result.Routes) != 2 || result.Routes[0].Hostname != "a.example.com" || result.Routes[1].Service != "http://127.0.0.1:9090" {
|
|||
|
|
t.Fatalf("手工路由解析不符：%#v", result)
|
|||
|
|
}
|
|||
|
|
if result.CatchAll != "http_status:404" {
|
|||
|
|
t.Fatalf("catch-all 不符：%q", result.CatchAll)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestBuildManualIngress(t *testing.T) {
|
|||
|
|
ingress, hostnames, err := buildManualIngress([]RouteInput{
|
|||
|
|
{Hostname: "A.Example.com", Path: "*", Service: "http://127.0.0.1:8080"},
|
|||
|
|
{Hostname: "b.example.com", Service: "https://127.0.0.1:8443"},
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if len(ingress) != 3 || ingress[0].Hostname != "a.example.com" || ingress[1].Hostname != "b.example.com" {
|
|||
|
|
t.Fatalf("ingress 结果不符：%#v", ingress)
|
|||
|
|
}
|
|||
|
|
if ingress[0].Path != "" || ingress[1].Path != "" {
|
|||
|
|
t.Fatalf("通配路径不应写入 Cloudflare ingress：%#v", ingress)
|
|||
|
|
}
|
|||
|
|
if len(hostnames) != 2 || hostnames[1] != "b.example.com" {
|
|||
|
|
t.Fatalf("hostnames 结果不符：%#v", hostnames)
|
|||
|
|
}
|
|||
|
|
}
|