89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
|
// Copyright 2012 Aaron Jacobs. All Rights Reserved.
|
||
|
|
// Author: aaronjjacobs@gmail.com (Aaron Jacobs)
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
package oglematchers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"reflect"
|
||
|
|
)
|
||
|
|
|
||
|
|
var byteSliceType reflect.Type = reflect.TypeOf([]byte{})
|
||
|
|
|
||
|
|
// DeepEquals returns a matcher that matches based on 'deep equality', as
|
||
|
|
// defined by the reflect package. This matcher requires that values have
|
||
|
|
// identical types to x.
|
||
|
|
func DeepEquals(x interface{}) Matcher {
|
||
|
|
return &deepEqualsMatcher{x}
|
||
|
|
}
|
||
|
|
|
||
|
|
type deepEqualsMatcher struct {
|
||
|
|
x interface{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *deepEqualsMatcher) Description() string {
|
||
|
|
xDesc := fmt.Sprintf("%v", m.x)
|
||
|
|
xValue := reflect.ValueOf(m.x)
|
||
|
|
|
||
|
|
// Special case: fmt.Sprintf presents nil slices as "[]", but
|
||
|
|
// reflect.DeepEqual makes a distinction between nil and empty slices. Make
|
||
|
|
// this less confusing.
|
||
|
|
if xValue.Kind() == reflect.Slice && xValue.IsNil() {
|
||
|
|
xDesc = "<nil slice>"
|
||
|
|
}
|
||
|
|
|
||
|
|
return fmt.Sprintf("deep equals: %s", xDesc)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *deepEqualsMatcher) Matches(c interface{}) error {
|
||
|
|
// Make sure the types match.
|
||
|
|
ct := reflect.TypeOf(c)
|
||
|
|
xt := reflect.TypeOf(m.x)
|
||
|
|
|
||
|
|
if ct != xt {
|
||
|
|
return NewFatalError(fmt.Sprintf("which is of type %v", ct))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Special case: handle byte slices more efficiently.
|
||
|
|
cValue := reflect.ValueOf(c)
|
||
|
|
xValue := reflect.ValueOf(m.x)
|
||
|
|
|
||
|
|
if ct == byteSliceType && !cValue.IsNil() && !xValue.IsNil() {
|
||
|
|
xBytes := m.x.([]byte)
|
||
|
|
cBytes := c.([]byte)
|
||
|
|
|
||
|
|
if bytes.Equal(cBytes, xBytes) {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return errors.New("")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Defer to the reflect package.
|
||
|
|
if reflect.DeepEqual(m.x, c) {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Special case: if the comparison failed because c is the nil slice, given
|
||
|
|
// an indication of this (since its value is printed as "[]").
|
||
|
|
if cValue.Kind() == reflect.Slice && cValue.IsNil() {
|
||
|
|
return errors.New("which is nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
return errors.New("")
|
||
|
|
}
|