mirror of
https://github.com/juanfont/headscale.git
synced 2025-08-14 13:51:01 +02:00
fix
This commit is contained in:
parent
9b47f71f37
commit
baa4fb9615
@ -91,49 +91,30 @@ func TestAuthKeyLogoutAndReloginSameUser(t *testing.T) {
|
|||||||
assertLastSeenSet(t, node)
|
assertLastSeenSet(t, node)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
userMap, err := headscale.MapUsers()
|
||||||
|
assertNoErr(t, err)
|
||||||
|
|
||||||
// if the server is not running with HTTPS, we have to wait a bit before
|
// if the server is not running with HTTPS, we have to wait a bit before
|
||||||
// reconnection as the newest Tailscale client has a measure that will only
|
// reconnection as the newest Tailscale client has a measure that will only
|
||||||
// reconnect over HTTPS if they saw a noise connection previously.
|
// reconnect over HTTPS if they saw a noise connection previously.
|
||||||
// https://github.com/tailscale/tailscale/commit/1eaad7d3deb0815e8932e913ca1a862afa34db38
|
// https://github.com/tailscale/tailscale/commit/1eaad7d3deb0815e8932e913ca1a862afa34db38
|
||||||
// https://github.com/juanfont/headscale/issues/2164
|
// https://github.com/juanfont/headscale/issues/2164
|
||||||
if !https {
|
if !https {
|
||||||
userMap, err := headscale.MapUsers()
|
|
||||||
assertNoErr(t, err)
|
|
||||||
|
|
||||||
// Create auth keys once outside the retry loop
|
|
||||||
userKeys := make(map[string]string)
|
|
||||||
for _, userName := range spec.Users {
|
|
||||||
key, err := scenario.CreatePreAuthKey(userMap[userName].GetId(), true, false)
|
|
||||||
assertNoErr(t, err)
|
|
||||||
userKeys[userName] = key.GetKey()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Wait for the 2-minute noise dial memory to expire
|
// Wait for the 2-minute noise dial memory to expire
|
||||||
// The Tailscale commit shows clients remember noise dials for 2 minutes
|
// The Tailscale commit shows clients remember noise dials for 2 minutes
|
||||||
t.Logf("Waiting 2.5 minutes for Tailscale noise dial memory to expire...")
|
t.Logf("Waiting 2.5 minutes for Tailscale noise dial memory to expire...")
|
||||||
time.Sleep(2*time.Minute + 30*time.Second)
|
time.Sleep(2*time.Minute + 30*time.Second)
|
||||||
|
}
|
||||||
|
|
||||||
// Wait for clients to be ready to reconnect over HTTP after HTTPS
|
for _, userName := range spec.Users {
|
||||||
assert.EventuallyWithT(t, func(ct *assert.CollectT) {
|
key, err := scenario.CreatePreAuthKey(userMap[userName].GetId(), true, false)
|
||||||
for _, userName := range spec.Users {
|
if err != nil {
|
||||||
err = scenario.RunTailscaleUp(userName, headscale.GetEndpoint(), userKeys[userName])
|
t.Fatalf("failed to create pre-auth key for user %s: %s", userName, err)
|
||||||
assert.NoError(ct, err, "Client should be able to reconnect over HTTP")
|
}
|
||||||
}
|
|
||||||
}, 6*time.Minute, 30*time.Second)
|
|
||||||
} else {
|
|
||||||
userMap, err := headscale.MapUsers()
|
|
||||||
assertNoErr(t, err)
|
|
||||||
|
|
||||||
for _, userName := range spec.Users {
|
err = scenario.RunTailscaleUp(userName, headscale.GetEndpoint(), key.GetKey())
|
||||||
key, err := scenario.CreatePreAuthKey(userMap[userName].GetId(), true, false)
|
if err != nil {
|
||||||
if err != nil {
|
t.Fatalf("failed to run tailscale up for user %s: %s", userName, err)
|
||||||
t.Fatalf("failed to create pre-auth key for user %s: %s", userName, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = scenario.RunTailscaleUp(userName, headscale.GetEndpoint(), key.GetKey())
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("failed to run tailscale up for user %s: %s", userName, err)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user