@@ -13,6 +13,11 @@ import (
1313"strings"
1414"testing"
1515
16+ "github.com/go-chi/chi/v5"
17+ "github.com/google/uuid"
18+ "github.com/stretchr/testify/require"
19+ "go.uber.org/mock/gomock"
20+
1621"cdr.dev/slog"
1722"cdr.dev/slog/sloggers/slogtest"
1823"github.com/coder/coder/v2/coderd/database"
@@ -28,10 +33,6 @@ import (
2833"github.com/coder/coder/v2/tailnet/tailnettest"
2934"github.com/coder/coder/v2/testutil"
3035"github.com/coder/websocket"
31- "github.com/go-chi/chi/v5"
32- "github.com/google/uuid"
33- "github.com/stretchr/testify/require"
34- "go.uber.org/mock/gomock"
3536)
3637
3738type fakeAgentProvider struct {
@@ -143,8 +144,11 @@ func TestWatchAgentContainers(t *testing.T) {
143144
144145// And: Dial the WebSocket
145146wsURL := strings .Replace (srv .URL ,"http://" ,"ws://" ,1 )
146- conn ,_ ,err := websocket .Dial (ctx ,fmt .Sprintf ("%s/workspaceagents/%s/containers/watch" ,wsURL ,agentID ),nil )
147+ conn ,resp ,err := websocket .Dial (ctx ,fmt .Sprintf ("%s/workspaceagents/%s/containers/watch" ,wsURL ,agentID ),nil )
147148require .NoError (t ,err )
149+ if resp .Body != nil {
150+ defer resp .Body .Close ()
151+ }
148152
149153// And: Create a streaming decoder
150154decoder := wsjson .NewDecoder [codersdk.WorkspaceAgentListContainersResponse ](conn ,websocket .MessageText ,logger )