TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-05-04 14:41:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

derp: add (*Server).IsClientConnectedForTest func. (#4331)

Browse Source 
This allows tests to verfiy that a DERP connection was actually
established.

Related to #4326
Updates tailscale/corp#2579

Signed-off-by: Maisem Ali <maisem@tailscale.com>
This commit is contained in:
Maisem Ali 2022-03-30 10:50:50 -07:00 committed by GitHub
parent 3d180c0376
commit 9f604f2bd3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
derp/derp_server.go
View File

@ -391,6 +391,18 @@ func (s *Server) isClosed() bool {
return s.closed return s.closed
} }
// IsClientConnectedForTest reports whether the client with specified key is connected.
// This is used in tests to verify that nodes are connected.
func (s *Server) IsClientConnectedForTest(k key.NodePublic) bool {
s.mu.Lock()
defer s.mu.Unlock()
x, ok := s.clients[k]
if !ok {
return false
}
return x.ActiveClient() != nil
}
// Accept adds a new connection to the server and serves it. // Accept adds a new connection to the server and serves it.
// //
// The provided bufio ReadWriter must be already connected to nc. // The provided bufio ReadWriter must be already connected to nc.