fixup! Add new function RenameYamlKey

Author: stefanhaller

pkg/utils/yaml_utils/yaml_utils_test.go

@@ -150,7 +150,7 @@ func TestRenameYamlKey(t *testing.T) {
 path: []string{"foo"},
 newKey: "bar",
 expectedOut: "42\n",
-expectedErr: "yaml document is not a dictionary",
+expectedErr: "yaml node in path is not a dictionary",
 },
 {
 name: "not all path elements are dictionaries",
@@ -171,18 +171,15 @@ func TestRenameYamlKey(t *testing.T) {
 }
 
 for _, test := range tests {
-test := test
 t.Run(test.name, func(t *testing.T) {
-out, err := RenameYamlKey([]byte(test.in), test.path, test.newKey)
-if test.expectedErr != "" {
-if err == nil {
-t.Errorf("expected error %q but got none", test.expectedErr)
-}
-} else if err != nil {
-t.Errorf("unexpected error: %v", err)
-} else if string(out) != test.expectedOut {
-t.Errorf("expected %q but got %q", test.expectedOut, string(out))
+out, actualErr := RenameYamlKey([]byte(test.in), test.path, test.newKey)
+if test.expectedErr == "" {
+assert.NoError(t, actualErr)
+} else {
+assert.EqualError(t, actualErr, test.expectedErr)
 }
+
+assert.Equal(t, test.expectedOut, string(out))
 })
 }
 }