Function TestPollerOwnership

integration/storage/poller_test.go:42–163

(t *testing.T)

}

func TestPollerOwnership(t *testing.T) {
	storageBackendTestPermutations := []struct {
		name   string
		prefix string
	}{
		{
			name:   "empty-string-prefix",
			prefix: "",
		},
		{
			name: "no-prefix",
		},
		{
			name:   "prefix",
			prefix: "a/b/c/",
		},
		{
			name:   "prefix-no-trailing-slash",
			prefix: "a/b/c",
		},
	}

	logger := log.NewLogfmtLogger(os.Stdout)

	for _, pc := range storageBackendTestPermutations {
		t.Run(pc.name, func(t *testing.T) {
			util.RunIntegrationTests(t, util.TestHarnessConfig{
				DeploymentMode: util.DeploymentModeNone,
				Backends:       util.BackendObjectStorageAll,
			}, func(h *util.TempoHarness) {
				// Get the config to determine backend type and settings
				cfg, err := h.GetConfig()
				require.NoError(t, err)

				var rr backend.RawReader
				var ww backend.RawWriter
				var cc backend.Compactor

				listBlockConcurrency := 10

				// Get backend endpoint from the harness
				objStorage := h.Services[util.ServiceObjectStorage]
				e := objStorage.Endpoint(objStorage.HTTPPort())

				switch cfg.StorageConfig.Trace.Backend {
				case backend.S3:
					cfg.StorageConfig.Trace.S3.ListBlocksConcurrency = listBlockConcurrency
					cfg.StorageConfig.Trace.S3.Endpoint = e
					cfg.StorageConfig.Trace.S3.Prefix = pc.prefix
					rr, ww, cc, err = s3.New(cfg.StorageConfig.Trace.S3)
				case backend.GCS:
					cfg.StorageConfig.Trace.GCS.ListBlocksConcurrency = listBlockConcurrency
					cfg.StorageConfig.Trace.GCS.Endpoint = e
					cfg.StorageConfig.Trace.GCS.Prefix = pc.prefix
					rr, ww, cc, err = gcs.New(cfg.StorageConfig.Trace.GCS)
				case backend.Azure:
					cfg.StorageConfig.Trace.Azure.Endpoint = e
					cfg.StorageConfig.Trace.Azure.Prefix = pc.prefix