diff --git a/codersdk/workspacesdk/agentconn_test.go b/codersdk/workspacesdk/agentconn_test.go index c77cbae3ce..a2cce9d5cd 100644 --- a/codersdk/workspacesdk/agentconn_test.go +++ b/codersdk/workspacesdk/agentconn_test.go @@ -243,7 +243,9 @@ func stitchTailnet(t *testing.T, conns map[uuid.UUID]*tailnet.Conn) { Node: protoNode, Kind: proto.CoordinateResponse_PeerUpdate_NODE, }}) - assert.NoError(t, err) + if err != nil && !errors.Is(err, tailnet.ErrConnClosed) { + assert.NoError(t, err) + } } } diff --git a/tailnet/conn_test.go b/tailnet/conn_test.go index 7e8ac53474..b2c9cd575e 100644 --- a/tailnet/conn_test.go +++ b/tailnet/conn_test.go @@ -2,6 +2,7 @@ package tailnet_test import ( "context" + "errors" "net" "net/netip" "strings" @@ -483,7 +484,9 @@ func stitch(t *testing.T, dst, src *tailnet.Conn) { Node: pn, Kind: proto.CoordinateResponse_PeerUpdate_NODE, }}) - assert.NoError(t, err) + if err != nil && !errors.Is(err, tailnet.ErrConnClosed) { + assert.NoError(t, err) + } }) // ensures we don't send callbacks after the test ends and connections are closed. t.Cleanup(func() {