Function TestNewKeyFailoverTransport

aibridge/keypool/failover_test.go:28–69

(t *testing.T)

}

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

	pool, err := keypool.New([]string{"k0"}, quartz.NewMock(t))
	require.NoError(t, err)

	tests := []struct {
		name string
		// Constructor input.
		config keypool.KeyFailoverConfig
		// Whether the constructor returns inner unchanged.
		expectSame bool
	}{
		{
			// Pool is nil: failover is disabled, inner is returned unchanged.
			name:       "pool_nil_returns_inner",
			config:     keypool.KeyFailoverConfig{},
			expectSame: true,
		},
		{
			// Pool is set: inner is wrapped in a key-failover transport.
			name:       "pool_set_returns_wrapper",
			config:     keypool.KeyFailoverConfig{Pool: pool},
			expectSame: false,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			inner := &fakeRoundTripper{}
			got := keypool.NewKeyFailoverTransport(inner, tc.config)

			if tc.expectSame {
				assert.Same(t, inner, got)
			} else {
				assert.NotSame(t, inner, got)
			}
		})
	}
}