From a0997fb8e5893fed2c2275ff0cfbfa892b261601 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Wed, 23 Aug 2017 20:12:10 +0200
Subject: Make Wrappers modify.

---
 src/mailman_pgp/mta/tests/test_personalized.py | 20 ++++++++------------
 1 file changed, 8 insertions(+), 12 deletions(-)

(limited to 'src/mailman_pgp/mta/tests/test_personalized.py')

diff --git a/src/mailman_pgp/mta/tests/test_personalized.py b/src/mailman_pgp/mta/tests/test_personalized.py
index 7f4caa8..953ef55 100644
--- a/src/mailman_pgp/mta/tests/test_personalized.py
+++ b/src/mailman_pgp/mta/tests/test_personalized.py
@@ -89,13 +89,11 @@ Subject: test
         out_wrapped = PGPWrapper(out_msg)
         self.assertTrue(out_wrapped.is_encrypted())
 
-        decrypted = out_wrapped.decrypt(self.list_key)
-        wrapped = PGPWrapper(decrypted)
-        self.assertTrue(wrapped.is_signed())
+        decrypted = out_wrapped.copy().decrypt(self.list_key)
+        self.assertTrue(decrypted.is_signed())
 
-        decrypted = out_wrapped.decrypt(self.anne_key)
-        wrapped = PGPWrapper(decrypted)
-        self.assertTrue(wrapped.is_signed())
+        decrypted = out_wrapped.copy().decrypt(self.anne_key)
+        self.assertTrue(decrypted.is_signed())
 
     def test_encrypt(self):
         with transaction():
@@ -113,13 +111,11 @@ Subject: test
         out_wrapped = PGPWrapper(out_msg)
         self.assertTrue(out_wrapped.is_encrypted())
 
-        decrypted = out_wrapped.decrypt(self.list_key)
-        wrapped = PGPWrapper(decrypted)
-        self.assertFalse(wrapped.is_signed())
+        decrypted = out_wrapped.copy().decrypt(self.list_key)
+        self.assertFalse(decrypted.is_signed())
 
-        decrypted = out_wrapped.decrypt(self.anne_key)
-        wrapped = PGPWrapper(decrypted)
-        self.assertFalse(wrapped.is_signed())
+        decrypted = out_wrapped.copy().decrypt(self.anne_key)
+        self.assertFalse(decrypted.is_signed())
 
     def test_sign(self):
         with transaction():
-- 
cgit v1.2.3-70-g09d2