1package matchers
 2
 3import (
 4	"bytes"
 5	"fmt"
 6	"reflect"
 7
 8	"github.com/onsi/gomega/format"
 9)
10
11type EqualMatcher struct {
12	Expected interface{}
13}
14
15func (matcher *EqualMatcher) Match(actual interface{}) (success bool, err error) {
16	if actual == nil && matcher.Expected == nil {
17		return false, fmt.Errorf("Refusing to compare <nil> to <nil>.\nBe explicit and use BeNil() instead.  This is to avoid mistakes where both sides of an assertion are erroneously uninitialized.")
18	}
19	// Shortcut for byte slices.
20	// Comparing long byte slices with reflect.DeepEqual is very slow,
21	// so use bytes.Equal if actual and expected are both byte slices.
22	if actualByteSlice, ok := actual.([]byte); ok {
23		if expectedByteSlice, ok := matcher.Expected.([]byte); ok {
24			return bytes.Equal(actualByteSlice, expectedByteSlice), nil
25		}
26	}
27	return reflect.DeepEqual(actual, matcher.Expected), nil
28}
29
30func (matcher *EqualMatcher) FailureMessage(actual interface{}) (message string) {
31	actualString, actualOK := actual.(string)
32	expectedString, expectedOK := matcher.Expected.(string)
33	if actualOK && expectedOK {
34		return format.MessageWithDiff(actualString, "to equal", expectedString)
35	}
36
37	return format.Message(actual, "to equal", matcher.Expected)
38}
39
40func (matcher *EqualMatcher) NegatedFailureMessage(actual interface{}) (message string) {
41	return format.Message(actual, "not to equal", matcher.Expected)
42}