Function Test_CAS_Backoff

pkg/ring/kv/dynamodb/client_test.go:48–103

(t *testing.T)

}

func Test_CAS_Backoff(t *testing.T) {
	testCases := []struct {
		name               string
		setupMocks         func(*MockDynamodbClient, *CodecMock, *DescMock, map[dynamodbKey]dynamodbItem, []dynamodbKey)
		expectedQueryCalls int
		expectedBatchCalls int
	}{
		{
			name: "query_fails_and_backs_off",
			setupMocks: func(ddbMock *MockDynamodbClient, codecMock *CodecMock, descMock *DescMock, expectedBatch map[dynamodbKey]dynamodbItem, expectedDelete []dynamodbKey) {
				ddbMock.On("Query").Return(map[string]dynamodbItem{}, errors.Errorf("query failed")).Once()
				ddbMock.On("Query").Return(map[string]dynamodbItem{}, nil).Once()
				ddbMock.On("Batch", context.TODO(), expectedBatch, expectedDelete).Return(false, nil).Once()
			},
			expectedQueryCalls: 2,
			expectedBatchCalls: 1,
		},
		{
			name: "batch_fails_and_backs_off",
			setupMocks: func(ddbMock *MockDynamodbClient, codecMock *CodecMock, descMock *DescMock, expectedBatch map[dynamodbKey]dynamodbItem, expectedDelete []dynamodbKey) {
				ddbMock.On("Query").Return(map[string]dynamodbItem{}, nil).Twice()
				ddbMock.On("Batch", context.TODO(), expectedBatch, expectedDelete).Return(true, errors.Errorf("batch failed")).Once()
				ddbMock.On("Batch", context.TODO(), expectedBatch, expectedDelete).Return(false, nil).Once()
			},
			expectedQueryCalls: 2,
			expectedBatchCalls: 2,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			ddbMock := NewDynamodbClientMock()
			codecMock := &CodecMock{}
			descMock := &DescMock{}
			c := NewClientMock(ddbMock, codecMock, TestLogger{}, prometheus.NewPedanticRegistry(), defaultPullTime, defaultBackoff)

			expectedBatch := map[dynamodbKey]dynamodbItem{}
			expectedDelete := []dynamodbKey{{primaryKey: "test", sortKey: "childkey"}}

			tc.setupMocks(ddbMock, codecMock, descMock, expectedBatch, expectedDelete)

			codecMock.On("DecodeMultiKey").Return(descMock, nil).Times(tc.expectedQueryCalls)
			descMock.On("Clone").Return(descMock).Times(tc.expectedQueryCalls)
			descMock.On("FindDifference", descMock).Return(descMock, []string{"childkey"}, nil).Times(tc.expectedBatchCalls)
			codecMock.On("EncodeMultiKey").Return(map[string][]byte{}, nil).Times(tc.expectedBatchCalls)

			err := c.CAS(context.TODO(), key, func(in any) (out any, retry bool, err error) {
				return descMock, true, nil
			})

			require.NoError(t, err)
			ddbMock.AssertNumberOfCalls(t, "Query", tc.expectedQueryCalls)
			ddbMock.AssertNumberOfCalls(t, "Batch", tc.expectedBatchCalls)
		})
	}
}

func Test_CAS_Failed(t *testing.T) {