tests: Fix exception type not handled

LedgerHQ · Dec 7, 2023 · cabd753 · cabd753
1 parent 7de5ad7
commit cabd753
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/speculos/u2f/test_u2f_fake_channel_security_crc.py b/tests/speculos/u2f/test_u2f_fake_channel_security_crc.py
@@ -1,4 +1,5 @@
 import pytest
+import socket
 
 from fido2.ctap1 import ApduError, Ctap1
 from fido2.hid import CTAPHID
@@ -43,5 +44,5 @@ def test_register_raw_u2f_fake_channel_security_crc(client):
                                   p2=0x00,
                                   data=data)
 
-    with pytest.raises((AssertionError, ConnectionResetError, TimeoutError)) as e:
+    with pytest.raises((AssertionError, ConnectionResetError, TimeoutError, socket.timeout)) as e:
         response = client.ctap1.device.recv(CTAPHID.MSG)
diff --git a/tests/speculos/u2f/test_u2f_fake_channel_security_length.py b/tests/speculos/u2f/test_u2f_fake_channel_security_length.py
@@ -1,4 +1,5 @@
 import pytest
+import socket
 
 from fido2.ctap1 import ApduError, Ctap1
 from fido2.hid import CTAPHID
@@ -44,5 +45,5 @@ def test_register_raw_u2f_fake_channel_security_length(client):
                                   p2=0x00,
                                   data=data)
 
-    with pytest.raises((AssertionError, ConnectionResetError, TimeoutError)) as e:
+    with pytest.raises((AssertionError, ConnectionResetError, TimeoutError, socket.timeout)) as e:
         response = client.ctap1.device.recv(CTAPHID.MSG)