Test for new strategy

pkg/bloombuild/planner/strategies/chunksize_test.go

@@ -0,0 +1,51 @@
+package strategies
+
+import (
+	"context"
+	"github.com/grafana/loki/v3/pkg/bloombuild/planner/test_utils"
+	"testing"
+
+	"github.com/go-kit/log"
+	"github.com/stretchr/testify/require"
+
+	"github.com/grafana/loki/v3/pkg/bloombuild/protos"
+	"github.com/grafana/loki/v3/pkg/storage/stores/shipper/bloomshipper"
+)
+
+func Test_ChunkSizeStrategy_Plan(t *testing.T) {
+	for _, tc := range []struct {
+		name          string
+		originalMetas []bloomshipper.Meta
+		tsdbs         TSDBSet
+		expectedTasks []*protos.Task
+	}{
+		{
+			name: "no previous blocks and metas",
+			// TODO: Build TSDB with mock ForEach with equally sized series
+		},
+		{
+			name: "Test case 2",
+		},
+	} {
+		t.Run(tc.name, func(t *testing.T) {
+			logger := log.NewNopLogger()
+			//logger := log.NewLogfmtLogger(os.Stdout)
+
+			strategy, err := NewChunkSizeStrategy(fakeChunkSizeLimits{TargetSize: 1000}, logger)
+			require.NoError(t, err)
+
+			actual, err := strategy.Plan(context.Background(), test_utils.TestTable, "fake", tc.tsdbs, tc.originalMetas)
+			require.NoError(t, err)
+
+			require.ElementsMatch(t, tc.expectedTasks, actual)
+		})
+	}
+}
+
+type fakeChunkSizeLimits struct {
+	TargetSize uint64
+}
+
+func (f fakeChunkSizeLimits) BloomTaskTargetChunkSizeBytes(tenantID string) uint64 {
+	return f.TargetSize
+}