2018-01-29 19:44:18 +00:00
package matchers
import (
2018-09-05 15:36:14 +00:00
"bytes"
2018-01-29 19:44:18 +00:00
"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." )
}
2018-09-05 15:36:14 +00:00
// 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
}
}
2018-01-29 19:44:18 +00:00
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 )
}