diff --git a/api/backend_test.go b/api/backend_test.go index 8603b0912..88d92faea 100644 --- a/api/backend_test.go +++ b/api/backend_test.go @@ -842,7 +842,7 @@ func TestLoginAccount(t *testing.T) { c <- struct{}{} } }) - defer signal.SetMobileSignalHandler(nil) + t.Cleanup(signal.ResetMobileSignalHandler) waitForLogin := func(chan interface{}) { select { case <-c: @@ -1403,6 +1403,7 @@ func TestCreateWallet(t *testing.T) { c <- struct{}{} } }) + t.Cleanup(signal.ResetMobileSignalHandler) account, err := b.CreateAccountAndLogin(createAccountRequest) require.NoError(t, err) @@ -1463,6 +1464,7 @@ func TestSetFleet(t *testing.T) { c <- struct{}{} } }) + t.Cleanup(signal.ResetMobileSignalHandler) newAccount, err := b.CreateAccountAndLogin(createAccountRequest) require.NoError(t, err) @@ -1531,6 +1533,7 @@ func TestWalletConfigOnLoginAccount(t *testing.T) { c <- struct{}{} } }) + t.Cleanup(signal.ResetMobileSignalHandler) newAccount, err := b.CreateAccountAndLogin(createAccountRequest) require.NoError(t, err) diff --git a/services/connector/commands/personal_sign_test.go b/services/connector/commands/personal_sign_test.go index e983f7fb5..7a9b46985 100644 --- a/services/connector/commands/personal_sign_test.go +++ b/services/connector/commands/personal_sign_test.go @@ -135,6 +135,7 @@ func TestPersonalSignWithSignalAccepted(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) response, err := cmd.Execute(request) assert.NoError(t, err) @@ -177,6 +178,7 @@ func TestPersonalSignWithSignalRejected(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) _, err = cmd.Execute(request) assert.Equal(t, ErrPersonalSignRejectedByUser, err) diff --git a/services/connector/commands/request_accounts_test.go b/services/connector/commands/request_accounts_test.go index ada6d2400..b68899275 100644 --- a/services/connector/commands/request_accounts_test.go +++ b/services/connector/commands/request_accounts_test.go @@ -88,6 +88,7 @@ func TestRequestAccountsAcceptedAndRequestAgain(t *testing.T) { dAppPermissionGranted = true } })) + t.Cleanup(signal.ResetMobileSignalHandler) expectedResponse := FormatAccountAddressToResponse(accountAddress) response, err := cmd.Execute(request) @@ -142,6 +143,7 @@ func TestRequestAccountsRejected(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) _, err = cmd.Execute(request) assert.Equal(t, ErrRequestAccountsRejectedByUser, err) diff --git a/services/connector/commands/revoke_permissions_test.go b/services/connector/commands/revoke_permissions_test.go index 74fc4b142..4613e5e3a 100644 --- a/services/connector/commands/revoke_permissions_test.go +++ b/services/connector/commands/revoke_permissions_test.go @@ -56,6 +56,7 @@ func TestRevokePermissionsSucceeded(t *testing.T) { dAppPermissionRevoked = true } })) + t.Cleanup(signal.ResetMobileSignalHandler) err := PersistDAppData(db, testDAppData, sharedAccount, 0x123) assert.NoError(t, err) diff --git a/services/connector/commands/send_transaction_test.go b/services/connector/commands/send_transaction_test.go index ad7201c95..b4349ba67 100644 --- a/services/connector/commands/send_transaction_test.go +++ b/services/connector/commands/send_transaction_test.go @@ -129,6 +129,7 @@ func TestSendTransactionWithSignalAccepted(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) response, err := cmd.Execute(request) assert.NoError(t, err) @@ -169,6 +170,7 @@ func TestSendTransactionWithSignalRejected(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) _, err = cmd.Execute(request) assert.Equal(t, ErrSendTransactionRejectedByUser, err) diff --git a/services/connector/commands/switch_ethereum_chain_test.go b/services/connector/commands/switch_ethereum_chain_test.go index 774b1d13b..3d0f46446 100644 --- a/services/connector/commands/switch_ethereum_chain_test.go +++ b/services/connector/commands/switch_ethereum_chain_test.go @@ -102,6 +102,7 @@ func TestSwitchEthereumChainSuccess(t *testing.T) { chainIdSwitched = true } })) + t.Cleanup(signal.ResetMobileSignalHandler) cmd := &SwitchEthereumChainCommand{ Db: db, diff --git a/services/connector/connector_flows_test.go b/services/connector/connector_flows_test.go index d5905c449..e2df94619 100644 --- a/services/connector/connector_flows_test.go +++ b/services/connector/connector_flows_test.go @@ -86,6 +86,7 @@ func TestRequestAccountsSwitchChainAndSendTransactionFlow(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) // Request accounts, now for real request := "{\"method\": \"eth_requestAccounts\", \"params\": [], \"url\": \"http://testDAppURL123\", \"name\": \"testDAppName\", \"iconUrl\": \"http://testDAppIconUrl\" }" @@ -221,6 +222,7 @@ func TestRequestAccountsAfterPermisasionsRevokeTest(t *testing.T) { assert.NoError(t, err) } })) + t.Cleanup(signal.ResetMobileSignalHandler) for range [10]int{} { dAppPermissionRevoked = false diff --git a/services/local-notifications/core_test.go b/services/local-notifications/core_test.go index ae1a74808..baaeddf9a 100644 --- a/services/local-notifications/core_test.go +++ b/services/local-notifications/core_test.go @@ -91,6 +91,7 @@ func TestTransactionNotification(t *testing.T) { }) signal.SetMobileSignalHandler(signalHandler) + t.Cleanup(signal.ResetMobileSignalHandler) feed := &event.Feed{} require.NoError(t, s.SubscribeWallet(feed)) diff --git a/services/wallet/router/router_v2_test.go b/services/wallet/router/router_v2_test.go index 170d84eca..ae1e76a5d 100644 --- a/services/wallet/router/router_v2_test.go +++ b/services/wallet/router/router_v2_test.go @@ -144,10 +144,10 @@ func setupSignalHandler(t *testing.T) (chan SuggestedRoutesV2Response, func()) { } }) signal.SetMobileSignalHandler(signalHandler) + t.Cleanup(signal.ResetMobileSignalHandler) closeFn := func() { close(suggestedRoutesCh) - signal.SetMobileSignalHandler(nil) } return suggestedRoutesCh, closeFn diff --git a/signal/signals.go b/signal/signals.go index 81aed7ab4..c6f37c049 100644 --- a/signal/signals.go +++ b/signal/signals.go @@ -111,6 +111,10 @@ func SetMobileSignalHandler(handler MobileSignalHandler) { mobileSignalHandler = handler } +func ResetMobileSignalHandler() { + mobileSignalHandler = nil +} + // SetSignalEventCallback set callback // this function uses C implementation (see `signals.c` file) func SetSignalEventCallback(cb unsafe.Pointer) {