gomega_transform.go

package gomegahelpers

import (
	"github.com/onsi/gomega/types"
)

func Transform(transform func(oldValue interface{}) interface{}, matcher types.GomegaMatcher) types.GomegaMatcher {
	return &TransformMatcher{
		Transform: transform,
		Matcher:   matcher,
	}
}

type TransformMatcher struct {
	Transform func(oldValue interface{}) interface{}
	Matcher   types.GomegaMatcher
}

func (m *TransformMatcher) Match(actual interface{}) (success bool, err error) {
	return m.Matcher.Match(m.Transform(actual))
}

func (m *TransformMatcher) FailureMessage(actual interface{}) (message string) {
	return m.Matcher.FailureMessage(m.Transform(actual))
}

func (m *TransformMatcher) NegatedFailureMessage(actual interface{}) (message string) {
	return m.Matcher.NegatedFailureMessage(m.Transform(actual))
}