From ccf40fefafddebca11e4a4ba2f8df71248af8f08 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= Date: Thu, 4 Sep 2025 13:15:38 +0200 Subject: [PATCH] Fix proxy tests --- client/iface/wgproxy/proxy_linux_test.go | 2 +- client/iface/wgproxy/proxy_seed_test.go | 3 ++- client/iface/wgproxy/proxy_test.go | 3 ++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/client/iface/wgproxy/proxy_linux_test.go b/client/iface/wgproxy/proxy_linux_test.go index 1f81077a3..9526e91d2 100644 --- a/client/iface/wgproxy/proxy_linux_test.go +++ b/client/iface/wgproxy/proxy_linux_test.go @@ -62,7 +62,7 @@ func seedProxyForProxyCloseByRemoteConn() ([]proxyInstance, error) { closeFn: func() error { return nil }, } pl = append(pl, pUDP) - wgAddress, err := wgaddr.ParseWGAddress("10.0.0.1") + wgAddress, err := wgaddr.ParseWGAddress("10.0.0.1/32") if err != nil { return nil, err } diff --git a/client/iface/wgproxy/proxy_seed_test.go b/client/iface/wgproxy/proxy_seed_test.go index c21b5275d..4d244f18a 100644 --- a/client/iface/wgproxy/proxy_seed_test.go +++ b/client/iface/wgproxy/proxy_seed_test.go @@ -6,6 +6,7 @@ import ( "net" "github.com/netbirdio/netbird/client/iface/bind" + "github.com/netbirdio/netbird/client/iface/wgaddr" bindproxy "github.com/netbirdio/netbird/client/iface/wgproxy/bind" ) @@ -17,7 +18,7 @@ func seedProxies() ([]proxyInstance, error) { func seedProxyForProxyCloseByRemoteConn() ([]proxyInstance, error) { pl := make([]proxyInstance, 0) - wgAddress, err := wgaddr.ParseWGAddress("10.0.0.1") + wgAddress, err := wgaddr.ParseWGAddress("10.0.0.1/32") if err != nil { return nil, err } diff --git a/client/iface/wgproxy/proxy_test.go b/client/iface/wgproxy/proxy_test.go index bbbb6dd51..1aeab66b7 100644 --- a/client/iface/wgproxy/proxy_test.go +++ b/client/iface/wgproxy/proxy_test.go @@ -93,8 +93,9 @@ func TestProxyCloseByRemoteConn(t *testing.T) { for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { + addr, _ := net.ResolveUDPAddr("udp", "100.108.135.221:51892") relayedConn := newMockConn() - err := tt.proxy.AddTurnConn(ctx, nil, relayedConn) + err := tt.proxy.AddTurnConn(ctx, addr, relayedConn) if err != nil { t.Errorf("error: %v", err) }