openrtb/native/response/raw_json_test.go

package response_test

import (
	"encoding/json"

	. "github.com/mxmCherry/openrtb/native/response"

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

var _ = Describe("RawJSON", func() {

	var _ json.Marshaler = (RawJSON)(nil)
	var _ json.Unmarshaler = (*RawJSON)(nil)

	It("should encode JSON", func() {
		subject := RawJSON(`true`)

		actual, err := subject.MarshalJSON()
		Expect(err).NotTo(HaveOccurred())
		Expect(actual).To(Equal([]byte(`true`)))
	})

	It("should decode JSON", func() {
		subject := RawJSON(nil)

		err := subject.UnmarshalJSON([]byte(`true`))
		Expect(err).NotTo(HaveOccurred())
		Expect(subject).To(Equal(RawJSON(`true`)))
	})

	It("should decode JSON when embedded into struct", func() {
		wrapper := struct {
			Raw RawJSON `json:"raw"`
		}{
			Raw: nil,
		}

		err := json.Unmarshal([]byte(`{"raw":true}`), &wrapper)
		Expect(err).NotTo(HaveOccurred())
		Expect(wrapper.Raw).To(Equal(RawJSON(`true`)))
	})

	It("should encode JSON when embedded into struct", func() {
		wrapper := struct {
			Raw RawJSON `json:"raw"`
		}{
			Raw: RawJSON(`true`),
		}

		actual, err := json.Marshal(wrapper)
		Expect(err).NotTo(HaveOccurred())
		Expect(actual).To(MatchJSON(`{"raw":true}`))
	})

})