mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 09:50:47 +01:00
ddea001170
Rely on net.ListenUDP to get an available port for wireguard in case the configured one is in use --------- Co-authored-by: Viktor Liu <17948409+lixmal@users.noreply.github.com>
62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package internal
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func Test_freePort(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
port int
|
|
want int
|
|
shouldMatch bool
|
|
}{
|
|
{
|
|
name: "not provided, fallback to default",
|
|
port: 0,
|
|
want: 51820,
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
name: "provided and available",
|
|
port: 51821,
|
|
want: 51821,
|
|
shouldMatch: true,
|
|
},
|
|
{
|
|
name: "provided and not available",
|
|
port: 51830,
|
|
want: 51830,
|
|
shouldMatch: false,
|
|
},
|
|
}
|
|
c1, err := net.ListenUDP("udp", &net.UDPAddr{Port: 51830})
|
|
if err != nil {
|
|
t.Errorf("freePort error = %v", err)
|
|
}
|
|
defer func(c1 *net.UDPConn) {
|
|
_ = c1.Close()
|
|
}(c1)
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := freePort(tt.port)
|
|
|
|
if err != nil {
|
|
t.Errorf("got an error while getting free port: %v", err)
|
|
}
|
|
|
|
if tt.shouldMatch && got != tt.want {
|
|
t.Errorf("got a different port %v, want %v", got, tt.want)
|
|
}
|
|
|
|
if !tt.shouldMatch && got == tt.want {
|
|
t.Errorf("got the same port %v, want a different port", tt.want)
|
|
}
|
|
})
|
|
|
|
}
|
|
}
|