From ee8f96e7ba4dc5d6bd52372348a5e339a8498627 Mon Sep 17 00:00:00 2001
From: Spike Curtis <spike@coder.com>
Date: Tue, 6 Feb 2024 20:07:42 +0400
Subject: [PATCH] fix: start timer before subscribing to avoid test race

---
 coderd/database/pubsub/watchdog.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/coderd/database/pubsub/watchdog.go b/coderd/database/pubsub/watchdog.go
index 2b8d929fa7e38..687129fc5bcc2 100644
--- a/coderd/database/pubsub/watchdog.go
+++ b/coderd/database/pubsub/watchdog.go
@@ -103,6 +103,8 @@ func (w *Watchdog) publishLoop() {
 
 func (w *Watchdog) subscribeMonitor() {
 	defer w.wg.Done()
+	tmr := w.clock.Timer(periodTimeout)
+	defer tmr.Stop()
 	beats := make(chan struct{})
 	unsub, err := w.ps.Subscribe(EventPubsubWatchdog, func(context.Context, []byte) {
 		w.logger.Debug(w.ctx, "got heartbeat for pubsub watchdog")
@@ -117,8 +119,6 @@ func (w *Watchdog) subscribeMonitor() {
 		return
 	}
 	defer unsub()
-	tmr := w.clock.Timer(periodTimeout)
-	defer tmr.Stop()
 	for {
 		select {
 		case <-w.ctx.Done():