Deflake ThrottledLogger test

This commit is contained in:
Jordan Liggitt 2020-11-19 13:59:27 -05:00
parent 3af376d3ad
commit c84df5d66e

View File

@ -2228,14 +2228,18 @@ func TestRequestPreflightCheck(t *testing.T) {
func TestThrottledLogger(t *testing.T) { func TestThrottledLogger(t *testing.T) {
now := time.Now() now := time.Now()
oldClock := globalThrottledLogger.clock
defer func() {
globalThrottledLogger.clock = oldClock
}()
clock := clock.NewFakeClock(now) clock := clock.NewFakeClock(now)
globalThrottledLogger.clock = clock globalThrottledLogger.clock = clock
logMessages := 0 logMessages := 0
for i := 0; i < 10000; i++ { for i := 0; i < 1000; i++ {
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(100) wg.Add(10)
for j := 0; j < 100; j++ { for j := 0; j < 10; j++ {
go func() { go func() {
if _, ok := globalThrottledLogger.attemptToLog(); ok { if _, ok := globalThrottledLogger.attemptToLog(); ok {
logMessages++ logMessages++
@ -2248,7 +2252,7 @@ func TestThrottledLogger(t *testing.T) {
clock.SetTime(now) clock.SetTime(now)
} }
if a, e := logMessages, 1000; a != e { if a, e := logMessages, 100; a != e {
t.Fatalf("expected %v log messages, but got %v", e, a) t.Fatalf("expected %v log messages, but got %v", e, a)
} }
} }