Files
3x-ui/tools/openapigen/walker_test.go
T
YoungReckless4 35e09191de feat(amneziawg): support persistent keepalive ranges (#6881)
AmneziaWG clients accept a randomised PersistentKeepalive range such as 15-25,
which makes keepalive timing harder to fingerprint; the panel only stored a
single number. A client's keepalive may now be a range, validated the way
AmneziaWG parses it. The downloadable .conf and vpn:// profile carry the range,
Clash/Mihomo output gets its lower bound, and plain WireGuard stays scalar-only.
The range lives in an additive wg_keep_alive_range column, so the integer
column and an older panel binary keep working after a rollback.
2026-10-24 06:36:49 +02:00

71 lines
2.4 KiB
Go

package main
import (
"bytes"
"os"
"path/filepath"
"strings"
"testing"
)
func TestWalkPackagesExportsAllowedStructsWithoutDIInterfaces(t *testing.T) {
dir := t.TempDir()
source := `package service
type HappLinkGenerator interface {
Generate()
}
type HappLinkResult struct {
EncryptedLink string
}
`
if err := os.WriteFile(filepath.Join(dir, "service.go"), []byte(source), 0o644); err != nil {
t.Fatalf("write service fixture: %v", err)
}
schemas, aliases, err := walkPackages([]packageRequest{{
Path: dir,
StructAllow: setOf("HappLinkResult"),
}})
if err != nil {
t.Fatalf("walk packages: %v", err)
}
var generated bytes.Buffer
if err := emitTypes(&generated, schemas, aliases); err != nil {
t.Fatalf("emit types: %v", err)
}
apiSurface := generated.String()
if !strings.Contains(apiSurface, "export interface HappLinkResult") {
t.Fatalf("generated types omit allowed response schema:\n%s", apiSurface)
}
if strings.Contains(apiSurface, "HappLinkGenerator") {
t.Fatalf("generated types expose DI interface:\n%s", apiSurface)
}
}
func TestKeepAliveOverrideEmitsNumberOrString(t *testing.T) {
dir := t.TempDir()
source := "package model\ntype Client struct { KeepAlive *int `json:\"keepAlive,omitempty\"` }\n"
if err := os.WriteFile(filepath.Join(dir, "client.go"), []byte(source), 0o644); err != nil {
t.Fatalf("write fixture: %v", err)
}
union := TypeRef{Kind: KindUnion, Variants: []TypeRef{{Kind: KindInt}, {Kind: KindString}}}
schemas, _, err := walkPackages([]packageRequest{{Path: dir, StructAllow: setOf("Client"), Overrides: map[string][]walkOverride{"Client": {{Field: "KeepAlive", Type: union}}}}})
if err != nil {
t.Fatalf("walk fixture: %v", err)
}
var types bytes.Buffer
if err := emitTypes(&types, schemas, nil); err != nil || !strings.Contains(types.String(), "keepAlive?: number | string;") {
t.Fatalf("generated TypeScript keepAlive = %s, err = %v", types.String(), err)
}
var zod bytes.Buffer
if err := emitZod(&zod, schemas, nil); err != nil || !strings.Contains(zod.String(), "z.union([z.number().int(), z.string()]).optional()") {
t.Fatalf("generated Zod keepAlive = %s, err = %v", zod.String(), err)
}
var schema bytes.Buffer
if err := emitJSONSchema(&schema, schemas, nil); err != nil || !strings.Contains(schema.String(), `"oneOf"`) {
t.Fatalf("generated JSON schema keepAlive = %s, err = %v", schema.String(), err)
}
}