Fix pfx dedup and add tests (#233)

net/ip_cache_test.go

@@ -0,0 +1,25 @@
+package net
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestIPCache(t *testing.T) {
+	a := &IP{
+		higher:   100,
+		lower:    200,
+		isLegacy: false,
+	}
+	b := &IP{
+		higher:   100,
+		lower:    200,
+		isLegacy: false,
+	}
+
+	x := a.Dedup()
+	y := b.Dedup()
+
+	assert.Equal(t, true, x == y)
+}

net/prefix.go

@@ -26,7 +26,7 @@ func (p *Prefix) Dedup() *Prefix {
 
 // Less compares prefixes for use in btree.Btree
 func (p *Prefix) Less(other btree.Item) bool {
-	if uintptr(unsafe.Pointer(p)) < uintptr(unsafe.Pointer(other.(*Prefix))) {
+	if uintptr(unsafe.Pointer(p.addr)) < uintptr(unsafe.Pointer(other.(*Prefix).addr)) {
 		return true
 	}
 

net/prefix_cache_test.go

@@ -0,0 +1,34 @@
+package net
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPrefixCache(t *testing.T) {
+	a := &Prefix{
+		addr: &IP{
+			higher:   100,
+			lower:    200,
+			isLegacy: false,
+		},
+		pfxlen: 64,
+	}
+	b := &Prefix{
+		addr: &IP{
+			higher:   100,
+			lower:    200,
+			isLegacy: false,
+		},
+		pfxlen: 64,
+	}
+
+	a.addr = a.addr.Dedup()
+	b.addr = b.addr.Dedup()
+
+	x := a.Dedup()
+	y := b.Dedup()
+
+	assert.Equal(t, true, x == y)
+}