fixup! chore: refactor agent stats streaming

coadler · coadler · commit aaef131e7239 · 2022-11-17T19:10:29.000-06:00
diff --git a/codersdk/workspaceagents_test.go b/codersdk/workspaceagents_test.go
@@ -6,6 +6,7 @@ import (
 	"net/http/httptest"
 	"net/url"
 	"strconv"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -54,9 +55,9 @@ func TestWorkspaceAgentMetadata(t *testing.T) {
 func TestAgentReportStats(t *testing.T) {
 	t.Parallel()
 
-	var numReports int
+	var numReports atomic.Int64
 	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		numReports++
+		numReports.Add(1)
 		httpapi.Write(context.Background(), w, http.StatusOK, codersdk.AgentStatsResponse{
 			ReportInterval: 5 * time.Millisecond,
 		})
@@ -73,7 +74,7 @@ func TestAgentReportStats(t *testing.T) {
 	defer closeStream.Close()
 
 	require.Eventually(t,
-		func() bool { return numReports >= 3 },
+		func() bool { return numReports.Load() >= 3 },
 		testutil.WaitMedium, testutil.IntervalFast,
 	)
 }
