From c0ae8b109a21fad1c2ff56db0768c16b78f6a5f7 Mon Sep 17 00:00:00 2001 From: Ruben Pollan Date: Wed, 6 Jul 2016 15:59:10 +0200 Subject: [feat] discover encr and sign address --- tests/test_protection.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/test_protection.py b/tests/test_protection.py index f83d3f7..59effaa 100644 --- a/tests/test_protection.py +++ b/tests/test_protection.py @@ -1,4 +1,3 @@ -import base64 import unittest from email.parser import Parser from zope.interface import implementer @@ -32,8 +31,20 @@ class ProtectTest(unittest.TestCase): self.assertEqual(encmsg.get_payload(1).get_payload(), encrypter.encstr) self.assertEqual(BODY, encrypter.data[1:-1]) # remove '\n' + self.assertEqual([TO], encrypter.encraddr) + self.assertEqual(FROM, encrypter.singaddr) self.assertEqual(encmsg.get_content_type(), "multipart/encrypted") + def test_unobscured_headers(self): + p = Parser() + msg = p.parsestr(EMAIL) + encrypter = Encrypter() + encmsg = protect(msg, encrypter, obscure=False) + + self.assertEqual(encmsg['from'], FROM) + self.assertEqual(encmsg['to'], TO) + self.assertEqual(encmsg['subject'], SUBJECT) + @implementer(OpenPGP) class Encrypter(object): @@ -41,6 +52,8 @@ class Encrypter(object): def encrypt(self, data, encraddr, singaddr): self.data = data + self.encraddr = encraddr + self.singaddr = singaddr return self.encstr -- cgit v1.2.3