Function TestMarkTemporary

aibridge/keypool/keypool_test.go:138–217

(t *testing.T)

}

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

	tests := []struct {
		name               string
		cooldown           time.Duration
		setup              func(t *testing.T, pool *keypool.Pool, clk *quartz.Mock) *keypool.Key
		expectedState      keypool.KeyState
		expectedTransition bool
	}{
		{
			// valid -> temporary: key becomes unavailable.
			name:     "valid_to_temporary",
			cooldown: 60 * time.Second,
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				return key
			},
			expectedState:      keypool.KeyStateTemporary,
			expectedTransition: true,
		},
		{
			// temporary -> temporary: new cooldown is longer,
			// so the deadline is extended.
			name:     "temporary_to_temporary_extends_cooldown",
			cooldown: 60 * time.Second,
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkTemporary(10 * time.Second)
				return key
			},
			expectedState:      keypool.KeyStateTemporary,
			expectedTransition: false,
		},
		{
			// temporary -> temporary: new cooldown is shorter,
			// so the existing longer deadline is preserved.
			name:     "temporary_to_temporary_keeps_longer_cooldown",
			cooldown: 10 * time.Second,
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkTemporary(60 * time.Second)
				return key
			},
			expectedState:      keypool.KeyStateTemporary,
			expectedTransition: false,
		},
		{
			// permanent -> permanent: no-op, permanent is irreversible.
			name:     "permanent_to_temporary_is_no_op",
			cooldown: 60 * time.Second,
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkPermanent()