github.com/onsi/gomega@v1.32.0/matchers/be_key_of_matcher.go (about)

     1  package matchers
     2  
     3  import (
     4  	"fmt"
     5  	"reflect"
     6  
     7  	"github.com/onsi/gomega/format"
     8  )
     9  
    10  type BeKeyOfMatcher struct {
    11  	Map interface{}
    12  }
    13  
    14  func (matcher *BeKeyOfMatcher) Match(actual interface{}) (success bool, err error) {
    15  	if !isMap(matcher.Map) {
    16  		return false, fmt.Errorf("BeKeyOf matcher needs expected to be a map type")
    17  	}
    18  
    19  	if reflect.TypeOf(actual) == nil {
    20  		return false, fmt.Errorf("BeKeyOf matcher expects actual to be typed")
    21  	}
    22  
    23  	var lastError error
    24  	for _, key := range reflect.ValueOf(matcher.Map).MapKeys() {
    25  		matcher := &EqualMatcher{Expected: key.Interface()}
    26  		success, err := matcher.Match(actual)
    27  		if err != nil {
    28  			lastError = err
    29  			continue
    30  		}
    31  		if success {
    32  			return true, nil
    33  		}
    34  	}
    35  
    36  	return false, lastError
    37  }
    38  
    39  func (matcher *BeKeyOfMatcher) FailureMessage(actual interface{}) (message string) {
    40  	return format.Message(actual, "to be a key of", presentable(valuesOf(matcher.Map)))
    41  }
    42  
    43  func (matcher *BeKeyOfMatcher) NegatedFailureMessage(actual interface{}) (message string) {
    44  	return format.Message(actual, "not to be a key of", presentable(valuesOf(matcher.Map)))
    45  }