Add golden config tests to severity operator (#71)

* Add golden config tests

operator/helper/severity_test.go

@@ -16,12 +16,15 @@ package helper
 
 import (
  "fmt"
+ "io/ioutil"
+ "path"
  "strings"
  "testing"
 
  "github.com/open-telemetry/opentelemetry-log-collection/entry"
  "github.com/open-telemetry/opentelemetry-log-collection/testutil"
  "github.com/stretchr/testify/require"
+ yaml "gopkg.in/yaml.v2"
 )
 
 type severityTestCase struct {
@@ -392,3 +395,122 @@ func (tc severityTestCase) run(parseFrom entry.Field) func(*testing.T) {
  require.Equal(t, tc.expected, ent.Severity)
  }
 }
+
+type severityConfigTestCase struct {
+ name string
+ expectErr bool
+ expect *SeverityParserConfig
+}
+
+func TestGoldenSeverityParserConfig(t *testing.T) {
+ cases := []severityConfigTestCase{
+ {
+ "default",
+ false,
+ defaultSeverityCfg(),
+ },
+ {
+ "parse_from_simple",
+ false,
+ func() *SeverityParserConfig {
+ cfg := defaultSeverityCfg()
+ newParse := entry.NewRecordField("from")
+ cfg.ParseFrom = &newParse
+ return cfg
+ }(),
+ },
+ {
+ "mapping",
+ false,
+ func() *SeverityParserConfig {
+ cfg := defaultSeverityCfg()
+ cfg.Mapping = map[interface{}]interface{}{
+ "critical": "5xx",
+ "error": "4xx",
+ "info": "3xx",
+ "debug": "2xx",
+ }
+ return cfg
+ }(),
+ },
+ {
+ "preserve_to",
+ false,
+ func() *SeverityParserConfig {
+ cfg := defaultSeverityCfg()
+ preserve := entry.NewRecordField("aField")
+ cfg.PreserveTo = &preserve
+ return cfg
+ }(),
+ },
+ {
+ "preset",
+ false,
+ func() *SeverityParserConfig {
+ cfg := defaultSeverityCfg()
+ cfg.Preset = "default"
+ return cfg
+ }(),
+ },
+ }
+
+ for _, tc := range cases {
+ t.Run("yaml/"+tc.name, func(t *testing.T) {
+ cfgFromYaml, yamlErr := configFromFileViaYaml(path.Join(".", "severitytestdata", fmt.Sprintf("%s.yaml", tc.name)))
+ if tc.expectErr {
+ require.Error(t, yamlErr)
+ } else {
+ require.NoError(t, yamlErr)
+ require.Equal(t, tc.expect, cfgFromYaml)
+ }
+ })
+ t.Run("mapstructure/"+tc.name, func(t *testing.T) {
+ cfgFromMapstructure := defaultSeverityCfg()
+ mapErr := configFromFileViaMapstructure(
+ path.Join(".", "severitytestdata", fmt.Sprintf("%s.yaml", tc.name)),
+ cfgFromMapstructure,
+ )
+ if tc.expectErr {
+ require.Error(t, mapErr)
+ } else {
+ require.NoError(t, mapErr)
+ require.Equal(t, tc.expect, cfgFromMapstructure)
+ }
+ })
+ }
+}
+
+func configFromFileViaYaml(file string) (*SeverityParserConfig, error) {
+ bytes, err := ioutil.ReadFile(file)
+ if err != nil {
+ return nil, fmt.Errorf("could not find config file: %s", err)
+ }
+
+ config := defaultSeverityCfg()
+ if err := yaml.Unmarshal(bytes, config); err != nil {
+ return nil, fmt.Errorf("failed to read config file as yaml: %s", err)
+ }
+
+ return config, nil
+}
+
+func configFromFileViaMapstructure(file string, result *SeverityParserConfig) error {
+ bytes, err := ioutil.ReadFile(file)
+ if err != nil {
+ return fmt.Errorf("could not find config file: %s", err)
+ }
+
+ raw := map[string]interface{}{}
+
+ if err := yaml.Unmarshal(bytes, raw); err != nil {
+ return fmt.Errorf("failed to read data from yaml: %s", err)
+ }
+
+ err = UnmarshalMapstructure(raw, result)
+ return err
+}
+
+func defaultSeverityCfg() *SeverityParserConfig {
+ newCfg := NewSeverityParserConfig()
+ return &newCfg
+}

operator/helper/severitytestdata/default.yaml

@@ -0,0 +1 @@
+type: severity_parser

operator/helper/severitytestdata/mapping.yaml

@@ -0,0 +1,6 @@
+type: severity_parser
+mapping:
+ critical: 5xx
+ error: 4xx
+ info: 3xx
+ debug: 2xx

operator/helper/severitytestdata/on_error_drop.yaml

@@ -0,0 +1,2 @@
+type: severity_parser
+on_error: drop

operator/helper/severitytestdata/parse_from_simple.yaml

@@ -0,0 +1,2 @@
+type: severity_parser
+parse_from: $.from

operator/helper/severitytestdata/preserve_to.yaml

@@ -0,0 +1,2 @@
+type: severity_parser
+preserve_to: aField

operator/helper/severitytestdata/preset.yaml

@@ -0,0 +1,2 @@
+type: severity_parser
+preset: default

operator/helper/severitytestdata/severity.yaml

@@ -0,0 +1,3 @@
+type: severity_parser
+severity:
+ parse_from: severity_field