fix(dbapi): ensure Cursor.__aexit__() is called with the required arguments [backport 1.14] (#6110)

Backport ce589b5 from #6002 to 1.14.

## Changes

Updates the dbapi_async implementation to pass along the `exc_...`
parameters to the `__wrapped__.__aexit__` calls.

This should not introduce any breaking changes or cause any changes to
performance.

This resolves #6001 

## Checklist

- [x] Change(s) are motivated and described in the PR description.
- [x] Testing strategy is described if automated tests are not included
in the PR.
- [x] Risk is outlined (performance impact, potential for breakage,
maintainability, etc).
- [x] Change is maintainable (easy to change, telemetry, documentation).
- [x] [Library release note
guidelines](https://ddtrace.readthedocs.io/en/stable/contributing.html#Release-Note-Guidelines)
are followed.
- [x] Documentation is included (in-code, generated user docs, [public
corp docs](https://github.com/DataDog/documentation/)).

## Reviewer Checklist

- [x] Title is accurate.
- [x] No unnecessary changes are introduced.
- [x] Description motivates each change.
- [x] Avoids breaking
[API](https://ddtrace.readthedocs.io/en/stable/versioning.html#interfaces)
changes unless absolutely necessary.
- [x] Testing strategy adequately addresses listed risk(s).
- [x] Change is maintainable (easy to change, telemetry, documentation).
- [x] Release note makes sense to a user of the library.
- [x] Reviewer has explicitly acknowledged and discussed the performance
implications of this PR as reported in the benchmarks PR comment.

Co-authored-by: Sean <[email protected]>

Author: github-actions[bot]

ddtrace/contrib/dbapi_async/__init__.py

@@ -33,7 +33,7 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
         # previous versions of the dbapi didn't support context managers. let's
         # reference the func that would be called to ensure that error
         # messages will be the same.
-        return await self.__wrapped__.__aexit__()
+        return await self.__wrapped__.__aexit__(exc_type, exc_val, exc_tb)
 
     async def _trace_method(self, method, name, resource, extra_tags, dbm_propagator, *args, **kwargs):
         """
@@ -221,7 +221,7 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
         # previous versions of the dbapi didn't support context managers. let's
         # reference the func that would be called to ensure that errors
         # messages will be the same.
-        return await self.__wrapped__.__aexit__()
+        return await self.__wrapped__.__aexit__(exc_type, exc_val, exc_tb)
 
     async def _trace_method(self, method, name, extra_tags, *args, **kwargs):
         pin = Pin.get_from(self)

releasenotes/notes/fix-dbapi-async-aexit-0df48d739dfd2a3a.yaml

@@ -0,0 +1,4 @@
+---
+fixes:
+  - |
+    psycopg: Resolves ``TypeError`` raised when an async cursor object is traced. This fix ensures `exc_type`, `exc_val`, and `exc_tb` are passed down to the wrapped object on `__aexit__`.

tests/contrib/psycopg/test_psycopg_async.py

@@ -319,6 +319,31 @@ async def test_connection_execute(self):
         assert len(rows) == 1, rows
         assert rows[0][0] == "one"
 
+    @mark_asyncio
+    async def test_connection_context_execute(self):
+        """Checks whether connection context manager works as normal."""
+
+        query = SQL("""select 'one' as x""")
+        async with (await psycopg.AsyncConnection.connect(**POSTGRES_CONFIG)) as conn:
+            cur = await conn.execute(query)
+            rows = await cur.fetchall()
+
+            assert len(rows) == 1, rows
+            assert rows[0][0] == "one"
+
+    @mark_asyncio
+    async def test_cursor_context_execute(self):
+        """Checks whether cursor context manager works as normal."""
+
+        query = SQL("""select 'one' as x""")
+        async with (await psycopg.AsyncConnection.connect(**POSTGRES_CONFIG)).cursor() as cur:
+            breakpoint()
+            await cur.execute(query)
+            rows = await cur.fetchall()
+
+            assert len(rows) == 1, rows
+            assert rows[0][0] == "one"
+
     @mark_asyncio
     async def test_cursor_from_connection_shortcut(self):
         """Checks whether connection execute shortcute method works as normal"""