| package matchers |
| |
| import ( |
| "bytes" |
| "fmt" |
| "reflect" |
| |
| "github.com/onsi/gomega/format" |
| ) |
| |
| type EqualMatcher struct { |
| Expected interface{} |
| } |
| |
| func (matcher *EqualMatcher) Match(actual interface{}) (success bool, err error) { |
| if actual == nil && matcher.Expected == nil { |
| 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.") |
| } |
| // Shortcut for byte slices. |
| // Comparing long byte slices with reflect.DeepEqual is very slow, |
| // so use bytes.Equal if actual and expected are both byte slices. |
| if actualByteSlice, ok := actual.([]byte); ok { |
| if expectedByteSlice, ok := matcher.Expected.([]byte); ok { |
| return bytes.Equal(actualByteSlice, expectedByteSlice), nil |
| } |
| } |
| return reflect.DeepEqual(actual, matcher.Expected), nil |
| } |
| |
| func (matcher *EqualMatcher) FailureMessage(actual interface{}) (message string) { |
| actualString, actualOK := actual.(string) |
| expectedString, expectedOK := matcher.Expected.(string) |
| if actualOK && expectedOK { |
| return format.MessageWithDiff(actualString, "to equal", expectedString) |
| } |
| |
| return format.Message(actual, "to equal", matcher.Expected) |
| } |
| |
| func (matcher *EqualMatcher) NegatedFailureMessage(actual interface{}) (message string) { |
| return format.Message(actual, "not to equal", matcher.Expected) |
| } |