agent/ports_supported_internal_test.go

Rama main · réplica de lectura del repositorio original.

//go:build linux || (windows && amd64)

package agent

import (
	"net"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func TestOSListeningPortsGetter(t *testing.T) {
	t.Parallel()

	uut := &osListeningPortsGetter{
		cacheDuration: 1 * time.Hour,
	}

	l, err := net.Listen("tcp", "localhost:0")
	require.NoError(t, err)
	defer l.Close()

	ports, err := uut.GetListeningPorts()
	require.NoError(t, err)
	found := false
	for _, port := range ports {
		// #nosec G115 - Safe conversion as TCP port numbers are within uint16 range (0-65535)
		if port.Port == uint16(l.Addr().(*net.TCPAddr).Port) {
			found = true
			break
		}
	}
	require.True(t, found)

	// check that we cache the ports
	err = l.Close()
	require.NoError(t, err)
	portsNew, err := uut.GetListeningPorts()
	require.NoError(t, err)
	require.Equal(t, ports, portsNew)

	// note that it's unsafe to try to assert that a port does not exist in the response
	// because the OS may reallocate the port very quickly.
}

func TestOSListeningPortsGetter_IPv6(t *testing.T) {
	t.Parallel()

	uut := &osListeningPortsGetter{
		cacheDuration: 1 * time.Hour,
	}

	// Many dev servers (e.g. Next.js) bind to the IPv6 wildcard address,
	// which is dual-stack and only shows up in the IPv6 socket table.
	l, err := net.Listen("tcp", "[::]:0")
	if err != nil {
		t.Skipf("unable to listen on IPv6 wildcard address: %s", err)
	}
	defer l.Close()

	// #nosec G115 - Safe conversion as TCP port numbers are within uint16 range (0-65535)
	want := uint16(l.Addr().(*net.TCPAddr).Port)

	ports, err := uut.GetListeningPorts()
	require.NoError(t, err)
	found := false
	for _, port := range ports {
		if port.Port == want {
			found = true
			break
		}
	}
	require.True(t, found, "port %d not found in %v", want, ports)
}