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

     1  // untested sections: 3
     2  
     3  package matchers
     4  
     5  import (
     6  	"fmt"
     7  	"time"
     8  
     9  	"github.com/onsi/gomega/format"
    10  )
    11  
    12  type BeTemporallyMatcher struct {
    13  	Comparator string
    14  	CompareTo  time.Time
    15  	Threshold  []time.Duration
    16  }
    17  
    18  func (matcher *BeTemporallyMatcher) FailureMessage(actual interface{}) (message string) {
    19  	return format.Message(actual, fmt.Sprintf("to be %s", matcher.Comparator), matcher.CompareTo)
    20  }
    21  
    22  func (matcher *BeTemporallyMatcher) NegatedFailureMessage(actual interface{}) (message string) {
    23  	return format.Message(actual, fmt.Sprintf("not to be %s", matcher.Comparator), matcher.CompareTo)
    24  }
    25  
    26  func (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  
    51  func (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  }