have_cap_matcher.go 802 B

12345678910111213141516171819202122232425262728293031
  1. // untested sections: 2
  2. package matchers
  3. import (
  4. "fmt"
  5. "github.com/onsi/gomega/format"
  6. )
  7. type HaveCapMatcher struct {
  8. Count int
  9. }
  10. func (matcher *HaveCapMatcher) Match(actual interface{}) (success bool, err error) {
  11. length, ok := capOf(actual)
  12. if !ok {
  13. return false, fmt.Errorf("HaveCap matcher expects a array/channel/slice. Got:\n%s", format.Object(actual, 1))
  14. }
  15. return length == matcher.Count, nil
  16. }
  17. func (matcher *HaveCapMatcher) FailureMessage(actual interface{}) (message string) {
  18. return fmt.Sprintf("Expected\n%s\nto have capacity %d", format.Object(actual, 1), matcher.Count)
  19. }
  20. func (matcher *HaveCapMatcher) NegatedFailureMessage(actual interface{}) (message string) {
  21. return fmt.Sprintf("Expected\n%s\nnot to have capacity %d", format.Object(actual, 1), matcher.Count)
  22. }