HGKeeper

// Convey

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

package yaml

import (

"testing"

"github.com/stretchr/testify/assert"

"gopkg.in/yaml.v2"

)

func unmarshal(t *testing.T, str string) StringOrSlice {

data := StringOrSlice{}

err := yaml.Unmarshal([]byte(str), &data)

assert.Nil(t, err)

return data

}

func TestEmpty(t *testing.T) {

data := unmarshal(t, "")

assert.Equal(t, StringOrSlice{}, data)

}

func TestString(t *testing.T) {

data := unmarshal(t, "string")

assert.Equal(t, StringOrSlice{"string"}, data)

}

func TestSlice(t *testing.T) {

data := unmarshal(t, "['abc', '123']")

assert.Equal(t, StringOrSlice{"abc", "123"}, data)

}