Function TestDropTraceCmd

cmd/tempo-cli/cmd-rewrite-blocks_test.go:27–62

(t *testing.T)

)

func TestDropTraceCmd(t *testing.T) {
	testCase := func(t *testing.T, blocksNum int, tracesNum int, deleteEvery int) {
		cmd := dropTracesCmd{
			backendOptions: backendOptions{
				Backend: "local",
				Bucket:  t.TempDir(),
			},
			TenantID:  "single-tenant",
			DropTrace: true,
		}
		generateTestBlocks(t, cmd.Bucket, cmd.TenantID, blocksNum, tracesNum)

		before := getAllTraceIDs(t, cmd.Bucket, cmd.TenantID)

		var expectedAfter, toRemove []string
		for i, traceID := range before {
			if i%deleteEvery == 0 {
				toRemove = append(toRemove, traceID)
			} else {
				expectedAfter = append(expectedAfter, traceID)
			}
		}
		cmd.TraceIDs = strings.Join(toRemove, ",")

		err := cmd.Run(&globalOptions{})
		require.NoError(t, err)

		after := getAllTraceIDs(t, cmd.Bucket, cmd.TenantID)

		require.ElementsMatch(t, after, expectedAfter)
	}

	testCase(t, 1, 10, 3)
	testCase(t, 2, 5, 3)
	testCase(t, 2, 5, 1)
}

func generateTestBlocks(t *testing.T, tempDir string, tenantID string, blockCount int, traceCount int) {
	t.Helper()