From 9f9998ab4bd5f0bc385b8ff2ffbd2bbd7acd87c8 Mon Sep 17 00:00:00 2001
From: Shogo Hyodo <mmoshg8u@gmail.com>
Date: Tue, 13 Dec 2022 15:33:35 +0900
Subject: [PATCH] Use `defer`

---
 p2p/transport_test.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/p2p/transport_test.go b/p2p/transport_test.go
index f448f6ba9..58555a671 100644
--- 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