56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package proxy
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestStorePersistsExitsAndFiltersLocal(t *testing.T) {
|
|||
|
|
store, _ := newTestStore(t)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
rule, err := store.Create(ctx, CreateInput{
|
|||
|
|
Upstream: "http://192.168.1.10:8080",
|
|||
|
|
ListenPort: 8443,
|
|||
|
|
ListenIPv4: true,
|
|||
|
|
Hosts: []string{"cf.example.com"},
|
|||
|
|
Enabled: true,
|
|||
|
|
Exits: []string{ExitCloudflare},
|
|||
|
|
CFTunnelID: 7,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if rule.CFTunnelID != 7 || len(rule.Exits) != 1 || rule.Exits[0] != ExitCloudflare {
|
|||
|
|
t.Fatalf("Cloudflare 出口未持久化：%#v", rule)
|
|||
|
|
}
|
|||
|
|
localRules, err := store.ListEnabledLocal(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if len(localRules) != 0 {
|
|||
|
|
t.Fatalf("仅 Cloudflare 规则不应生成本机 Nginx：%#v", localRules)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
enabled := true
|
|||
|
|
updated, err := store.Update(ctx, rule.ID, UpdateInput{
|
|||
|
|
Exits: &[]string{ExitLocal, ExitCloudflare},
|
|||
|
|
Enabled: &enabled,
|
|||
|
|
CFTunnelID: int64Ptr(7),
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if len(updated.Exits) != 2 {
|
|||
|
|
t.Fatalf("本机 + Cloudflare 出口未保存：%#v", updated.Exits)
|
|||
|
|
}
|
|||
|
|
localRules, err = store.ListEnabledLocal(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if len(localRules) != 1 || localRules[0].ID != rule.ID {
|
|||
|
|
t.Fatalf("启用本机出口后应进入 Nginx 规则：%#v", localRules)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func int64Ptr(v int64) *int64 { return &v }
|