1// untested sections: 3
 2
 3package matchers
 4
 5import (
 6	"fmt"
 7	"time"
 8
 9	"github.com/onsi/gomega/format"
10)
11
12type BeTemporallyMatcher struct {
13	Comparator string
14	CompareTo  time.Time
15	Threshold  []time.Duration
16}
17
18func (matcher *BeTemporallyMatcher) FailureMessage(actual interface{}) (message string) {
19	return format.Message(actual, fmt.Sprintf("to be %s", matcher.Comparator), matcher.CompareTo)
20}
21
22func (matcher *BeTemporallyMatcher) NegatedFailureMessage(actual interface{}) (message string) {
23	return format.Message(actual, fmt.Sprintf("not to be %s", matcher.Comparator), matcher.CompareTo)
24}
25
26func (matcher *BeTemporallyMatcher) Match(actual interface{}) (bool, error) {
27	// predicate to test for time.Time type
28	isTime := func(t interface{}) bool {
29		_, ok := t.(time.Time)
30		return ok
31	}
32
33	if !isTime(actual) {
34		return false, fmt.Errorf("Expected a time.Time.  Got:\n%s", format.Object(actual, 1))
35	}
36
37	switch matcher.Comparator {
38	case "==", "~", ">", ">=", "<", "<=":
39	default:
40		return false, fmt.Errorf("Unknown comparator: %s", matcher.Comparator)
41	}
42
43	var threshold = time.Millisecond
44	if len(matcher.Threshold) == 1 {
45		threshold = matcher.Threshold[0]
46	}
47
48	return matcher.matchTimes(actual.(time.Time), matcher.CompareTo, threshold), nil
49}
50
51func (matcher *BeTemporallyMatcher) matchTimes(actual, compareTo time.Time, threshold time.Duration) (success bool) {
52	switch matcher.Comparator {
53	case "==":
54		return actual.Equal(compareTo)
55	case "~":
56		diff := actual.Sub(compareTo)
57		return -threshold <= diff && diff <= threshold
58	case ">":
59		return actual.After(compareTo)
60	case ">=":
61		return !actual.Before(compareTo)
62	case "<":
63		return actual.Before(compareTo)
64	case "<=":
65		return !actual.After(compareTo)
66	}
67	return false
68}