78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
package integration
|
|
|
|
import (
|
|
"net/netip"
|
|
"testing"
|
|
|
|
"github.com/juanfont/headscale/integration/tsic"
|
|
)
|
|
|
|
func TestPingAll(t *testing.T) {
|
|
IntegrationSkip(t)
|
|
|
|
scenario, err := NewScenario()
|
|
if err != nil {
|
|
t.Errorf("failed to create scenario: %s", err)
|
|
}
|
|
|
|
spec := map[string]int{
|
|
"namespace1": len(TailscaleVersions),
|
|
"namespace2": len(TailscaleVersions),
|
|
}
|
|
|
|
err = scenario.CreateHeadscaleEnv(spec)
|
|
if err != nil {
|
|
t.Errorf("failed to create headscale environment: %s", err)
|
|
}
|
|
|
|
var allIps []netip.Addr
|
|
var allClients []*tsic.TailscaleInContainer
|
|
|
|
for namespace, count := range spec {
|
|
ips, err := scenario.GetIPs(namespace)
|
|
if err != nil {
|
|
t.Errorf("failed to get tailscale ips: %s", err)
|
|
}
|
|
|
|
if len(ips) != count*2 {
|
|
t.Errorf(
|
|
"got the wrong amount of tailscale ips, %d != %d",
|
|
len(ips),
|
|
count*2,
|
|
)
|
|
}
|
|
|
|
clients, err := scenario.GetClients(namespace)
|
|
if err != nil {
|
|
t.Errorf("failed to get tailscale clients: %s", err)
|
|
}
|
|
|
|
allIps = append(allIps, ips...)
|
|
allClients = append(allClients, clients...)
|
|
}
|
|
|
|
err = scenario.WaitForTailscaleSync()
|
|
if err != nil {
|
|
t.Errorf("failed wait for tailscale clients to be in sync: %s", err)
|
|
}
|
|
|
|
success := 0
|
|
|
|
for _, client := range allClients {
|
|
for _, ip := range allIps {
|
|
err := client.Ping(ip)
|
|
if err != nil {
|
|
t.Errorf("failed to ping %s from %s: %s", ip, client.Hostname, err)
|
|
} else {
|
|
success++
|
|
}
|
|
}
|
|
}
|
|
|
|
t.Logf("%d successful pings out of %d", success, len(allClients)*len(allIps))
|
|
|
|
err = scenario.Shutdown()
|
|
if err != nil {
|
|
t.Errorf("failed to tear down scenario: %s", err)
|
|
}
|
|
}
|