Function TestWalkerIndependence

aibridge/keypool/keypool_test.go:612–647

TestWalkerIndependence simulates two requests using the same pool. The first request marks key-0 temporary and key-1 permanent, then gets key-2. The second request sees the updated pool state and also gets key-2.

(t *testing.T)

// permanent, then gets key-2. The second request sees the
// updated pool state and also gets key-2.
func TestWalkerIndependence(t *testing.T) {
	t.Parallel()

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

	walker := pool.Walker()

	// First attempt: get key-0.
	key, keyPoolErr := walker.Next()
	require.Nil(t, keyPoolErr)
	assert.Equal(t, "key-0", key.Value())

	// Simulate 429: mark key-0 temporary.
	key.MarkTemporary(60 * time.Second)

	// Second attempt: walker advances to key-1.
	key, keyPoolErr = walker.Next()
	require.Nil(t, keyPoolErr)
	assert.Equal(t, "key-1", key.Value())

	// Simulate 401: mark key-1 permanent.
	key.MarkPermanent()

	// Third attempt: walker advances to key-2.
	key, keyPoolErr = walker.Next()
	require.Nil(t, keyPoolErr)
	assert.Equal(t, "key-2", key.Value())

	// A new walker should skip key-0 (temporary) and key-1
	// (permanent), and return key-2.
	key2, keyPoolErr := pool.Walker().Next()
	require.Nil(t, keyPoolErr)
	assert.Equal(t, "key-2", key2.Value())
}