From 3f55a532558eb17b272b69d2e6c91b2f9af9b38f Mon Sep 17 00:00:00 2001 From: Michael Greene Date: Wed, 23 Nov 2022 16:23:15 -0800 Subject: [PATCH] fix this test --- tests/test_10_exceptions.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/tests/test_10_exceptions.py b/tests/test_10_exceptions.py index 043f711..6307671 100644 --- a/tests/test_10_exceptions.py +++ b/tests/test_10_exceptions.py @@ -230,13 +230,9 @@ class TestPGPKey(object): rsa_pub.subkeys['EEE097A017B979CA'].encrypt(PGPMessage.new('asdf'), cipher=SymmetricKeyAlgorithm.CAST5) - w = recwarn.pop(UserWarning) - assert str(w.message) == "Selected symmetric algorithm not in key preferences" - assert w.filename == __file__ - - w = recwarn.pop(UserWarning) - assert str(w.message) == "Selected compression algorithm not in key preferences" - assert w.filename == __file__ + relevant_warning_messages = [ str(w.message) for w in recwarn if w.category is UserWarning ] + assert "Selected symmetric algorithm not in key preferences" in relevant_warning_messages + assert "Selected compression algorithm not in key preferences" in relevant_warning_messages def test_sign_bad_prefs(self, rsa_sec, recwarn): rsa_sec.subkeys['2A834D8E5918E886'].sign(PGPMessage.new('asdf'), hash=HashAlgorithm.MD5)