Function TestBackendScheduler

integration/storage/backend_scheduler_test.go:30–268

(t *testing.T)

)

func TestBackendScheduler(t *testing.T) {
	util.RunIntegrationTests(t, util.TestHarnessConfig{
		Components:    util.ComponentsBackendWork,
		Backends:      util.BackendObjectStorageAll,
		ConfigOverlay: "config-backend-scheduler.yaml",
	}, func(h *util.TempoHarness) {
		ctx, cancel := context.WithCancel(context.Background())
		defer cancel()

		s := h.TestScenario

		// Get the config from the harness
		cfg, err := h.GetConfig()
		require.NoError(t, err)

		objStorage := h.Services[util.ServiceObjectStorage]
		backendEndpoint := objStorage.HTTPEndpoint()
		t.Logf("Endpoint: %s", backendEndpoint)

		// Get the scheduler and worker services from the harness
		scheduler := h.Services[util.ServiceBackendScheduler]
		worker := h.Services[util.ServiceBackendWorker]

		// Stop the worker initially - we'll start it later after populating data
		require.NoError(t, s.Stop(worker))

		// Setup tempodb with local backend
		tempodbWriter := setupBackendWithEndpoint(t, &cfg.StorageConfig.Trace, backendEndpoint)

		cases := []struct {
			name                      string
			tenantCount               int
			blockCount                int
			expectedBlocks            int // accumulated expected blocks.  Each test adds this to the total
			expectedOutstandingBlocks int // accumulated expected outstanding blocks.  Each test adds this to the total, but 1 outstanding block is between 2-4 blocks.
		}{
			{
				name:                      "a bunch of tenants with 1 block each is 0 outstanding",
				tenantCount:               11,
				blockCount:                1,
				expectedBlocks:            1,
				expectedOutstandingBlocks: 0,
			},
			{
				name:                      "a bunch of tenants with 12 blocks each 12 outstanding",
				tenantCount:               11,
				blockCount:                12,
				expectedBlocks:            12,
				expectedOutstandingBlocks: 12,
			},
		}
		var tenants []string

		type expectations struct {
			blocks      int
			outstanding int
		}