1616# SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
1717
1818from __future__ import print_function
19- import unittest , os , shutil , errno , imp , sys , time , pprint , base64 , os .path
19+ import unittest , os , shutil , errno , imp , sys , time , pprint , os .path
20+
21+ try :
22+ from base64 import encodebytes as base64_encode # python3 only
23+ except ImportError :
24+ # python2 and deplricated in 3
25+ from base64 import encodestring as base64_encode
26+
2027import logging , cgi
2128from . import gpgmelib
2229from email import message_from_string
@@ -2624,9 +2631,9 @@ def dummy_snd(s, to, msg, res=res) :
26242631 self .assertTrue ("New submission from admin" in mail_msg )
26252632 self .assertTrue ("one two" in mail_msg )
26262633 self .assertTrue ("File 'test1.txt' not attached" not in mail_msg )
2627- self .assertTrue (b2s (base64 . encodestring (s2b ("xxx" ))).rstrip () in mail_msg )
2634+ self .assertTrue (b2s (base64_encode (s2b ("xxx" ))).rstrip () in mail_msg )
26282635 self .assertTrue ("File 'test2.txt' not attached" in mail_msg )
2629- self .assertTrue (b2s (base64 . encodestring (s2b ("yyy" ))).rstrip () not in mail_msg )
2636+ self .assertTrue (b2s (base64_encode (s2b ("yyy" ))).rstrip () not in mail_msg )
26302637 finally :
26312638 roundupdb ._ = old_translate_
26322639 Mailer .smtp_send = backup
@@ -2657,7 +2664,7 @@ def dummy_snd(s, to, msg, res=res) :
26572664 self .assertTrue ("New submission from admin" in mail_msg )
26582665 self .assertTrue ("one two" in mail_msg )
26592666 self .assertTrue ("Hello world" in mail_msg )
2660- self .assertTrue (b2s (base64 . encodestring (b"\x01 \x02 \x03 \xfe \xff " )).rstrip () in mail_msg )
2667+ self .assertTrue (b2s (base64_encode (b"\x01 \x02 \x03 \xfe \xff " )).rstrip () in mail_msg )
26612668 finally :
26622669 roundupdb ._ = old_translate_
26632670 Mailer .smtp_send = backup
@@ -2701,9 +2708,9 @@ def dummy_snd(s, to, msg, res=res) :
27012708 self .assertTrue ("New submission from admin" in mail_msg )
27022709 self .assertTrue ("one two" in mail_msg )
27032710 self .assertTrue ("File 'test1.txt' not attached" not in mail_msg )
2704- self .assertTrue (b2s (base64 . encodestring (s2b ("xxx" ))).rstrip () in mail_msg )
2711+ self .assertTrue (b2s (base64_encode (s2b ("xxx" ))).rstrip () in mail_msg )
27052712 self .assertTrue ("File 'test2.txt' not attached" in mail_msg )
2706- self .assertTrue (b2s (base64 . encodestring (s2b ("yyy" ))).rstrip () not in mail_msg )
2713+ self .assertTrue (b2s (base64_encode (s2b ("yyy" ))).rstrip () not in mail_msg )
27072714 mail_msg = str (res [1 ]["mail_msg" ])
27082715 parts = message_from_string (mail_msg ).get_payload ()
27092716 self .assertEqual (len (parts ),2 )
@@ -2720,9 +2727,9 @@ def dummy_snd(s, to, msg, res=res) :
27202727 self .assertTrue ("New submission from admin" in mail_msg )
27212728 self .assertTrue ("one two" in mail_msg )
27222729 self .assertTrue ("File 'test1.txt' not attached" not in mail_msg )
2723- self .assertTrue (b2s (base64 . encodestring (s2b ("xxx" ))).rstrip () in mail_msg )
2730+ self .assertTrue (b2s (base64_encode (s2b ("xxx" ))).rstrip () in mail_msg )
27242731 self .assertTrue ("File 'test2.txt' not attached" in mail_msg )
2725- self .assertTrue (b2s (base64 . encodestring (s2b ("yyy" ))).rstrip () not in mail_msg )
2732+ self .assertTrue (b2s (base64_encode (s2b ("yyy" ))).rstrip () not in mail_msg )
27262733 finally :
27272734 roundupdb ._ = old_translate_
27282735 Mailer .smtp_send = backup
0 commit comments