Function TestAddIngestBlocks

cmd/tempo-cli/cmd-migrate-config_test.go:165–208

(t *testing.T)

}

func TestAddIngestBlocks(t *testing.T) {
	t.Run("monolithic mode skips kafka", func(t *testing.T) {
		m := map[string]interface{}{}
		err := addIngestBlocks(m, modeMonolithic, "", "")
		require.NoError(t, err)
		assert.NotContains(t, m, "ingest")
	})

	t.Run("microservices mode requires kafka address", func(t *testing.T) {
		m := map[string]interface{}{}
		err := addIngestBlocks(m, modeMicroservices, "", "tempo")
		require.Error(t, err)
		assert.Contains(t, err.Error(), "--kafka-address is required")
	})

	t.Run("microservices mode adds kafka config", func(t *testing.T) {
		m := map[string]interface{}{}
		err := addIngestBlocks(m, modeMicroservices, "kafka:9092", "my-topic")
		require.NoError(t, err)

		ingest := m["ingest"].(map[string]interface{})
		kafka := ingest["kafka"].(map[string]interface{})
		assert.Equal(t, "kafka:9092", kafka["address"])
		assert.Equal(t, "my-topic", kafka["topic"])
	})

	t.Run("microservices mode merges into existing ingest", func(t *testing.T) {
		m := map[string]interface{}{
			"ingest": map[string]interface{}{
				"kafka": map[string]interface{}{
					"client_id": "my-client",
				},
			},
		}
		err := addIngestBlocks(m, modeMicroservices, "kafka:9092", "tempo")
		require.NoError(t, err)

		ingest := m["ingest"].(map[string]interface{})
		kafka := ingest["kafka"].(map[string]interface{})
		assert.Equal(t, "kafka:9092", kafka["address"])
		assert.Equal(t, "tempo", kafka["topic"])
		assert.Equal(t, "my-client", kafka["client_id"])
	})
}

func TestModifyOverrides(t *testing.T) {
	t.Run("creates overrides and sets compaction_disabled", func(t *testing.T) {