mirror of
https://github.com/libp2p/go-reuseport.git
synced 2025-09-26 19:01:11 +08:00
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package reuseport
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func testDialFromListeningPort(t *testing.T, network, host string) {
|
|
lc := net.ListenConfig{
|
|
Control: Control,
|
|
}
|
|
ctx := context.Background()
|
|
ll, err := lc.Listen(ctx, network, host+":0")
|
|
if err != nil && strings.Contains(err.Error(), "cannot assign requested address") {
|
|
t.Skip(err)
|
|
}
|
|
require.NoError(t, err)
|
|
rl, err := lc.Listen(ctx, network, host+":0")
|
|
require.NoError(t, err)
|
|
d := net.Dialer{
|
|
LocalAddr: ll.Addr(),
|
|
Control: Control,
|
|
}
|
|
c, err := d.Dial(network, rl.Addr().String())
|
|
require.NoError(t, err)
|
|
c.Close()
|
|
}
|
|
|
|
func TestDialFromListeningPort(t *testing.T) {
|
|
testDialFromListeningPort(t, "tcp", "localhost")
|
|
}
|
|
|
|
func TestDialFromListeningPortTcp6(t *testing.T) {
|
|
testDialFromListeningPort(t, "tcp6", "[::1]")
|
|
}
|
|
|
|
func TestListenPacketWildcardAddress(t *testing.T) {
|
|
pc, err := ListenPacket("udp", ":0")
|
|
require.NoError(t, err)
|
|
pc.Close()
|
|
}
|
|
|
|
func TestErrorWhenDialUnresolvable(t *testing.T) {
|
|
_, err := Dial("asd", "127.0.0.1:1234", "127.0.0.1:1234")
|
|
require.ErrorIs(t, err, net.UnknownNetworkError("asd"))
|
|
_, err = Dial("tcp", "a.b.c.d:1234", "a.b.c.d:1235")
|
|
require.Error(t, err)
|
|
}
|