mirror of
https://github.com/jwhited/wgsd.git
synced 2025-01-18 22:09:34 +08:00
42 lines
641 B
Go
42 lines
641 B
Go
package wgsd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/caddyserver/caddy"
|
|
)
|
|
|
|
func TestSetup(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
input string
|
|
expectErr bool
|
|
}{
|
|
{
|
|
"valid input",
|
|
"wgsd example.com. wg0",
|
|
false,
|
|
},
|
|
{
|
|
"missing token",
|
|
"wgsd example.com.",
|
|
true,
|
|
},
|
|
{
|
|
"too many tokens",
|
|
"wgsd example.com. wg0 extra",
|
|
true,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
c := caddy.NewTestController("dns", tc.input)
|
|
err := setup(c)
|
|
if (err != nil) != tc.expectErr {
|
|
t.Fatalf("expectErr: %v, got err=%v", tc.expectErr, err)
|
|
}
|
|
})
|
|
}
|
|
}
|