mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-25 08:09:30 +01:00
0c039274a4
This update adds new relay integration for NetBird clients. The new relay is based on web sockets and listens on a single port. - Adds new relay implementation with websocket with single port relaying mechanism - refactor peer connection logic, allowing upgrade and downgrade from/to P2P connection - peer connections are faster since it connects first to relay and then upgrades to P2P - maintains compatibility with old clients by not using the new relay - updates infrastructure scripts with new relay service
104 lines
2.1 KiB
Go
104 lines
2.1 KiB
Go
package healthcheck
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
// override the health check interval to speed up the test
|
|
healthCheckInterval = 2 * time.Second
|
|
healthCheckTimeout = 100 * time.Millisecond
|
|
code := m.Run()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func TestNewHealthPeriod(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
hc := NewSender()
|
|
go hc.StartHealthCheck(ctx)
|
|
|
|
iterations := 0
|
|
for i := 0; i < 3; i++ {
|
|
select {
|
|
case <-hc.HealthCheck:
|
|
iterations++
|
|
hc.OnHCResponse()
|
|
case <-hc.Timeout:
|
|
t.Fatalf("health check is timed out")
|
|
case <-time.After(healthCheckInterval + 100*time.Millisecond):
|
|
t.Fatalf("health check not received")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNewHealthFailed(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
hc := NewSender()
|
|
go hc.StartHealthCheck(ctx)
|
|
|
|
select {
|
|
case <-hc.Timeout:
|
|
case <-time.After(healthCheckInterval + healthCheckTimeout + 100*time.Millisecond):
|
|
t.Fatalf("health check is not timed out")
|
|
}
|
|
}
|
|
|
|
func TestNewHealthcheckStop(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
hc := NewSender()
|
|
go hc.StartHealthCheck(ctx)
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
cancel()
|
|
|
|
select {
|
|
case _, ok := <-hc.HealthCheck:
|
|
if ok {
|
|
t.Fatalf("health check on received")
|
|
}
|
|
case _, ok := <-hc.Timeout:
|
|
if ok {
|
|
t.Fatalf("health check on received")
|
|
}
|
|
case <-ctx.Done():
|
|
// expected
|
|
case <-time.After(10 * time.Second):
|
|
t.Fatalf("is not exited")
|
|
}
|
|
}
|
|
|
|
func TestTimeoutReset(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
hc := NewSender()
|
|
go hc.StartHealthCheck(ctx)
|
|
|
|
iterations := 0
|
|
for i := 0; i < 3; i++ {
|
|
select {
|
|
case <-hc.HealthCheck:
|
|
iterations++
|
|
hc.OnHCResponse()
|
|
case <-hc.Timeout:
|
|
t.Fatalf("health check is timed out")
|
|
case <-time.After(healthCheckInterval + 100*time.Millisecond):
|
|
t.Fatalf("health check not received")
|
|
}
|
|
}
|
|
|
|
select {
|
|
case <-hc.HealthCheck:
|
|
case <-hc.Timeout:
|
|
// expected
|
|
case <-ctx.Done():
|
|
t.Fatalf("context is done")
|
|
case <-time.After(10 * time.Second):
|
|
t.Fatalf("is not exited")
|
|
}
|
|
}
|