x/wsme
Code Issues Proposed changes

fix binary type tests for python 3

Browse Source
This commit is contained in:
Doug Hellmann 2013-01-18 16:53:33 -05:00
parent a896345971
commit 31497bff37
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wsme/tests/test_types.py
View File

@ -401,13 +401,14 @@ class TestTypes(unittest.TestCase):
def test_binary_to_base(self): def test_binary_to_base(self):
import base64 import base64
assert types.binary.tobasetype(None) is None assert types.binary.tobasetype(None) is None
assert types.binary.tobasetype('abcdef') == base64.encodestring('abcdef') expected = base64.encodestring(six.b('abcdef'))
assert types.binary.tobasetype(six.b('abcdef')) == expected
def test_binary_from_base(self): def test_binary_from_base(self):
import base64 import base64
assert types.binary.frombasetype(None) is None assert types.binary.frombasetype(None) is None
encoded = base64.encodestring('abcdef') encoded = base64.encodestring(six.b('abcdef'))
assert types.binary.frombasetype(encoded) == 'abcdef' assert types.binary.frombasetype(encoded) == six.b('abcdef')
def test_wsattr_weakref_datatype(self): def test_wsattr_weakref_datatype(self):
# If the datatype inside the wsattr ends up a weakref, it # If the datatype inside the wsattr ends up a weakref, it