WIP

peer/forwarder_test.go

@@ -11,12 +11,11 @@ import (
 // no real WireGuard interface is needed.
 func newTestForwarder(t *testing.T) *Forwarder {
 	t.Helper()
-	f := NewForwarder(netip.MustParseAddr("127.0.0.1"))
-	t.Cleanup(func() {
-		for b := range 256 {
-			f.RemoveEndpoint(byte(b))
-		}
-	})
+	f, err := NewForwarder(netip.MustParseAddr("127.0.0.1"))
+	if err != nil {
+		t.Fatalf("NewForwarder: %v", err)
+	}
+	t.Cleanup(f.Close)
 	return f
 }