multi_writer_test.go

package peanut_test

import (
	"errors"

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

	"github.com/jimsmart/peanut"
)

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

	It("should call the methods of the writers it wraps", func() {

		w1 := &peanut.MockWriter{}
		w2 := &peanut.MockWriter{}
		w := peanut.MultiWriter(w1, w2)

		testWritesAndCloseSequential(w)

		Expect(w1.CalledWrite).To(Equal(10))
		Expect(w1.CalledClose).To(Equal(1))
		Expect(w1.CalledCancel).To(Equal(1))

		// Did it actually write to both?
		Expect(w1).To(Equal(w2))
	})

	It("should forward errors from Close", func() {

		w1 := &peanut.MockWriter{}
		w2 := &failWriter{}
		w := peanut.MultiWriter(w1, w2)

		err := w.Close()
		Expect(err).NotTo(BeNil())
	})

	It("should forward errors from Cancel", func() {

		w1 := &peanut.MockWriter{}
		w2 := &failWriter{}
		w := peanut.MultiWriter(w1, w2)

		err := w.Cancel()
		Expect(err).NotTo(BeNil())
	})
})

type failWriter struct{}

func (failWriter) Write(x interface{}) error {
	return errors.New("fail")
}

func (failWriter) Close() error {
	return errors.New("fail")
}

func (failWriter) Cancel() error {
	return errors.New("fail")
}