| Server IP : 91.108.119.186 / Your IP : 216.73.217.6 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/go/pkg/mod/github.com/coder/quartz@v0.1.2/ |
Upload File : |
package quartz_test
import (
"context"
"sync"
"testing"
"time"
"github.com/coder/quartz"
)
type exampleTickCounter struct {
ctx context.Context
mu sync.Mutex
ticks int
clock quartz.Clock
}
func (c *exampleTickCounter) Ticks() int {
c.mu.Lock()
defer c.mu.Unlock()
return c.ticks
}
func (c *exampleTickCounter) count() {
_ = c.clock.TickerFunc(c.ctx, time.Hour, func() error {
c.mu.Lock()
defer c.mu.Unlock()
c.ticks++
return nil
}, "mytag")
}
func newExampleTickCounter(ctx context.Context, clk quartz.Clock) *exampleTickCounter {
tc := &exampleTickCounter{ctx: ctx, clock: clk}
go tc.count()
return tc
}
// TestExampleTickerFunc demonstrates how to test the use of TickerFunc.
func TestExampleTickerFunc(t *testing.T) {
t.Parallel()
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
mClock := quartz.NewMock(t)
// Because the ticker is started on a goroutine, we can't immediately start
// advancing the clock, or we will race with the start of the ticker. If we
// win that race, the clock gets advanced _before_ the ticker starts, and
// our ticker will not get a tick.
//
// To handle this, we set a trap for the call to TickerFunc(), so that we
// can assert it has been called before advancing the clock.
trap := mClock.Trap().TickerFunc("mytag")
defer trap.Close()
tc := newExampleTickCounter(ctx, mClock)
// Here, we wait for our trap to be triggered.
call, err := trap.Wait(ctx)
if err != nil {
t.Fatal("ticker never started")
}
// it's good practice to release calls before any possible t.Fatal() calls
// so that we don't leave dangling goroutines waiting for the call to be
// released.
call.Release()
if call.Duration != time.Hour {
t.Fatal("unexpected duration")
}
if tks := tc.Ticks(); tks != 0 {
t.Fatalf("expected 0 got %d ticks", tks)
}
// Now that we know the ticker is started, we can advance the time.
mClock.Advance(time.Hour).MustWait(ctx)
if tks := tc.Ticks(); tks != 1 {
t.Fatalf("expected 1 got %d ticks", tks)
}
}
type exampleLatencyMeasurer struct {
mu sync.Mutex
lastLatency time.Duration
}
func newExampleLatencyMeasurer(ctx context.Context, clk quartz.Clock) *exampleLatencyMeasurer {
m := &exampleLatencyMeasurer{}
clk.TickerFunc(ctx, 10*time.Second, func() error {
start := clk.Now()
// m.doSomething()
latency := clk.Since(start)
m.mu.Lock()
defer m.mu.Unlock()
m.lastLatency = latency
return nil
})
return m
}
func (m *exampleLatencyMeasurer) LastLatency() time.Duration {
m.mu.Lock()
defer m.mu.Unlock()
return m.lastLatency
}
func TestExampleLatencyMeasurer(t *testing.T) {
t.Parallel()
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
mClock := quartz.NewMock(t)
trap := mClock.Trap().Since()
defer trap.Close()
lm := newExampleLatencyMeasurer(ctx, mClock)
w := mClock.Advance(10 * time.Second) // triggers first tick
c := trap.MustWait(ctx) // call to Since()
mClock.Advance(33 * time.Millisecond)
c.Release()
w.MustWait(ctx)
if l := lm.LastLatency(); l != 33*time.Millisecond {
t.Fatalf("expected 33ms got %s", l.String())
}
// Next tick is in 10s - 33ms, but if we don't want to calculate, we can use:
d, w2 := mClock.AdvanceNext()
c = trap.MustWait(ctx)
mClock.Advance(17 * time.Millisecond)
c.Release()
w2.MustWait(ctx)
expectedD := 10*time.Second - 33*time.Millisecond
if d != expectedD {
t.Fatalf("expected %s got %s", expectedD.String(), d.String())
}
if l := lm.LastLatency(); l != 17*time.Millisecond {
t.Fatalf("expected 17ms got %s", l.String())
}
}