tor: replace defer cleanup with t.Cleanup

Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>

tor/cmd_info_test.go

@@ -25,7 +25,7 @@ func TestCheckOnionServiceSucceed(t *testing.T) {
 
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 	server := proxy.serverConn
 
 	// Assign a fake service ID to the controller.
@@ -47,7 +47,7 @@ func TestCheckOnionServiceFailOnServiceIDNotMatch(t *testing.T) {
 
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 	server := proxy.serverConn
 
 	// Assign a fake service ID to the controller.
@@ -69,7 +69,7 @@ func TestCheckOnionServiceSucceedOnMultipleServices(t *testing.T) {
 
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 	server := proxy.serverConn
 
 	// Assign a fake service ID to the controller.
@@ -92,7 +92,7 @@ func TestCheckOnionServiceFailOnClosedConnection(t *testing.T) {
 
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 	server := proxy.serverConn
 
 	// Assign a fake service ID to the controller.

tor/controller_test.go

@@ -153,7 +153,7 @@ func createTestProxy(t *testing.T) *testProxy {
 func TestReadResponse(t *testing.T) {
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 	server := proxy.serverConn
 
 	// Create a dummy tor controller.
@@ -300,7 +300,7 @@ func TestReconnectTCMustBeRunning(t *testing.T) {
 func TestReconnectSucceed(t *testing.T) {
 	// Create mock server and client connection.
 	proxy := createTestProxy(t)
-	defer proxy.cleanUp()
+	t.Cleanup(proxy.cleanUp)
 
 	// Create a tor controller and mark the controller as started.
 	c := &Controller{