Remove flakiness of google chat tests (#40592)

This commit is contained in:
Yuriy Tseretyan
2021-10-19 16:18:44 -04:00
committed by GitHub
parent 38709df33b
commit 36e12e2b26
4 changed files with 36 additions and 10 deletions
@@ -0,0 +1,23 @@
package channels
import "time"
// mockTimeNow replaces function timeNow to return constant time.
// It returns a function that resets the variable back to its original value.
// This allows usage of this function with defer:
// func Test (t *testing.T) {
// now := time.Now()
// defer mockTimeNow(now)()
// ...
// }
func mockTimeNow(constTime time.Time) func() {
timeNow = func() time.Time {
return constTime
}
return resetTimeNow
}
// resetTimeNow resets the global variable timeNow to the default value, which is time.Now
func resetTimeNow() {
timeNow = time.Now
}