From 528607d5b5b67c1b9c2edb39176b7e005db371f6 Mon Sep 17 00:00:00 2001
From: =?utf8?q?S=C3=B6ren=20Oldag?= <soeren.oldag@kiwi.ki>
Date: Tue, 24 Oct 2023 14:09:03 +0200
Subject: [PATCH] Add bind casts for BIT on asyncpg

Fixes: #10532
---
 lib/sqlalchemy/dialects/postgresql/asyncpg.py |  6 ++
 test/dialect/postgresql/test_types.py         | 60 +++++++++++++++++++
 2 files changed, 66 insertions(+)

diff --git a/lib/sqlalchemy/dialects/postgresql/asyncpg.py b/lib/sqlalchemy/dialects/postgresql/asyncpg.py
index 4fa188cb02..ca35bf9607 100644
--- a/lib/sqlalchemy/dialects/postgresql/asyncpg.py
+++ b/lib/sqlalchemy/dialects/postgresql/asyncpg.py
@@ -205,6 +205,7 @@ from .base import PGExecutionContext
 from .base import PGIdentifierPreparer
 from .base import REGCLASS
 from .base import REGCONFIG
+from .types import BIT
 from .types import BYTEA
 from .types import CITEXT
 from ... import exc
@@ -237,6 +238,10 @@ class AsyncpgTime(sqltypes.Time):
     render_bind_cast = True
 
 
+class AsyncpgBit(BIT):
+    render_bind_cast = True
+
+
 class AsyncpgByteA(BYTEA):
     render_bind_cast = True
 
@@ -1016,6 +1021,7 @@ class PGDialect_asyncpg(PGDialect):
         {
             sqltypes.String: AsyncpgString,
             sqltypes.ARRAY: AsyncpgARRAY,
+            BIT: AsyncpgBit,
             CITEXT: CITEXT,
             REGCONFIG: AsyncpgREGCONFIG,
             sqltypes.Time: AsyncpgTime,
diff --git a/test/dialect/postgresql/test_types.py b/test/dialect/postgresql/test_types.py
index 95bbb16636..4380d0273b 100644
--- a/test/dialect/postgresql/test_types.py
+++ b/test/dialect/postgresql/test_types.py
@@ -46,6 +46,7 @@ from sqlalchemy.dialects.postgresql import array
 from sqlalchemy.dialects.postgresql import array_agg
 from sqlalchemy.dialects.postgresql import asyncpg
 from sqlalchemy.dialects.postgresql import base
+from sqlalchemy.dialects.postgresql import BIT
 from sqlalchemy.dialects.postgresql import BYTEA
 from sqlalchemy.dialects.postgresql import CITEXT
 from sqlalchemy.dialects.postgresql import DATEMULTIRANGE
@@ -6291,3 +6292,62 @@ class PGInsertManyValuesTest(fixtures.TestBase):
             set(connection.scalars(select(t.c.value))),
             {value},
         )
+
+    @testing.variation("sort_by_parameter_order", [True, False])
+    @testing.variation("multiple_rows", [True, False])
+    @testing.only_on("postgresql+asyncpg")
+    @testing.requires.insert_returning
+    def test_imv_returning_datatypes_asyncpg_bit(
+        self,
+        connection,
+        metadata,
+        sort_by_parameter_order,
+        multiple_rows,
+    ):
+        """test #10532
+
+        this tests insertmanyvalues in conjunction with the BIT datatype
+        on asyncpg.
+
+        These tests are particularly for the asyncpg driver which needs
+        most types to be explicitly cast for the new IMV format
+
+        """
+        from asyncpg import BitString
+
+        t = Table(
+            "d_t",
+            metadata,
+            Column("id", Integer, primary_key=True),
+            Column("value", BIT(3)),
+        )
+
+        t.create(connection)
+
+        value = BitString.from_int(3, length=3)
+
+        result = connection.execute(
+            t.insert().returning(
+                t.c.id,
+                t.c.value,
+                sort_by_parameter_order=bool(sort_by_parameter_order),
+            ),
+            [{"value": value} for i in range(10)]
+            if multiple_rows
+            else {"value": value},
+        )
+
+        if multiple_rows:
+            i_range = range(1, 11)
+        else:
+            i_range = range(1, 2)
+
+        eq_(
+            set(result),
+            {(id_, value) for id_ in i_range},
+        )
+
+        eq_(
+            set(connection.scalars(select(t.c.value))),
+            {value},
+        )
-- 
2.47.3