test_wrong_certificate could also raise AssertionError

MSLNZ · Sep 20, 2023 · b650cfb · b650cfb
1 parent 5ec2f46
commit b650cfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -98,7 +98,7 @@ def test_wrong_certificate():
     assert cryptography.generate_certificate(path=cert, key_path=key) == cert
     kwargs = manager.kwargs.copy()
     kwargs['cert_file'] = cert
-    with pytest.raises(ConnectionError) as e:
+    with pytest.raises(ConnectionError, AssertionError) as e:
         connect(**kwargs)
     msg = str(e.value)
     assert 'Perhaps the Network Manager is using a new certificate' in msg