use BlockUntilContext

Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
This commit is contained in:
Nicolas De Loof 2025-02-10 15:09:18 +01:00 committed by Nicolas De loof
parent fa4cfb652e
commit ed72c21871
2 changed files with 8 additions and 4 deletions

View File

@ -147,7 +147,8 @@ func TestWatch_Sync(t *testing.T) {
watcher.Events() <- watch.NewFileEvent("/sync/changed")
watcher.Events() <- watch.NewFileEvent("/sync/changed/sub")
clock.BlockUntil(3)
err := clock.BlockUntilContext(ctx, 3)
assert.NilError(t, err)
clock.Advance(watch.QuietPeriod)
select {
case actual := <-syncer.synced:
@ -161,7 +162,8 @@ func TestWatch_Sync(t *testing.T) {
watcher.Events() <- watch.NewFileEvent("/rebuild")
watcher.Events() <- watch.NewFileEvent("/sync/changed")
clock.BlockUntil(4)
err = clock.BlockUntilContext(ctx, 4)
assert.NilError(t, err)
clock.Advance(watch.QuietPeriod)
select {
case batch := <-syncer.synced:

View File

@ -40,7 +40,8 @@ func Test_BatchDebounceEvents(t *testing.T) {
ch <- FileEvent(path)
}
// we sent 100 events + the debouncer
clock.BlockUntil(101)
err := clock.BlockUntilContext(ctx, 101)
assert.NilError(t, err)
clock.Advance(QuietPeriod)
select {
case batch := <-eventBatchCh:
@ -51,7 +52,8 @@ func Test_BatchDebounceEvents(t *testing.T) {
case <-time.After(50 * time.Millisecond):
t.Fatal("timed out waiting for events")
}
clock.BlockUntil(1)
err = clock.BlockUntilContext(ctx, 1)
assert.NilError(t, err)
clock.Advance(QuietPeriod)
// there should only be a single batch