Commit d2e12dfe authored by Jérome Perrin's avatar Jérome Perrin Committed by Arnaud Fontaine

py3: fix return type of DummyMailHostMixin.getMessageList

mail body is str
parent 0675f7f4
...@@ -125,7 +125,7 @@ class TestPasswordTool(ERP5TypeTestCase): ...@@ -125,7 +125,7 @@ class TestPasswordTool(ERP5TypeTestCase):
self.assertEqual(['userA@example.invalid'], mto) self.assertEqual(['userA@example.invalid'], mto)
reset_key, = list(six.iterkeys(self.portal.portal_password._password_request_dict)) reset_key, = list(six.iterkeys(self.portal.portal_password._password_request_dict))
self.assertIn( self.assertIn(
('PasswordTool_viewResetPassword?reset_key=' + reset_key).encode(), ('PasswordTool_viewResetPassword?reset_key=' + reset_key),
mbody) mbody)
ret = self.portal.portal_password.changeUserPassword( ret = self.portal.portal_password.changeUserPassword(
......
import re import re
from Products.ERP5Type.Utils import bytes2str last_message_text = context.getMessageList()[-1][2]
last_message_text = bytes2str(context.getMessageList()[-1][2])
__traceback_info__ = last_message_text
return container.REQUEST.RESPONSE.redirect(re.findall(r"http.*", last_message_text)[0]) return container.REQUEST.RESPONSE.redirect(re.findall(r"http.*", last_message_text)[0])
...@@ -30,6 +30,7 @@ ...@@ -30,6 +30,7 @@
""" """
import contextlib import contextlib
from datetime import datetime from datetime import datetime
import email
import errno import errno
import os import os
import logging import logging
...@@ -49,7 +50,6 @@ from Zope2.Startup.datatypes import ZopeDatabase ...@@ -49,7 +50,6 @@ from Zope2.Startup.datatypes import ZopeDatabase
from Testing import ZopeTestCase from Testing import ZopeTestCase
import Products.ERP5Type import Products.ERP5Type
from Products.MailHost.MailHost import MailHost from Products.MailHost.MailHost import MailHost
from email import message_from_string
from Products.ERP5Type.Globals import PersistentMapping from Products.ERP5Type.Globals import PersistentMapping
from Products.ERP5Type.Utils import simple_decorator from Products.ERP5Type.Utils import simple_decorator
from Products.ZSQLCatalog.SQLCatalog import Catalog from Products.ZSQLCatalog.SQLCatalog import Catalog
...@@ -106,14 +106,19 @@ class DummyMailHostMixin(object): ...@@ -106,14 +106,19 @@ class DummyMailHostMixin(object):
@staticmethod @staticmethod
def _decodeMessage(messageText): def _decodeMessage(messageText):
# type: (bytes) -> str
""" Decode message""" """ Decode message"""
if six.PY3 and isinstance(messageText, bytes): if six.PY3:
messageText = messageText.decode() message = email.message_from_bytes(messageText)
else:
message = email.message_from_string(messageText)
message_text = messageText message_text = messageText
for part in message_from_string(messageText).walk(): for part in message.walk():
if part.get_content_type() in ['text/plain', 'text/html' ] \ if part.get_content_type() in ['text/plain', 'text/html' ] \
and not part.is_multipart(): and not part.is_multipart():
message_text = part.get_payload(decode=1) message_text = part.get_payload(decode=True)
if six.PY3:
message_text = message_text.decode(part.get_content_charset('ascii'))
return message_text return message_text
security.declarePrivate('getMessageList') security.declarePrivate('getMessageList')
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment