Update assertions.go

fix for value.
pull/618/merge
Alexander Pantyukhin 2019-01-09 12:30:14 +04:00 committed by George Lesica
parent 834f27f4b7
commit 363ebb24d0
1 changed files with 3 additions and 3 deletions

View File

@ -629,7 +629,7 @@ func NotEqual(t TestingT, expected, actual interface{}, msgAndArgs ...interface{
func includeElement(list interface{}, element interface{}) (ok, found bool) { func includeElement(list interface{}, element interface{}) (ok, found bool) {
listValue := reflect.ValueOf(list) listValue := reflect.ValueOf(list)
listValueKind := listValue.Kind() listKind := reflect.TypeOf(list).Kind()
defer func() { defer func() {
if e := recover(); e != nil { if e := recover(); e != nil {
ok = false ok = false
@ -637,12 +637,12 @@ func includeElement(list interface{}, element interface{}) (ok, found bool) {
} }
}() }()
if listValueKind == reflect.String { if listKind == reflect.String {
elementValue := reflect.ValueOf(element) elementValue := reflect.ValueOf(element)
return true, strings.Contains(listValue.String(), elementValue.String()) return true, strings.Contains(listValue.String(), elementValue.String())
} }
if listValueKind == reflect.Map { if listKind == reflect.Map {
mapKeys := listValue.MapKeys() mapKeys := listValue.MapKeys()
for i := 0; i < len(mapKeys); i++ { for i := 0; i < len(mapKeys); i++ {
if ObjectsAreEqual(mapKeys[i].Interface(), element) { if ObjectsAreEqual(mapKeys[i].Interface(), element) {