Skip to content

Commit

Permalink
internal/daemon/worker: improve runtime and reduce noise
Browse files Browse the repository at this point in the history
TestRotationTicking would previously wait for 5 rotations, which is
pretty slow considering each rotation takes 20 seconds. Reduce
the number to 3, and use require instead of assert to remove
noise during flaky test runs.

This doesn't fix the flakiness of this test, but it should reduce the log spam.
  • Loading branch information
johanbrandhorst committed Dec 20, 2024
1 parent 6ed077c commit b949b33
Showing 1 changed file with 14 additions and 5 deletions.
19 changes: 14 additions & 5 deletions internal/daemon/worker/auth_rotation_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,8 @@ import (
// TestRotationTicking ensures that we see new credential information for a
// worker on both the controller side and worker side in a periodic fashion
func TestRotationTicking(t *testing.T) {
t.Parallel()

require, assert := require.New(t), assert.New(t)
logger := hclog.New(&hclog.LoggerOptions{
Level: hclog.Trace,
Expand Down Expand Up @@ -95,7 +97,7 @@ func TestRotationTicking(t *testing.T) {
// Verify we see authorized credentials now
auths, err = workerAuthRepo.List(c.Context(), (*types.NodeInformation)(nil))
require.NoError(err)
assert.Len(auths, 2)
require.Len(auths, 2)
// Fetch creds and store current key
currNodeCreds, err := types.LoadNodeCredentials(
w.Context(),
Expand All @@ -112,21 +114,28 @@ func TestRotationTicking(t *testing.T) {
// Make sure we have a failsafe in case the below loop never finds what it's
// looking for; at expiration the List will fail and we'll hit the Require,
// exiting
deadlineCtx, deadlineCtxCancel := context.WithTimeout(c.Context(), 10*time.Minute)
testTimeout := 10 * time.Minute
testDeadline, ok := t.Deadline()
if ok && time.Until(testDeadline) < testTimeout {
// If the test deadline is less than the default timeout, use the deadline
testTimeout = time.Until(testDeadline) - time.Second
}
deadlineCtx, deadlineCtxCancel := context.WithTimeout(c.Context(), testTimeout)
defer deadlineCtxCancel()

rotationCount := 2
for {
if rotationCount > 5 {
if rotationCount > 3 {
break
}
nextRotation := w.Worker().AuthRotationNextRotation.Load()
time.Sleep((*nextRotation).Sub(time.Now()) + 5*time.Second)
// Sleep until 5 seconds after next rotation
time.Sleep(time.Until(*nextRotation) + 5*time.Second)

// Verify we see 2- after credentials have rotated, we should see current and previous
auths, err = workerAuthRepo.List(deadlineCtx, (*types.NodeInformation)(nil))
require.NoError(err)
assert.Len(auths, 2)
require.Len(auths, 2)

// Fetch creds and compare current key
currNodeCreds, err := types.LoadNodeCredentials(
Expand Down

0 comments on commit b949b33

Please sign in to comment.