Added test for BYTES typecaster

This commit is contained in:
Daniele Varrazzo 2019-01-18 15:10:17 +00:00
parent fd31a7d41b
commit 4ab4247189

View File

@ -170,6 +170,17 @@ class QuotingTestCase(ConnectingTestCase):
self.assertEqual(res, data) self.assertEqual(res, data)
self.assert_(not self.conn.notices) self.assert_(not self.conn.notices)
def test_bytes(self):
snowman = u"\u2603"
conn = self.connect()
conn.set_client_encoding('UNICODE')
psycopg2.extensions.register_type(psycopg2.extensions.BYTES, conn)
curs = conn.cursor()
curs.execute("select %s::text", (snowman,))
x = curs.fetchone()[0]
self.assert_(isinstance(x, bytes))
self.assertEqual(x, snowman.encode('utf8'))
class TestQuotedString(ConnectingTestCase): class TestQuotedString(ConnectingTestCase):
def test_encoding_from_conn(self): def test_encoding_from_conn(self):