Function TestOverridesExtension_UnmarshalYAML

modules/overrides/extension_test.go:235–317

(t *testing.T)

}

func TestOverridesExtension_UnmarshalYAML(t *testing.T) {
	t.Run("overrides", func(t *testing.T) {
		resetRegistryForTesting(t)
		get := RegisterExtension(&testExtension{})

		input := `
ingestion:
  max_traces_per_user: 1000
test_extension:
  field_a: from_yaml
  field_b: 11
`
		var o Overrides
		require.NoError(t, yaml.Unmarshal([]byte(input), &o))

		assert.Equal(t, 1000, o.Ingestion.MaxLocalTracesPerUser)
		ext := get(&o)
		require.NotNil(t, ext)
		assert.Equal(t, "from_yaml", ext.FieldA)
		assert.Equal(t, 11, *ext.FieldB)
	})

	t.Run("Overrides_strict_decoder_absorbs_extension_key", func(t *testing.T) {
		resetRegistryForTesting(t)
		RegisterExtension(&testExtension{})

		input := `
ingestion:
  max_traces_per_user: 500
test_extension:
  field_a: strict_ok
`
		var o Overrides
		decoder := yaml.NewDecoder(strings.NewReader(input))
		decoder.SetStrict(true)
		require.NoError(t, decoder.Decode(&o), "strict YAML decoder must not error on registered extension keys")
	})

	t.Run("overrides unregistered key", func(t *testing.T) {
		resetRegistryForTesting(t)

		input := `
ingestion:
  max_traces_per_user: 1
unknown_ext:
  x: 1
`
		var o Overrides
		err := yaml.Unmarshal([]byte(input), &o)
		require.ErrorContains(t, err, "unknown extension key")
	})

	t.Run("legacy overrides", func(t *testing.T) {
		resetRegistryForTesting(t)
		get := RegisterExtension(&testExtension{})

		input := `
max_traces_per_user: 1000