Fix tests

OriginProtocol · Apr 9, 2024 · 4f41d63 · 4f41d63
1 parent 706b5dd
commit 4f41d63
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/tests/utils.py b/tests/utils.py
@@ -58,12 +58,10 @@ def run_test(contract, data, backend, firmware, navigator, test_name, value=0, g
     client = EthAppClient(backend)
 
     # first setup the external plugin
-    with client.set_external_plugin(PLUGIN_NAME,
-                                    contract.address,
-                                    # Extract function selector from the encoded data
-                                    get_selector_from_data(data)):
-        pass
-
+    client.set_external_plugin(PLUGIN_NAME,
+                               contract.address,
+                               # Extract function selector from the encoded data
+                               get_selector_from_data(data))
     # send the transaction
     with client.sign("m/44'/60'/1'/0/0", {
              "nonce": 20,
@@ -74,7 +72,7 @@ def run_test(contract, data, backend, firmware, navigator, test_name, value=0, g
              "value": value,
              "chainId": ChainId.ETH,
              "data": data
-         }):
+        }):
         # Validate the on-screen request by performing the navigation appropriate for this device
         if firmware.device.startswith("nano"):
             navigator.navigate_until_text_and_compare(NavInsID.RIGHT_CLICK,