Function TestWalkerNext

aibridge/keypool/keypool_test.go:282–527

(t *testing.T)

}

func TestWalkerNext(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name          string
		keys          []string
		setup         func(t *testing.T, pool *keypool.Pool)
		advance       time.Duration
		expectedValid []string
		expectedErr   *keypool.Error
	}{
		{
			// Given: key-0: valid, key-1: valid, key-2: valid.
			// Then: key-0: valid, key-1: valid, key-2: valid.
			name:          "all_keys_valid",
			keys:          []string{"key-0", "key-1", "key-2"},
			setup:         func(_ *testing.T, _ *keypool.Pool) {},
			expectedValid: []string{"key-0", "key-1", "key-2"},
			expectedErr:   &keypool.Error{Kind: keypool.ErrorKindRateLimited},
		},
		{
			// Given: key-0: temporary, key-1: valid, key-2: valid.
			// Then: key-0: temporary, key-1: valid, key-2: valid.
			name: "skips_temporary_keys",
			keys: []string{"key-0", "key-1", "key-2"},
			setup: func(t *testing.T, pool *keypool.Pool) {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkTemporary(60 * time.Second)
			},
			expectedValid: []string{"key-1", "key-2"},
			expectedErr:   &keypool.Error{Kind: keypool.ErrorKindRateLimited},
		},
		{
			// Given: key-0: permanent, key-1: permanent, key-2: valid.
			// Then: key-0: permanent, key-1: permanent, key-2: valid.
			name: "skips_permanent_keys",
			keys: []string{"key-0", "key-1", "key-2"},
			setup: func(t *testing.T, pool *keypool.Pool) {
				walker := pool.Walker()
				key0, keyPoolErr := walker.Next()
				require.Nil(t, keyPoolErr)
				key0.MarkPermanent()
				key1, keyPoolErr := walker.Next()
				require.Nil(t, keyPoolErr)
				key1.MarkPermanent()
			},
			expectedValid: []string{"key-2"},
			expectedErr:   &keypool.Error{Kind: keypool.ErrorKindRateLimited},
		},
		{
			// Given: key-0: temporary (30s), key-1: valid.
			// When: 35s pass.
			// Then: key-0: valid, key-1: valid.
			name: "expired_temporary_is_available",
			keys: []string{"key-0", "key-1"},
			setup: func(t *testing.T, pool *keypool.Pool) {
				key, keyPoolErr := pool.Walker().Next()