Function Test_instance_updateProcessors

modules/generator/instance_test.go:141–447

(t *testing.T)

}

func Test_instance_updateProcessors(t *testing.T) {
	cfg := Config{}
	cfg.RegisterFlagsAndApplyDefaults("", &flag.FlagSet{})
	logger := log.NewLogfmtLogger(log.NewSyncWriter(os.Stdout))
	overrides := mockOverrides{}

	instance, err := newInstance(&cfg, "test", &overrides, &noopStorage{}, logger)
	assert.NoError(t, err)

	// stop the update goroutine
	close(instance.shutdownCh)

	// no processors should be present initially
	assert.Len(t, instance.processors, 0)

	t.Run("add servicegraphs processors", func(t *testing.T) {
		overrides.processors = map[string]struct{}{
			processor.ServiceGraphsName: {},
		}
		err := instance.updateProcessors()
		assert.NoError(t, err)

		assert.Len(t, instance.processors, 1)
		assert.Equal(t, instance.processors[processor.ServiceGraphsName].Name(), processor.ServiceGraphsName)
	})

	t.Run("ignore unknown processor", func(t *testing.T) {
		overrides.processors = map[string]struct{}{
			"span-metricsss": {}, // typo in the overrides
		}
		err := instance.updateProcessors()
		assert.NoError(t, err)

		// unknown processors are ignored and therefore the desired processor set is empty.
		assert.Len(t, instance.processors, 0)
	})

	t.Run("ignore removed local-blocks processor and keep valid processors", func(t *testing.T) {
		overrides.processors = map[string]struct{}{
			"local-blocks":              {},
			processor.ServiceGraphsName: {},
		}
		err := instance.updateProcessors()
		assert.NoError(t, err)

		assert.Len(t, instance.processors, 1)
		assert.Equal(t, processor.ServiceGraphsName, instance.processors[processor.ServiceGraphsName].Name())
	})

	t.Run("add spanmetrics processor", func(t *testing.T) {
		overrides.processors = map[string]struct{}{
			processor.ServiceGraphsName: {},
			processor.SpanMetricsName:   {},
		}
		err := instance.updateProcessors()
		assert.NoError(t, err)

		assert.Len(t, instance.processors, 2)