mirror of
https://github.com/tailscale/tailscale.git
synced 2025-07-30 07:43:42 +00:00
1677fb1905
Updates tailscale/corp#30583 Updates tailscale/corp#30534 Updates tailscale/corp#30557 Signed-off-by: Dylan Bargatze <dylan@tailscale.com> Signed-off-by: Jordan Whited <jordan@tailscale.com> Co-authored-by: Dylan Bargatze <dylan@tailscale.com>
81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package magicsock
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"tailscale.com/disco"
|
|
"tailscale.com/types/key"
|
|
"tailscale.com/util/set"
|
|
)
|
|
|
|
func TestRelayManagerInitAndIdle(t *testing.T) {
|
|
rm := relayManager{}
|
|
rm.startUDPRelayPathDiscoveryFor(&endpoint{}, addrQuality{}, false)
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.stopWork(&endpoint{})
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.handleCallMeMaybeVia(&endpoint{c: &Conn{discoPrivate: key.NewDisco()}}, addrQuality{}, false, &disco.CallMeMaybeVia{UDPRelayEndpoint: disco.UDPRelayEndpoint{ServerDisco: key.NewDisco().Public()}})
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.handleRxDiscoMsg(&Conn{discoPrivate: key.NewDisco()}, &disco.BindUDPRelayEndpointChallenge{}, key.NodePublic{}, key.DiscoPublic{}, epAddr{})
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.handleRelayServersSet(make(set.Set[candidatePeerRelay]))
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.getServers()
|
|
<-rm.runLoopStoppedCh
|
|
|
|
rm = relayManager{}
|
|
rm.handleDERPHomeChange(key.NodePublic{}, 1)
|
|
<-rm.runLoopStoppedCh
|
|
}
|
|
|
|
func TestRelayManagerHandleDERPHomeChange(t *testing.T) {
|
|
rm := relayManager{}
|
|
servers := make(set.Set[candidatePeerRelay], 1)
|
|
c := candidatePeerRelay{
|
|
nodeKey: key.NewNode().Public(),
|
|
discoKey: key.NewDisco().Public(),
|
|
derpHomeRegionID: 1,
|
|
}
|
|
servers.Add(c)
|
|
rm.handleRelayServersSet(servers)
|
|
want := c
|
|
want.derpHomeRegionID = 2
|
|
rm.handleDERPHomeChange(c.nodeKey, 2)
|
|
got := rm.getServers()
|
|
if len(got) != 1 {
|
|
t.Fatalf("got %d servers, want 1", len(got))
|
|
}
|
|
_, ok := got[want]
|
|
if !ok {
|
|
t.Fatal("DERP home change failed to propagate")
|
|
}
|
|
}
|
|
|
|
func TestRelayManagerGetServers(t *testing.T) {
|
|
rm := relayManager{}
|
|
servers := make(set.Set[candidatePeerRelay], 1)
|
|
c := candidatePeerRelay{
|
|
nodeKey: key.NewNode().Public(),
|
|
discoKey: key.NewDisco().Public(),
|
|
}
|
|
servers.Add(c)
|
|
rm.handleRelayServersSet(servers)
|
|
got := rm.getServers()
|
|
if !servers.Equal(got) {
|
|
t.Errorf("got %v != want %v", got, servers)
|
|
}
|
|
}
|