set_test.go

package goml_test

import (
	. "github.com/JulzDiverse/goml"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	"github.com/smallfish/simpleyaml"
)

var _ = Describe("Set", func() {
	var yml *simpleyaml.Yaml
	var err error

	BeforeEach(func() {
		yaml := `map:
  name: foo

array:
- bar
- var
- zar

mapArray:
- foo: bar
  zoo: lion
  arr:
  - one
  - two
  - three
- foo: var
  boo: laa`

		yml, err = simpleyaml.NewYaml([]byte(yaml))
		Expect(err).NotTo(HaveOccurred())
	})

	It("should add an value to an array", func() {
		err = Set(yml, "array.2", "bumblebee")
		Expect(err).NotTo(HaveOccurred())
		err = Set(yml, "array.+", "optimusPrime")
		Expect(err).NotTo(HaveOccurred())
		err = Set(yml, "mapArray.0.foo", "wolverine")
		Expect(err).NotTo(HaveOccurred())
		err = Set(yml, "mapArray.foo:var.boo", "baymax")
		Expect(err).NotTo(HaveOccurred())

		Expect(Get(yml, "array.2")).To(Equal("bumblebee"))
		Expect(Get(yml, "array.3")).To(Equal("optimusPrime"))
		Expect(Get(yml, "mapArray.0.foo")).To(Equal("wolverine"))
		Expect(Get(yml, "mapArray.foo:var.boo")).To(Equal("baymax"))

		err = Set(yml, "array.:optimusPrime", "pikachu")
		Expect(Get(yml, "array.:pikachu")).To(Equal("pikachu"))
		err = Set(yml, "mapArray.foo:wolverine.arr.0", "new")
		Expect(Get(yml, "mapArray.foo:wolverine.arr.0")).To(Equal("new"))
	})
})