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