Use defer

Finschia · Dec 13, 2022 · 9f9998a · 9f9998a
1 parent fb950c3
commit 9f9998a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/p2p/transport_test.go b/p2p/transport_test.go
@@ -643,6 +643,10 @@ func TestTransportAddChannel(t *testing.T) {
 
 func TestTransportResolveIPs(t *testing.T) {
 	server := startFakeDNS()
+	defer func() {
+		err := server.Shutdown()
+		require.NoError(t, err)
+	}()
 
 	r := &net.Resolver{
 		PreferGo: true,
@@ -655,9 +659,6 @@ func TestTransportResolveIPs(t *testing.T) {
 	}
 	_, err := resolveIPs(r, &testTransportConn{})
 	require.Contains(t, err.Error(), "lookup test.local: i/o timeout")
-
-	err = server.Shutdown()
-	require.NoError(t, err)
 }
 
 // create listener