Update pgproto and add tests for client_encoding

Depends on MagicStack/py-pgproto#20
MagicStack · Feb 1, 2023 · a8b3188 · a8b3188
1 parent 7df9812
commit a8b3188
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/asyncpg/pgproto b/asyncpg/pgproto
diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -77,6 +77,15 @@ async def test_get_settings_01(self):
         self.assertEqual(
             self.con.get_settings().client_encoding,
             'UTF8')
+        await self.con.execute("set client_encoding to 'sql-ascii'")
+        self.assertEqual(self.con.get_settings().client_encoding, "SQL_ASCII")
+
+    async def test_client_encoding(self):
+        await self.con.execute("set client_encoding to 'sql-ascii'")
+        encoding = await self.con.fetchval("show client_encoding")
+        self.assertEqual(encoding, "SQL_ASCII")
+        encoding = await self.con.fetchval("select $1::text", encoding)
+        self.assertEqual(encoding, "SQL_ASCII")
 
     async def test_server_version_01(self):
         version = self.con.get_server_version()