1 // Copyright (c) 2023 Uber Technologies, Inc. 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining a copy 4 // of this software and associated documentation files (the "Software"), to deal 5 // in the Software without restriction, including without limitation the rights 6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 // copies of the Software, and to permit persons to whom the Software is 8 // furnished to do so, subject to the following conditions: 9 // 10 // The above copyright notice and this permission notice shall be included in 11 // all copies or substantial portions of the Software. 12 // 13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 // THE SOFTWARE. 20 21 package ztest 22 23 import ( 24 "sync/atomic" 25 "testing" 26 "time" 27 28 "github.com/stretchr/testify/assert" 29 ) 30 31 func TestMockClock_NewTicker(t *testing.T) { 32 var n atomic.Int32 33 clock := NewMockClock() 34 35 done := make(chan struct{}) 36 defer func() { <-done }() // wait for end 37 38 quit := make(chan struct{}) 39 // Create a channel to increment every microsecond. 40 go func(ticker *time.Ticker) { 41 defer close(done) 42 for { 43 select { 44 case <-quit: 45 ticker.Stop() 46 return 47 case <-ticker.C: 48 n.Add(1) 49 } 50 } 51 }(clock.NewTicker(time.Microsecond)) 52 53 // Move clock forward. 54 clock.Add(2 * time.Microsecond) 55 assert.Equal(t, int32(2), n.Load()) 56 close(quit) 57 } 58 59 func TestMockClock_NewTicker_slowConsumer(t *testing.T) { 60 clock := NewMockClock() 61 62 ticker := clock.NewTicker(time.Microsecond) 63 defer ticker.Stop() 64 65 // Two ticks, only one consumed. 66 clock.Add(2 * time.Microsecond) 67 <-ticker.C 68 69 select { 70 case <-ticker.C: 71 t.Fatal("unexpected tick") 72 default: 73 // ok 74 } 75 } 76 77 func TestMockClock_Add_negative(t *testing.T) { 78 clock := NewMockClock() 79 assert.Panics(t, func() { clock.Add(-1) }) 80 } 81