Function TestKVStores

integration/operations/kvstores_test.go:18–103

(t *testing.T)

const configKVStore = "config-kvstore.yaml"

func TestKVStores(t *testing.T) {
	testKVStores := []struct {
		name          string
		setupKVStore  func(*e2e.Scenario, map[string]any) error
		kvstoreConfig string
	}{
		{
			name: "memberlist",
			setupKVStore: func(_ *e2e.Scenario, templateData map[string]any) error {
				templateData["KVStoreConfig"] = `
      store: memberlist`
				return nil
			},
		},
		{
			name: "etcd",
			setupKVStore: func(s *e2e.Scenario, templateData map[string]any) error {
				etcd := e2edb.NewETCD()
				if err := s.StartAndWaitReady(etcd); err != nil {
					return err
				}
				templateData["KVStoreConfig"] = fmt.Sprintf(`
      store: etcd
      etcd:
        endpoints:
          - http://%s:%d`, etcd.Name(), etcd.HTTPPort())
				return nil
			},
		},
		{
			name: "consul",
			setupKVStore: func(s *e2e.Scenario, templateData map[string]any) error {
				consul := e2edb.NewConsul()
				if err := s.StartAndWaitReady(consul); err != nil {
					return err
				}
				templateData["KVStoreConfig"] = fmt.Sprintf(`
      store: consul
      consul:
        host: http://%s:%d`, consul.Name(), consul.HTTPPort())
				return nil
			},
		},
	}

	for _, tc := range testKVStores {
		t.Run(tc.name, func(t *testing.T) {
			util.RunIntegrationTests(t, util.TestHarnessConfig{
				ConfigOverlay: configKVStore,
				PreStartHook:  tc.setupKVStore,
			}, func(h *util.TempoHarness) {
				h.WaitTracesWritable(t)

				liveStoreA := h.Services[util.ServiceLiveStoreZoneA]
				liveStoreB := h.Services[util.ServiceLiveStoreZoneB]

				matchers := []*labels.Matcher{
					{Type: labels.MatchEqual, Name: "state", Value: "ACTIVE"},