59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
|
package frp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/fonu/fonu/internal/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGenerateNginxGateway(t *testing.T) {
|
||
|
|
cfg := config.Config{
|
||
|
|
DataDir: "/data",
|
||
|
|
NginxDefaultHTTPPort: 18080,
|
||
|
|
NginxDefaultHTTPSPort: 9443,
|
||
|
|
FrpPIDFile: "/data/frp/frpc.pid",
|
||
|
|
}
|
||
|
|
frpCfg := Config{
|
||
|
|
ServerAddr: "vps.example.com",
|
||
|
|
ServerPort: 7000,
|
||
|
|
TLSEnabled: true,
|
||
|
|
CustomDomains: []string{"nas.example.com", "*.example.com"},
|
||
|
|
}
|
||
|
|
out, err := Generate(cfg, frpCfg, "secret-token", 18080, 9443)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `serverAddr = "vps.example.com"`) {
|
||
|
|
t.Fatalf("missing server addr: %s", out)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `localPort = 18080`) {
|
||
|
|
t.Fatalf("missing http local port: %s", out)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `type = "https2https"`) {
|
||
|
|
t.Fatalf("missing https plugin: %s", out)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `localAddr = "127.0.0.1:9443"`) {
|
||
|
|
t.Fatalf("missing https local addr: %s", out)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `transport.tls.enable = true`) {
|
||
|
|
t.Fatalf("missing tls enable: %s", out)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenerateDisablesTLSByDefault(t *testing.T) {
|
||
|
|
cfg := config.Config{DataDir: "/data", FrpPIDFile: "/data/frp/frpc.pid"}
|
||
|
|
frpCfg := Config{
|
||
|
|
ServerAddr: "vps.example.com",
|
||
|
|
ServerPort: 7000,
|
||
|
|
CustomDomains: []string{"nas.example.com"},
|
||
|
|
}
|
||
|
|
out, err := Generate(cfg, frpCfg, "secret-token", 18080, 9443)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if !strings.Contains(out, `transport.tls.enable = false`) {
|
||
|
|
t.Fatalf("expected tls disabled: %s", out)
|
||
|
|
}
|
||
|
|
}
|