github/jwhited-wgsd
1
0
Fork 0
mirror of https://github.com/jwhited/wgsd.git synced 2025-04-03 18:49:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add tests for self-serve

Browse Source
This commit is contained in:
Tobias Krischer 2020-11-01 16:40:22 +01:00
parent 0edb1df552
commit 7c47784077
No known key found for this signature in database
GPG Key ID: E9D8FDB171E04060
2 changed files with 60 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
setup_test.go
View File

@ -22,9 +22,14 @@ func TestSetup(t *testing.T) {
"wgsd example.com.",
true,
},
{
"with local ip",
"wgsd example.com. wg0 127.0.0.1",
false,
},
{
"too many tokens",
"wgsd example.com. wg0 extra",
"wgsd example.com. wg0 127.0.0.1 extra",
true,
},
}

62
wgsd_test.go
View File

@ -16,13 +16,16 @@ import (
)
type mockClient struct {
host wgtypes.Peer
peers []wgtypes.Peer
}
func (m *mockClient) Device(d string) (*wgtypes.Device, error) {
return &wgtypes.Device{
Name: d,
Peers: m.peers,
Name: d,
PublicKey: m.host.PublicKey,
ListenPort: m.host.Endpoint.Port,
Peers: m.peers,
}, nil
}
@ -35,22 +38,32 @@ func base32EqualsPubKey(t *testing.T, b32 string, pubKey wgtypes.Key) bool {
}
func TestWGSD(t *testing.T) {
hostkey := [32]byte{}
hostkey[0] = 1
host := wgtypes.Peer{
Endpoint: &net.UDPAddr{
IP: net.ParseIP("127.0.0.1"),
Port: 1,
},
PublicKey: hostkey,
}
hostb32 := strings.ToLower(base32.StdEncoding.EncodeToString(host.PublicKey[:]))
key1 := [32]byte{}
key1[0] = 1
key1[0] = 2
peer1 := wgtypes.Peer{
Endpoint: &net.UDPAddr{
IP: net.ParseIP("1.1.1.1"),
Port: 1,
Port: 2,
},
PublicKey: key1,
}
peer1b32 := strings.ToLower(base32.StdEncoding.EncodeToString(peer1.PublicKey[:]))
key2 := [32]byte{}
key2[0] = 2
key2[0] = 3
peer2 := wgtypes.Peer{
Endpoint: &net.UDPAddr{
IP: net.ParseIP("::2"),
Port: 2,
Port: 3,
},
PublicKey: key2,
}
@ -58,10 +71,12 @@ func TestWGSD(t *testing.T) {
p := &WGSD{
Next: test.ErrorHandler(),
client: &mockClient{
host: host,
peers: []wgtypes.Peer{peer1, peer2},
},
zone: "example.com.",
device: "wg0",
wgIP: host.Endpoint.IP,
}
testCases := []test.Case{
@ -70,16 +85,39 @@ func TestWGSD(t *testing.T) {
Qtype: dns.TypePTR,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.PTR(fmt.Sprintf("_wireguard._udp.example.com. 0 IN PTR %s._wireguard._udp.example.com.", hostb32)),
test.PTR(fmt.Sprintf("_wireguard._udp.example.com. 0 IN PTR %s._wireguard._udp.example.com.", peer1b32)),
test.PTR(fmt.Sprintf("_wireguard._udp.example.com. 0 IN PTR %s._wireguard._udp.example.com.", peer2b32)),
},
},
test.Case{
Qname: fmt.Sprintf("_wireguard._udp.example.com."),
Qtype: dns.TypeSRV,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.SRV(fmt.Sprintf("_wireguard._udp.example.com. 0 IN SRV 0 0 1 %s.example.com.", hostb32)),
},
Extra: []dns.RR{
test.A(fmt.Sprintf("%s.example.com. 0 IN A %s", hostb32, host.Endpoint.IP.String())),
},
},
test.Case{
Qname: fmt.Sprintf("%s._wireguard._udp.example.com.", hostb32),
Qtype: dns.TypeSRV,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.SRV(fmt.Sprintf("%s._wireguard._udp.example.com. 0 IN SRV 0 0 1 %s.example.com.", hostb32, hostb32)),
},
Extra: []dns.RR{
test.A(fmt.Sprintf("%s.example.com. 0 IN A %s", hostb32, host.Endpoint.IP.String())),
},
},
test.Case{
Qname: fmt.Sprintf("%s._wireguard._udp.example.com.", peer1b32),
Qtype: dns.TypeSRV,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.SRV(fmt.Sprintf("%s._wireguard._udp.example.com. 0 IN SRV 0 0 1 %s.example.com.", peer1b32, peer1b32)),
test.SRV(fmt.Sprintf("%s._wireguard._udp.example.com. 0 IN SRV 0 0 2 %s.example.com.", peer1b32, peer1b32)),
},
Extra: []dns.RR{
test.A(fmt.Sprintf("%s.example.com. 0 IN A %s", peer1b32, peer1.Endpoint.IP.String())),
@ -90,12 +128,20 @@ func TestWGSD(t *testing.T) {
Qtype: dns.TypeSRV,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.SRV(fmt.Sprintf("%s._wireguard._udp.example.com. 0 IN SRV 0 0 2 %s.example.com.", peer2b32, peer2b32)),
test.SRV(fmt.Sprintf("%s._wireguard._udp.example.com. 0 IN SRV 0 0 3 %s.example.com.", peer2b32, peer2b32)),
},
Extra: []dns.RR{
test.AAAA(fmt.Sprintf("%s.example.com. 0 IN AAAA %s", peer2b32, peer2.Endpoint.IP.String())),
},
},
test.Case{
Qname: fmt.Sprintf("%s.example.com.", hostb32),
Qtype: dns.TypeA,
Rcode: dns.RcodeSuccess,
Answer: []dns.RR{
test.A(fmt.Sprintf("%s.example.com. 0 IN A %s", hostb32, host.Endpoint.IP.String())),
},
},
test.Case{
Qname: fmt.Sprintf("%s.example.com.", peer1b32),
Qtype: dns.TypeA,