Function TestKeyConcurrent

aibridge/keypool/keypool_test.go:532–606

TestKeyConcurrent exercises the documented concurrent-safety contract by hammering a single key with concurrent Mark calls and asserting the resulting state honors the pool's invariants.

(t *testing.T)

// contract by hammering a single key with concurrent Mark calls
// and asserting the resulting state honors the pool's invariants.
func TestKeyConcurrent(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name string
		// run is called concurrently from numGoroutines, each
		// with its own index.
		run func(idx int, key *keypool.Key)
		// verify asserts the final state. May advance the clock.
		verify func(t *testing.T, key *keypool.Key, clk *quartz.Mock)
	}{
		{
			// Half of the goroutines mark the key as temporary
			// with 60s, the other half with 10s. The longer
			// cooldown must win regardless of ordering.
			name: "longer_cooldown_wins",
			run: func(idx int, key *keypool.Key) {
				if idx%2 == 0 {
					key.MarkTemporary(60 * time.Second)
				} else {
					key.MarkTemporary(10 * time.Second)
				}
			},
			verify: func(t *testing.T, key *keypool.Key, clk *quartz.Mock) {
				// At 50s the 60s cooldown is still active.
				clk.Advance(50 * time.Second)
				assert.Equal(t, keypool.KeyStateTemporary, key.State())
				// At 65s the 60s cooldown has expired.
				clk.Advance(15 * time.Second)
				assert.Equal(t, keypool.KeyStateValid, key.State())
			},
		},
		{
			// Half of the goroutines mark the key as permanent,
			// the other half mark it as temporary. Permanent is
			// terminal: any permanent call wins.
			name: "permanent_wins_over_temporary",
			run: func(idx int, key *keypool.Key) {
				if idx%2 == 0 {
					key.MarkPermanent()
				} else {
					key.MarkTemporary(60 * time.Second)
				}
			},
			verify: func(t *testing.T, key *keypool.Key, _ *quartz.Mock) {
				assert.Equal(t, keypool.KeyStatePermanent, key.State())
			},
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			clk := quartz.NewMock(t)
			pool, err := keypool.New([]string{"key-0"}, clk)
			require.NoError(t, err)
			key, keyPoolErr := pool.Walker().Next()