Function TestMemcachedClient_GetMultiWithError

cache/memcached_client_test.go:160–230

(t *testing.T)

}

func TestMemcachedClient_GetMultiWithError(t *testing.T) {
	t.Run("returns error from backend", func(t *testing.T) {
		backendErr := fmt.Errorf("connection failed")
		backend := &erroringMockMemcachedClientBackend{
			mockMemcachedClientBackend: newMockMemcachedClientBackend(),
			err:                        backendErr,
		}
		client, err := newMemcachedClient(
			log.NewNopLogger(),
			backend,
			&mockServerSelector{
				servers: []mockServer{
					{addr: "127.0.0.1:11211"},
				},
			},
			MemcachedClientConfig{
				Addresses:           []string{"localhost"},
				MaxAsyncConcurrency: 1,
				MaxAsyncBufferSize:  10,
			},
			prometheus.NewPedanticRegistry(),
			"test",
		)
		require.NoError(t, err)

		ctx := context.Background()
		res, err := client.GetMultiWithError(ctx, []string{"foo"})
		require.ErrorIs(t, err, backendErr)
		require.Empty(t, res)
	})

	t.Run("returns partial results with error", func(t *testing.T) {
		backendErr := fmt.Errorf("connection failed")
		backend := &selectiveErrorMockMemcachedClientBackend{
			mockMemcachedClientBackend: newMockMemcachedClientBackend(),
			failOnCalls:                map[int]error{2: backendErr}, // Fail on 2nd batch
		}

		client, err := newMemcachedClient(
			log.NewNopLogger(),
			backend,
			&mockServerSelector{
				servers: []mockServer{
					{addr: "127.0.0.1:11211"},
				},
			},
			MemcachedClientConfig{
				Addresses:            []string{"localhost"},
				MaxAsyncConcurrency:  1,
				MaxAsyncBufferSize:   10,
				MaxGetMultiBatchSize: 1, // Force each key into its own batch
			},
			prometheus.NewPedanticRegistry(),
			"test",
		)
		require.NoError(t, err)

		// Set up 3 keys