HeshamTB
1a611616bd
- Working basic wg controls
- Working ip_pool allocation
- Working basic HTTP API
Signed-off-by: HeshamTB <hishaminv@gmail.com>
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package hvpnnode3
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestPoolIP4(t *testing.T) {
|
|
pool, err := NewPool("10.2.0.0/16")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if l := len(pool.available); l != 65534 {
|
|
t.Error("Expected 65534 available addresses, got", l)
|
|
}
|
|
|
|
ip, err := pool.Allocate()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if l := len(pool.available); l != 65533 {
|
|
t.Error("Expected 65533 available addresses, got", l)
|
|
}
|
|
|
|
if err := pool.Free(ip); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if l := len(pool.available); l != 65534 {
|
|
t.Error("Expected 65534 available addresses, got", l)
|
|
}
|
|
|
|
if ip.String() != "10.2.0.1" {
|
|
t.Error("Wrong ip", ip)
|
|
}
|
|
|
|
if err = pool.Free(net.ParseIP("1.2.3.4")); err == nil {
|
|
t.Error("expected error trying to free invalid IP")
|
|
}
|
|
|
|
if l := len(pool.available); l != 65534 {
|
|
t.Error("Expected 65534 available addresses, got", l)
|
|
}
|
|
if err := pool.Remove(net.ParseIP("10.2.0.100"), net.ParseIP("10.2.0.200")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if l := len(pool.available); l != 65532 {
|
|
t.Error("invalid number of available addresses, ", l)
|
|
}
|
|
for _, ip := range pool.available {
|
|
if ip.Equal(net.ParseIP("10.2.0.100")) {
|
|
t.Error("IP was not removed")
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIP4To6(t *testing.T) {
|
|
_, prefix, err := net.ParseCIDR("fdad:b10c:a::/48")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ip6 := ip4To6(net.IP{10, 1, 2, 3}, prefix)
|
|
if s := net.IP(ip6).String(); s != "fdad:b10c:a::a01:203" {
|
|
t.Log(s)
|
|
t.Error("invalid IP")
|
|
}
|
|
}
|
|
|