Function TestNoCompactFlag

tempodb/backend/raw_test.go:299–336

(t *testing.T)

}

func TestNoCompactFlag(t *testing.T) {
	ctx := context.Background()
	tenantID := "test-tenant"
	blockID := uuid.New()

	rawReader := &MockRawReader{}
	rawWriter := &MockRawWriter{}
	rawReader.ReadFn = func(_ context.Context, name string, keyPath KeyPath, _ *CacheInfo) (io.ReadCloser, int64, error) {
		key := strings.Join(keyPath, "/") + "/" + name
		val, ok := rawWriter.writeBuffer[key]
		if !ok {
			return nil, 0, ErrDoesNotExist
		}
		return io.NopCloser(bytes.NewReader(val)), int64(len(val)), nil
	}

	reader := NewReader(rawReader)
	writer := NewWriter(rawWriter)

	hasFlag, err := reader.HasNoCompactFlag(ctx, blockID, tenantID)
	require.NoError(t, err)
	assert.False(t, hasFlag)

	err = writer.WriteNoCompactFlag(ctx, blockID, tenantID)
	require.NoError(t, err)

	hasFlag, err = reader.HasNoCompactFlag(ctx, blockID, tenantID)
	require.NoError(t, err)
	assert.True(t, hasFlag)

	err = writer.DeleteNoCompactFlag(ctx, blockID, tenantID)
	require.NoError(t, err)
	assert.Equal(t, map[string]map[string]int{
		NoCompactFileName: {
			fmt.Sprintf("%s/%s", tenantID, blockID.String()): 1,
		},
	}, rawWriter.deleteCalls)
}