mirror of
https://github.com/juanfont/headscale.git
synced 2026-04-18 06:50:16 +02:00
f87b08676d
Extend the servertest harness with: - TestClient.Direct() accessor for advanced operations - TestClient.WaitForPeerCount and WaitForCondition helpers - TestHarness.ChangePolicy for ACL policy testing - AssertDERPMapPresent and AssertSelfHasAddresses New test suites: - content_test.go: self node, DERP map, peer properties, user profiles, update history monotonicity, and endpoint update propagation - policy_test.go: default allow-all, explicit policy, policy triggers updates on all nodes, multiple policy changes, multi-user mesh - ephemeral_test.go: ephemeral connect, cleanup after disconnect, mixed ephemeral/regular, reconnect prevents cleanup - routes_test.go: addresses in AllowedIPs, route advertise and approve, advertised routes via hostinfo, CGNAT range validation Also fix node_departs test to use WaitForCondition instead of assert.Eventually, and convert concurrent_join_and_leave to interleaved_join_and_leave with grace-period-tolerant assertions.
133 lines
4.0 KiB
Go
133 lines
4.0 KiB
Go
package servertest_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/juanfont/headscale/hscontrol/servertest"
|
|
"github.com/stretchr/testify/assert"
|
|
"tailscale.com/types/netmap"
|
|
)
|
|
|
|
// TestEphemeralNodes tests the lifecycle of ephemeral nodes,
|
|
// which should be automatically cleaned up when they disconnect.
|
|
func TestEphemeralNodes(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("ephemeral_connects_and_sees_peers", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
srv := servertest.NewServer(t,
|
|
servertest.WithEphemeralTimeout(5*time.Second))
|
|
user := srv.CreateUser(t, "eph-user")
|
|
|
|
regular := servertest.NewClient(t, srv, "eph-regular",
|
|
servertest.WithUser(user))
|
|
ephemeral := servertest.NewClient(t, srv, "eph-ephemeral",
|
|
servertest.WithUser(user), servertest.WithEphemeral())
|
|
|
|
// Both should see each other.
|
|
regular.WaitForPeers(t, 1, 10*time.Second)
|
|
ephemeral.WaitForPeers(t, 1, 10*time.Second)
|
|
|
|
_, found := regular.PeerByName("eph-ephemeral")
|
|
assert.True(t, found, "regular should see ephemeral peer")
|
|
|
|
_, found = ephemeral.PeerByName("eph-regular")
|
|
assert.True(t, found, "ephemeral should see regular peer")
|
|
})
|
|
|
|
t.Run("ephemeral_cleanup_after_disconnect", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// Use a short ephemeral timeout so the test doesn't take long.
|
|
srv := servertest.NewServer(t,
|
|
servertest.WithEphemeralTimeout(3*time.Second))
|
|
user := srv.CreateUser(t, "eph-cleanup-user")
|
|
|
|
regular := servertest.NewClient(t, srv, "eph-cleanup-regular",
|
|
servertest.WithUser(user))
|
|
ephemeral := servertest.NewClient(t, srv, "eph-cleanup-ephemeral",
|
|
servertest.WithUser(user), servertest.WithEphemeral())
|
|
|
|
regular.WaitForPeers(t, 1, 10*time.Second)
|
|
|
|
// Disconnect the ephemeral node.
|
|
ephemeral.Disconnect(t)
|
|
|
|
// After the grace period (10s) + ephemeral timeout (3s) +
|
|
// some propagation time, the regular node should no longer
|
|
// see the ephemeral node. This tests the full cleanup path.
|
|
regular.WaitForCondition(t, "ephemeral peer gone or offline",
|
|
60*time.Second,
|
|
func(nm *netmap.NetworkMap) bool {
|
|
for _, p := range nm.Peers {
|
|
hi := p.Hostinfo()
|
|
if hi.Valid() && hi.Hostname() == "eph-cleanup-ephemeral" {
|
|
// Still present -- check if offline.
|
|
isOnline, known := p.Online().GetOk()
|
|
if known && !isOnline {
|
|
return true // offline is acceptable
|
|
}
|
|
|
|
return false // still online
|
|
}
|
|
}
|
|
|
|
return true // gone
|
|
})
|
|
})
|
|
|
|
t.Run("ephemeral_and_regular_mixed", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
srv := servertest.NewServer(t,
|
|
servertest.WithEphemeralTimeout(5*time.Second))
|
|
user := srv.CreateUser(t, "mix-user")
|
|
|
|
r1 := servertest.NewClient(t, srv, "mix-regular-1",
|
|
servertest.WithUser(user))
|
|
r2 := servertest.NewClient(t, srv, "mix-regular-2",
|
|
servertest.WithUser(user))
|
|
e1 := servertest.NewClient(t, srv, "mix-eph-1",
|
|
servertest.WithUser(user), servertest.WithEphemeral())
|
|
|
|
// All three should see each other.
|
|
r1.WaitForPeers(t, 2, 15*time.Second)
|
|
r2.WaitForPeers(t, 2, 15*time.Second)
|
|
e1.WaitForPeers(t, 2, 15*time.Second)
|
|
|
|
servertest.AssertMeshComplete(t,
|
|
[]*servertest.TestClient{r1, r2, e1})
|
|
})
|
|
|
|
t.Run("ephemeral_reconnect_prevents_cleanup", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
srv := servertest.NewServer(t,
|
|
servertest.WithEphemeralTimeout(5*time.Second))
|
|
user := srv.CreateUser(t, "eph-recon-user")
|
|
|
|
regular := servertest.NewClient(t, srv, "eph-recon-regular",
|
|
servertest.WithUser(user))
|
|
ephemeral := servertest.NewClient(t, srv, "eph-recon-ephemeral",
|
|
servertest.WithUser(user), servertest.WithEphemeral())
|
|
|
|
regular.WaitForPeers(t, 1, 10*time.Second)
|
|
|
|
// Ensure the ephemeral node's long-poll is established.
|
|
ephemeral.WaitForPeers(t, 1, 10*time.Second)
|
|
|
|
// Disconnect and quickly reconnect.
|
|
ephemeral.Disconnect(t)
|
|
ephemeral.Reconnect(t)
|
|
|
|
// After reconnecting, the ephemeral node should still be visible.
|
|
regular.WaitForPeers(t, 1, 15*time.Second)
|
|
|
|
_, found := regular.PeerByName("eph-recon-ephemeral")
|
|
assert.True(t, found,
|
|
"ephemeral node should still be visible after quick reconnect")
|
|
})
|
|
}
|