mirror of
https://github.com/tailscale/tailscale.git
synced 2025-10-31 08:11:32 +01:00
16bc0a5558
For eventual use by net/udprelay.Server. Updates tailscale/corp#31164 Signed-off-by: Jordan Whited <jordan@tailscale.com>
38 lines
988 B
Go
38 lines
988 B
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package magicsock
|
|
|
|
import (
|
|
"net/netip"
|
|
"testing"
|
|
|
|
"tailscale.com/net/packet"
|
|
"tailscale.com/types/key"
|
|
)
|
|
|
|
func Test_peerMap_oneRelayEpAddrPerNK(t *testing.T) {
|
|
pm := newPeerMap()
|
|
nk := key.NewNode().Public()
|
|
ep := &endpoint{
|
|
nodeID: 1,
|
|
publicKey: nk,
|
|
}
|
|
ed := &endpointDisco{key: key.NewDisco().Public()}
|
|
ep.disco.Store(ed)
|
|
pm.upsertEndpoint(ep, key.DiscoPublic{})
|
|
vni := packet.VirtualNetworkID{}
|
|
vni.Set(1)
|
|
relayEpAddrA := epAddr{ap: netip.MustParseAddrPort("127.0.0.1:1"), vni: vni}
|
|
relayEpAddrB := epAddr{ap: netip.MustParseAddrPort("127.0.0.1:2"), vni: vni}
|
|
pm.setNodeKeyForEpAddr(relayEpAddrA, nk)
|
|
pm.setNodeKeyForEpAddr(relayEpAddrB, nk)
|
|
if len(pm.byEpAddr) != 1 {
|
|
t.Fatalf("expected 1 epAddr in byEpAddr, got: %d", len(pm.byEpAddr))
|
|
}
|
|
got := pm.relayEpAddrByNodeKey[nk]
|
|
if got != relayEpAddrB {
|
|
t.Fatalf("expected relay epAddr %v, got: %v", relayEpAddrB, got)
|
|
}
|
|
}
|