mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-10-19 13:49:53 +00:00
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
|
package matchers
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"reflect"
|
||
|
|
||
|
"github.com/onsi/gomega/format"
|
||
|
)
|
||
|
|
||
|
type BeKeyOfMatcher struct {
|
||
|
Map interface{}
|
||
|
}
|
||
|
|
||
|
func (matcher *BeKeyOfMatcher) Match(actual interface{}) (success bool, err error) {
|
||
|
if !isMap(matcher.Map) {
|
||
|
return false, fmt.Errorf("BeKeyOf matcher needs expected to be a map type")
|
||
|
}
|
||
|
|
||
|
if reflect.TypeOf(actual) == nil {
|
||
|
return false, fmt.Errorf("BeKeyOf matcher expects actual to be typed")
|
||
|
}
|
||
|
|
||
|
var lastError error
|
||
|
for _, key := range reflect.ValueOf(matcher.Map).MapKeys() {
|
||
|
matcher := &EqualMatcher{Expected: key.Interface()}
|
||
|
success, err := matcher.Match(actual)
|
||
|
if err != nil {
|
||
|
lastError = err
|
||
|
continue
|
||
|
}
|
||
|
if success {
|
||
|
return true, nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return false, lastError
|
||
|
}
|
||
|
|
||
|
func (matcher *BeKeyOfMatcher) FailureMessage(actual interface{}) (message string) {
|
||
|
return format.Message(actual, "to be a key of", presentable(valuesOf(matcher.Map)))
|
||
|
}
|
||
|
|
||
|
func (matcher *BeKeyOfMatcher) NegatedFailureMessage(actual interface{}) (message string) {
|
||
|
return format.Message(actual, "not to be a key of", presentable(valuesOf(matcher.Map)))
|
||
|
}
|