Function TestNoCompactFlag

tempodb/tempodb_test.go:850–953

(t *testing.T)

}

func TestNoCompactFlag(t *testing.T) {
	for _, tc := range []struct {
		name                    string
		createWithNoCompactFlag bool
		skipNoCompactBlocks     bool
		expectedNoCompactFlag   bool // flag after block completion
		isBlockExpected         bool // should we expect the block after pull
	}{
		{
			name:                    "default behaviour",
			createWithNoCompactFlag: false,
			skipNoCompactBlocks:     false,
			expectedNoCompactFlag:   false,
			isBlockExpected:         true,
		},
		{
			name:                    "blockbuilder behaviour",
			createWithNoCompactFlag: true,  // blockbuilder creates blocks with the flag
			skipNoCompactBlocks:     false, // but blockbuilder does not skip such block on pull
			expectedNoCompactFlag:   true,  // the flag should not be removed after block completion
			isBlockExpected:         true,  // blockbuilder sees the block although it has the flag
		},
		{
			name:                    "compactor sees its own blocks",
			createWithNoCompactFlag: false, // compactor does not create blocks with the flag
			skipNoCompactBlocks:     true,  // but does skip block with the flag on pull
			expectedNoCompactFlag:   false, // no flag on created block
			isBlockExpected:         true,  // compactor sees its own blocks
		},
		{
			name:                    "blockbuilder+compactor behaviour",
			createWithNoCompactFlag: true,  // blockbuilder creates the block with the flag
			skipNoCompactBlocks:     true,  // compactor skips the block with the flag on pull
			expectedNoCompactFlag:   true,  // the flag should not be removed after block completion
			isBlockExpected:         false, // compactor does not see the block
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			r, w, c, _ := testConfig(t, 0, func(c *Config) {
				c.Block.CreateWithNoCompactFlag = tc.createWithNoCompactFlag
			})

			ctx, cancel := context.WithCancel(context.Background())
			defer cancel()

			err := c.EnableCompaction(ctx, &CompactorConfig{
				MaxCompactionRange:      time.Hour,
				BlockRetention:          0,
				CompactedBlockRetention: 0,
			}, &mockSharder{}, &mockOverrides{})
			require.NoError(t, err)

			r.EnablePolling(ctx, &mockJobSharder{}, tc.skipNoCompactBlocks)

			// Create a test block
			blockID := backend.NewUUID()
			wal := w.WAL()
			meta := &backend.BlockMeta{BlockID: blockID, TenantID: testTenantID}