Function TestMarkPermanent

aibridge/keypool/keypool_test.go:219–280

(t *testing.T)

}

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

	tests := []struct {
		name               string
		setup              func(t *testing.T, pool *keypool.Pool) *keypool.Key
		expectedState      keypool.KeyState
		expectedTransition bool
	}{
		{
			// valid -> permanent: key becomes permanently unavailable.
			name: "valid_to_permanent",
			setup: func(t *testing.T, pool *keypool.Pool) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				return key
			},
			expectedState:      keypool.KeyStatePermanent,
			expectedTransition: true,
		},
		{
			// temporary -> permanent: escalation from rate limit
			// to auth failure.
			name: "temporary_to_permanent",
			setup: func(t *testing.T, pool *keypool.Pool) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkTemporary(60 * time.Second)
				return key
			},
			expectedState:      keypool.KeyStatePermanent,
			expectedTransition: true,
		},
		{
			// permanent -> permanent: no-op, already permanent.
			name: "permanent_to_permanent",
			setup: func(t *testing.T, pool *keypool.Pool) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkPermanent()
				return key
			},
			expectedState:      keypool.KeyStatePermanent,
			expectedTransition: false,
		},
	}

	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", "key-1"}, clk)
			require.NoError(t, err)

			key := tc.setup(t, pool)
			transition := key.MarkPermanent()

			assert.Equal(t, tc.expectedState, key.State())