From d8937a61da12d933df4816febe437b070c826630 Mon Sep 17 00:00:00 2001 From: Viktor Liu Date: Thu, 20 Aug 2026 14:19:47 +0200 Subject: [PATCH] Release the activation state test's resources with t.Cleanup --- client/internal/engine_test.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/client/internal/engine_test.go b/client/internal/engine_test.go index 290a8e3c7..b55819e99 100644 --- a/client/internal/engine_test.go +++ b/client/internal/engine_test.go @@ -478,7 +478,7 @@ func TestEngine_ModifiedPeerKeepsActivationState(t *testing.T) { require.NoError(t, err) ctx, cancel := context.WithCancel(CtxInitState(context.Background())) - defer cancel() + t.Cleanup(cancel) relayMgr := relayClient.NewManager(ctx, nil, key.PublicKey().String(), iface.DefaultMTU) engine := NewEngine(ctx, cancel, &EngineConfig{ @@ -528,11 +528,13 @@ func TestEngine_ModifiedPeerKeepsActivationState(t *testing.T) { } udpConn, err := net.ListenUDP("udp4", nil) require.NoError(t, err) + t.Cleanup(func() { _ = udpConn.Close() }) engine.udpMux = udpmux.NewUniversalUDPMuxDefault(udpmux.UniversalUDPMuxParams{UDPConn: udpConn, MTU: 1280}) engine.ctx = ctx engine.srWatcher = guard.NewSRWatcher(nil, nil, nil, icemaker.Config{}) engine.connMgr = NewConnMgr(engine.config, engine.statusRecorder, engine.peerStore, wgIface) engine.connMgr.Start(ctx) + t.Cleanup(engine.connMgr.Close) // No agent version: not lazy-capable, so the connection opens permanently. activePeer := &mgmtProto.RemotePeerConfig{