diff --git a/tests/test_form_pages.py b/tests/test_form_pages.py index 71c4aa1a2..c65065ebf 100644 --- a/tests/test_form_pages.py +++ b/tests/test_form_pages.py @@ -2837,7 +2837,7 @@ def test_form_table_field_submit(pub, emails): resp = resp.follow() assert 'The form has been recorded' in resp.text # check rst2html didn't fail - assert 'ee' in emails.emails['New form (test)']['msg'].get_payload()[1].get_payload() + assert b'ee' in emails.emails['New form (test)']['msg'].get_payload()[1].get_payload(decode=True) def test_form_table_rows_field_submit(pub, emails): formdef = create_formdef() @@ -2903,7 +2903,7 @@ def test_form_table_rows_field_submit(pub, emails): resp = resp.form.submit('submit') resp = resp.follow() assert 'The form has been recorded' in resp.text - assert 'ee' in emails.emails['New form (test)']['msg'].get_payload()[1].get_payload() + assert b'ee' in emails.emails['New form (test)']['msg'].get_payload()[1].get_payload(decode=True) def test_form_new_table_rows_field_draft_recall(pub): formdef = create_formdef() @@ -5908,7 +5908,7 @@ def test_email_actions(pub, emails): assert '/actions/' in action_url if docutils: assert len(email_data['payloads']) == 2 - assert action_url in email_data['payloads'][1] + assert action_url in force_text(email_data['payloads'][1]) app = get_app(pub) resp = app.get(action_url) diff --git a/tests/test_misc.py b/tests/test_misc.py index 3f8968f85..43eab9102 100644 --- a/tests/test_misc.py +++ b/tests/test_misc.py @@ -283,7 +283,7 @@ def test_email_signature_plain(emails): send_email('test', mail_body='Hello', email_rcpt='test@localhost', want_html=False) assert emails.count() == 1 assert not emails.emails['test']['msg'].is_multipart() - assert 'Footer\nText' in emails.emails['test']['msg'].get_payload() + assert b'Footer\nText' in emails.emails['test']['msg'].get_payload(decode=True) def test_email_from(emails): pub = create_temporary_pub() @@ -303,7 +303,7 @@ def test_email_from(emails): send_email('test', mail_body='Hello', email_rcpt='test@localhost', want_html=False) assert emails.count() == 1 assert emails.emails['test']['from'] == 'foo@localhost' - assert emails.emails['test']['msg']['From'] == '=?utf-8?q?HELLO?= ' + assert emails.emails['test']['msg']['From'] in ('=?utf-8?q?HELLO?= ', 'HELLO ') @pytest.mark.skipif('docutils is None') def test_email_signature_rst(emails): @@ -315,8 +315,8 @@ def test_email_signature_rst(emails): assert emails.emails['test']['msg'].get_content_subtype() == 'alternative' assert emails.emails['test']['msg'].get_payload()[0].get_content_type() == 'text/plain' assert emails.emails['test']['msg'].get_payload()[1].get_content_type() == 'text/html' - assert 'Footer\nText' in emails.emails['test']['msg'].get_payload()[0].get_payload() - assert '>Footer<' in emails.emails['test']['msg'].get_payload()[1].get_payload() + assert b'Footer\nText' in emails.emails['test']['msg'].get_payload()[0].get_payload(decode=True) + assert b'>Footer<' in emails.emails['test']['msg'].get_payload()[1].get_payload(decode=True) @pytest.mark.skipif('docutils is None') def test_email_signature_rst_pipes(emails): @@ -328,8 +328,8 @@ def test_email_signature_rst_pipes(emails): assert emails.emails['test']['msg'].get_content_subtype() == 'alternative' assert emails.emails['test']['msg'].get_payload()[0].get_content_type() == 'text/plain' assert emails.emails['test']['msg'].get_payload()[1].get_content_type() == 'text/html' - assert 'Footer\nText' in emails.emails['test']['msg'].get_payload()[0].get_payload() - assert '>Footer<' in emails.emails['test']['msg'].get_payload()[1].get_payload() + assert b'Footer\nText' in emails.emails['test']['msg'].get_payload()[0].get_payload(decode=True) + assert b'>Footer<' in emails.emails['test']['msg'].get_payload()[1].get_payload(decode=True) def test_email_plain_with_attachments(emails): pub = create_temporary_pub() @@ -360,7 +360,7 @@ def test_email_plain_with_attachments(emails): assert emails.emails['txt']['msg'].get_content_subtype() == 'mixed' assert emails.emails['txt']['msg'].get_payload()[0].get_content_type() == 'text/plain' assert emails.emails['txt']['msg'].get_payload()[1].get_content_type() == 'text/plain' - assert emails.emails['txt']['msg'].get_payload()[1].get_payload() == 'foo-text-bar' + assert emails.emails['txt']['msg'].get_payload()[1].get_payload(decode=True) == b'foo-text-bar' send_email('jpgodt', mail_body='Hello', email_rcpt='test@localhost', want_html=False, @@ -382,7 +382,7 @@ def test_email_plain_with_attachments(emails): assert emails.emails['unknown']['msg'].get_content_subtype() == 'mixed' assert emails.emails['unknown']['msg'].get_payload()[0].get_content_type() == 'text/plain' assert emails.emails['unknown']['msg'].get_payload()[1].get_content_type() == 'x-foo/x-bar' - assert emails.emails['unknown']['msg'].get_payload()[1].get_payload() == base64.b64encode('barfoo') + assert emails.emails['unknown']['msg'].get_payload()[1].get_payload(decode=False).strip() == 'YmFyZm9v' send_email('test-bad-attachment', mail_body='Hello', email_rcpt='test@localhost', want_html=False, @@ -452,12 +452,12 @@ M. Francis Kuntz assert emails.emails['test']['msg'].get_content_subtype() == 'alternative' assert emails.emails['test']['msg'].get_payload()[0].get_content_type() == 'text/plain' assert emails.emails['test']['msg'].get_payload()[1].get_content_type() == 'text/html' - text = emails.emails['test']['msg'].get_payload()[0].get_payload() - html = emails.emails['test']['msg'].get_payload()[1].get_payload() - assert html.count('