tor: replace defer cleanup with t.Cleanup

Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
This commit is contained in:
Eng Zer Jun 2022-08-27 15:07:51 +08:00
parent ff3f477ebf
commit 22cd790586
No known key found for this signature in database
GPG Key ID: DAEBBD2E34C111E6
2 changed files with 6 additions and 6 deletions

View File

@ -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.

View File

@ -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{