Skip to content

Commit 7388a3c

Browse files
committed
base64.encodestring is deprecated on python > 3.1. Eliminate warning
using encodebytes on py 3.
1 parent a2800bc commit 7388a3c

File tree

1 file changed

+15
-8
lines changed

1 file changed

+15
-8
lines changed

test/db_test_base.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,14 @@
1616
# SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
1717

1818
from __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+
2027
import logging, cgi
2128
from . import gpgmelib
2229
from 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

Comments
 (0)