Use Until instead of Sub in WithDeadline

benbjohnson · Nov 12, 2021 · 3e4357c · 3e4357c
1 parent 6754fcb
commit 3e4357c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/context.go b/context.go
@@ -18,7 +18,7 @@ func (m *Mock) WithDeadline(parent context.Context, deadline time.Time) (context
 	}
 	ctx := &timerCtx{clock: m, parent: parent, deadline: deadline, done: make(chan struct{})}
 	propagateCancel(parent, ctx)
-	dur := deadline.Sub(m.Now())
+	dur := m.Until(deadline)
 	if dur <= 0 {
 		ctx.cancel(context.DeadlineExceeded) // deadline has already passed
 		return ctx, func() { ctx.cancel(context.Canceled) }