1// untested sections: 6
 2
 3package matchers
 4
 5import (
 6	"fmt"
 7	"reflect"
 8
 9	"github.com/onsi/gomega/format"
10)
11
12type HaveKeyMatcher struct {
13	Key interface{}
14}
15
16func (matcher *HaveKeyMatcher) Match(actual interface{}) (success bool, err error) {
17	if !isMap(actual) {
18		return false, fmt.Errorf("HaveKey matcher expects a map.  Got:%s", format.Object(actual, 1))
19	}
20
21	keyMatcher, keyIsMatcher := matcher.Key.(omegaMatcher)
22	if !keyIsMatcher {
23		keyMatcher = &EqualMatcher{Expected: matcher.Key}
24	}
25
26	keys := reflect.ValueOf(actual).MapKeys()
27	for i := 0; i < len(keys); i++ {
28		success, err := keyMatcher.Match(keys[i].Interface())
29		if err != nil {
30			return false, fmt.Errorf("HaveKey's key matcher failed with:\n%s%s", format.Indent, err.Error())
31		}
32		if success {
33			return true, nil
34		}
35	}
36
37	return false, nil
38}
39
40func (matcher *HaveKeyMatcher) FailureMessage(actual interface{}) (message string) {
41	switch matcher.Key.(type) {
42	case omegaMatcher:
43		return format.Message(actual, "to have key matching", matcher.Key)
44	default:
45		return format.Message(actual, "to have key", matcher.Key)
46	}
47}
48
49func (matcher *HaveKeyMatcher) NegatedFailureMessage(actual interface{}) (message string) {
50	switch matcher.Key.(type) {
51	case omegaMatcher:
52		return format.Message(actual, "not to have key matching", matcher.Key)
53	default:
54		return format.Message(actual, "not to have key", matcher.Key)
55	}
56}