package config import ( "gopkg.in/yaml.v2" "reflect" "testing" ) // Unmarshal good configuration file and confirm data is flattened correctly func TestConfigFlattening(t *testing.T) { goodYamlConfig := []byte(`--- collectors: enabled: cpu,net,service log: level: debug`) var data map[string]interface{} err := yaml.Unmarshal(goodYamlConfig, &data) if err != nil { t.Error(err) } expectedResult := map[string]string{ "collectors.enabled": "cpu,net,service", "log.level": "debug", } flattenedValues := flatten(data) if !reflect.DeepEqual(expectedResult, flattenedValues) { t.Errorf("Flattened values do not match!\nExpected result: %s\nActual result: %s", expectedResult, flattenedValues) } }