yaml_test.go

// SPDX-License-Identifier: Apache-2.0

package output

import (
	"errors"
	"testing"
)

func TestOutput_YAML(t *testing.T) {
	// setup tests
	tests := []struct {
		failure bool
		input   interface{}
	}{
		{
			failure: false,
			input:   "hello",
		},
		{ // map
			failure: false,
			input:   map[string]string{"hello": "world"},
		},
		{ // slice
			failure: false,
			input:   []interface{}{1, 2, 3},
		},
		{ // slice complex
			failure: false,
			input:   []interface{}{struct{ Foo string }{Foo: "bar"}},
		},
		{ // complex
			failure: false,
			input:   []struct{ Foo string }{{"bar"}, {"baz"}},
		},
		{
			failure: true,
			input:   new(failMarshaler),
		},
		{
			failure: true,
			input:   nil,
		},
		{
			failure: true,
			input:   "",
		},
	}

	// run tests
	for _, test := range tests {
		err := YAML(test.input)

		if test.failure {
			if err == nil {
				t.Errorf("YAML should have returned err")
			}

			continue
		}

		if err != nil {
			t.Errorf("YAML returned err: %v", err)
		}
	}
}

func (f *failMarshaler) MarshalYAML() (interface{}, error) {
	return nil, errors.New("this is a marshaler that fails when you try to marshal")
}

func (f *failMarshaler) UnmarshalYAML(unmarshal func(interface{}) error) error {
	return errors.New("this is a marshaler that fails when you try to unmarshal")
}