40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
import unicodedata
|
||
|
|
from twisted.trial import unittest
|
||
|
|
from .. import util
|
||
|
|
|
||
|
|
class Utils(unittest.TestCase):
|
||
|
|
def test_to_bytes(self):
|
||
|
|
b = util.to_bytes(u"abc")
|
||
|
|
self.assertIsInstance(b, type(b""))
|
||
|
|
self.assertEqual(b, b"abc")
|
||
|
|
|
||
|
|
A = unicodedata.lookup("LATIN SMALL LETTER A WITH DIAERESIS")
|
||
|
|
b = util.to_bytes(A + u"bc")
|
||
|
|
self.assertIsInstance(b, type(b""))
|
||
|
|
self.assertEqual(b, b"\xc3\xa4\x62\x63")
|
||
|
|
|
||
|
|
def test_bytes_to_hexstr(self):
|
||
|
|
b = b"\x00\x45\x91\xfe\xff"
|
||
|
|
hexstr = util.bytes_to_hexstr(b)
|
||
|
|
self.assertIsInstance(hexstr, type(u""))
|
||
|
|
self.assertEqual(hexstr, u"004591feff")
|
||
|
|
|
||
|
|
def test_hexstr_to_bytes(self):
|
||
|
|
hexstr = u"004591feff"
|
||
|
|
b = util.hexstr_to_bytes(hexstr)
|
||
|
|
hexstr = util.bytes_to_hexstr(b)
|
||
|
|
self.assertIsInstance(b, type(b""))
|
||
|
|
self.assertEqual(b, b"\x00\x45\x91\xfe\xff")
|
||
|
|
|
||
|
|
def test_dict_to_bytes(self):
|
||
|
|
d = {"a": "b"}
|
||
|
|
b = util.dict_to_bytes(d)
|
||
|
|
self.assertIsInstance(b, type(b""))
|
||
|
|
self.assertEqual(b, b'{"a": "b"}')
|
||
|
|
|
||
|
|
def test_bytes_to_dict(self):
|
||
|
|
b = b'{"a": "b", "c": 2}'
|
||
|
|
d = util.bytes_to_dict(b)
|
||
|
|
self.assertIsInstance(d, dict)
|
||
|
|
self.assertEqual(d, {u"a": u"b", u"c": 2})
|