Function TestState

aibridge/keypool/keypool_test.go:58–136

(t *testing.T)

}

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

	tests := []struct {
		name          string
		setup         func(t *testing.T, pool *keypool.Pool, clk *quartz.Mock) *keypool.Key
		expectedState keypool.KeyState
	}{
		{
			// Fresh key is valid.
			name: "fresh_key_is_valid",
			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.KeyStateValid,
		},
		{
			// Active cooldown makes the key temporary.
			name: "active_cooldown_is_temporary",
			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,
		},
		{
			// Expired cooldown returns the key to valid.
			name: "expired_cooldown_is_valid",
			setup: func(t *testing.T, pool *keypool.Pool, clk *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkTemporary(30 * time.Second)
				clk.Advance(35 * time.Second)
				return key
			},
			expectedState: keypool.KeyStateValid,
		},
		{
			// Permanent key is permanent.
			name: "permanent_key",
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)
				key.MarkPermanent()
				return key
			},
			expectedState: keypool.KeyStatePermanent,
		},
		{
			// Permanent takes precedence over active cooldown.
			name: "permanent_with_cooldown_is_permanent",
			setup: func(t *testing.T, pool *keypool.Pool, _ *quartz.Mock) *keypool.Key {
				key, keyPoolErr := pool.Walker().Next()
				require.Nil(t, keyPoolErr)