Skip to content

Commit 9757433

Browse files
committed
refactor(tests): improve client closure handling in DHT tests
1 parent 2755235 commit 9757433

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

dht_config_test.go

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ func TestDHTModeDefault(t *testing.T) {
3030
client, err := NewClient(config)
3131
require.NoError(t, err)
3232
require.NotNil(t, client)
33-
defer client.Close()
33+
defer func() { _ = client.Close() }()
3434

3535
// Verify server mode was selected
3636
output := buf.String()
@@ -57,7 +57,7 @@ func TestDHTModeServer(t *testing.T) {
5757
client, err := NewClient(config)
5858
require.NoError(t, err)
5959
require.NotNil(t, client)
60-
defer client.Close()
60+
defer func() { _ = client.Close() }()
6161

6262
// Verify server mode was selected
6363
output := buf.String()
@@ -83,7 +83,7 @@ func TestDHTModeClient(t *testing.T) {
8383
client, err := NewClient(config)
8484
require.NoError(t, err)
8585
require.NotNil(t, client)
86-
defer client.Close()
86+
defer func() { _ = client.Close() }()
8787

8888
// Verify client mode was selected
8989
output := buf.String()
@@ -111,7 +111,7 @@ func TestDHTCleanupIntervalDefault(t *testing.T) {
111111
client, err := NewClient(config)
112112
require.NoError(t, err)
113113
require.NotNil(t, client)
114-
defer client.Close()
114+
defer func() { _ = client.Close() }()
115115

116116
// Verify cleanup interval configuration is NOT logged (uses libp2p default)
117117
output := buf.String()
@@ -139,7 +139,7 @@ func TestDHTCleanupIntervalCustom(t *testing.T) {
139139
client, err := NewClient(config)
140140
require.NoError(t, err)
141141
require.NotNil(t, client)
142-
defer client.Close()
142+
defer func() { _ = client.Close() }()
143143

144144
// Verify cleanup interval configuration was logged
145145
output := buf.String()
@@ -166,7 +166,7 @@ func TestDHTCleanupIntervalIgnoredInClientMode(t *testing.T) {
166166
client, err := NewClient(config)
167167
require.NoError(t, err)
168168
require.NotNil(t, client)
169-
defer client.Close()
169+
defer func() { _ = client.Close() }()
170170

171171
// Verify client mode was selected
172172
output := buf.String()
@@ -225,7 +225,7 @@ func TestDHTModeServerWithVariousCleanupIntervals(t *testing.T) {
225225
client, err := NewClient(config)
226226
require.NoError(t, err)
227227
require.NotNil(t, client)
228-
defer client.Close()
228+
defer func() { _ = client.Close() }()
229229

230230
// Verify cleanup interval was configured correctly
231231
output := buf.String()
@@ -301,7 +301,7 @@ func TestDHTConfigurationCombinations(t *testing.T) {
301301
client, err := NewClient(config)
302302
require.NoError(t, err, "client creation should succeed")
303303
require.NotNil(t, client, "client should not be nil")
304-
defer client.Close()
304+
defer func() { _ = client.Close() }()
305305

306306
output := buf.String()
307307

@@ -344,7 +344,7 @@ func TestDHTClientCanQuery(t *testing.T) {
344344
client, err := NewClient(config)
345345
require.NoError(t, err)
346346
require.NotNil(t, client)
347-
defer client.Close()
347+
defer func() { _ = client.Close() }()
348348

349349
// Verify client mode was selected (basic functional check)
350350
output := buf.String()
@@ -370,10 +370,10 @@ func TestP2PClientTypeAlias(t *testing.T) {
370370
p2pClient, err = NewClient(config)
371371
require.NoError(t, err)
372372
require.NotNil(t, p2pClient)
373-
defer p2pClient.Close()
373+
defer func() { _ = p2pClient.Close() }()
374374

375375
// Verify it's also a Client
376-
var client Client = p2pClient
376+
client := p2pClient
377377
assert.NotNil(t, client)
378378
}
379379

@@ -391,7 +391,7 @@ func BenchmarkDHTServerMode(b *testing.B) {
391391
for i := 0; i < b.N; i++ {
392392
client, _ := NewClient(config)
393393
if client != nil {
394-
client.Close()
394+
_ = client.Close()
395395
}
396396
}
397397
}
@@ -409,7 +409,7 @@ func BenchmarkDHTClientMode(b *testing.B) {
409409
for i := 0; i < b.N; i++ {
410410
client, _ := NewClient(config)
411411
if client != nil {
412-
client.Close()
412+
_ = client.Close()
413413
}
414414
}
415415
}

0 commit comments

Comments
 (0)