Function Test_WatchKey_UpdateStale

pkg/ring/kv/dynamodb/client_test.go:239–264

(t *testing.T)

}

func Test_WatchKey_UpdateStale(t *testing.T) {
	ddbMock := NewDynamodbClientMock()
	codecMock := &CodecMock{}
	c := NewClientMock(ddbMock, codecMock, TestLogger{}, prometheus.NewPedanticRegistry(), defaultPullTime, defaultBackoff)
	staleData := &DescMock{}

	ddbMock.On("Query").Return(map[string]dynamodbItem{}, nil).Once()
	codecMock.On("DecodeMultiKey").Return(staleData, nil)

	c.WatchKey(context.TODO(), key, func(i any) bool {
		ddbMock.AssertNumberOfCalls(t, "Query", 1)
		codecMock.AssertNumberOfCalls(t, "DecodeMultiKey", 1)
		require.EqualValues(t, staleData, i)
		return false
	})

	ddbMock.On("Query").Return(map[string]dynamodbItem{}, errors.Errorf("failed"))
	staleData.On("Clone").Return(staleData).Once()

	c.WatchKey(context.TODO(), key, func(i any) bool {
		ddbMock.AssertNumberOfCalls(t, "Query", 12)
		codecMock.AssertNumberOfCalls(t, "DecodeMultiKey", 1)
		require.EqualValues(t, staleData, i)
		return false
	})
}

func Test_WatchKey_AlwaysRetry(t *testing.T) {
	casBackoffConfig := backoff.Config{