Function TestOverridesExtension_MarshalYAML

modules/overrides/extension_test.go:184–233

(t *testing.T)

}

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

		fieldB := 3
		o := Overrides{}
		o.Ingestion.MaxLocalTracesPerUser = 1000
		o.Extensions = map[string]any{
			"test_extension": &testExtension{FieldA: "yaml_val", FieldB: &fieldB},
		}

		b, err := yaml.Marshal(&o)
		require.NoError(t, err)

		// The marshaled YAML must contain the nested extension key.
		assert.Contains(t, string(b), "test_extension:")

		// Round-trip: unmarshal recovers the extension (Overrides.UnmarshalYAML calls processExtensions).
		var o2 Overrides
		require.NoError(t, yaml.Unmarshal(b, &o2))

		assert.Equal(t, 1000, o2.Ingestion.MaxLocalTracesPerUser)
		ext := get(&o2)
		require.NotNil(t, ext)
		assert.Equal(t, "yaml_val", ext.FieldA)
		assert.Equal(t, 3, *ext.FieldB)
	})

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

		fieldB := 8
		o := Overrides{}
		o.Ingestion.MaxLocalTracesPerUser = 500
		o.Extensions = map[string]any{
			"test_extension": &testExtension{FieldA: "legacy_yaml", FieldB: &fieldB},
		}
		l := o.toLegacy()

		b, err := yaml.Marshal(&l)
		require.NoError(t, err)

		// Flat keys must appear at the top level; nested key must not.
		yamlStr := string(b)
		assert.Contains(t, yamlStr, "test_extension_field_a:")
		assert.NotContains(t, yamlStr, "test_extension:")
	})
}

func TestOverridesExtension_UnmarshalYAML(t *testing.T) {
	t.Run("overrides", func(t *testing.T) {