Add unit test

instrumentation/opentelemetry-instrumentation-asyncpg/src/opentelemetry/instrumentation/asyncpg/__init__.py

@@ -138,22 +138,13 @@ def _instrument(self, **kwargs):
             )
 
     def _uninstrument(self, **__):
-        for method in [
-            "execute",
-            "executemany",
-            "fetch",
-            "fetchval",
-            "fetchrow",
-        ]:
-            unwrap(asyncpg.connection.Connection, method)
-
-        for method in [
-            "fetch",
-            "forward",
-            "fetchrow",
-            "__anext__",
+        for cls, methods in [
+            (asyncpg.connection.Connection, ("execute", "executemany", "fetch", "fetchval", "fetchrow")),
+            (asyncpg.cursor.Cursor, ("forward", "fetch", "fetchrow")), 
+            (asyncpg.cursor.CursorIterator, ("__anext__", ))
         ]:
-            unwrap(asyncpg.cursor, method)
+            for method_name in methods:
+                unwrap(cls, method_name)
 
     async def _do_execute(self, func, instance, args, kwargs):
         exception = None

instrumentation/opentelemetry-instrumentation-asyncpg/tests/test_asyncpg_wrapper.py

@@ -1,8 +1,10 @@
-from asyncpg import Connection
+from asyncpg import Connection, cursor
 
 from opentelemetry.instrumentation.asyncpg import AsyncPGInstrumentor
 from opentelemetry.test.test_base import TestBase
 
+from wrapt import ObjectProxy
+
 
 class TestAsyncPGInstrumentation(TestBase):
     def test_duplicated_instrumentation_can_be_uninstrumented(self):
@@ -34,3 +36,21 @@ def test_duplicated_uninstrumentation(self):
             self.assertFalse(
                 hasattr(method, "_opentelemetry_ext_asyncpg_applied")
             )
+
+    def test_cursor_instrumentation(self):
+        def assert_wrapped(assert_fnc):
+            for cls, methods in [
+                (cursor.Cursor, ("forward", "fetch", "fetchrow")), 
+                (cursor.CursorIterator, ("__anext__", ))
+            ]:
+                for method_name in methods:
+                    method = getattr(cls, method_name, None)
+                    assert_fnc(isinstance(method, ObjectProxy), f"{method} isinstance {type(method)}")
+
+
+        assert_wrapped(self.assertFalse)
+        AsyncPGInstrumentor().instrument()
+        assert_wrapped(self.assertTrue)
+        AsyncPGInstrumentor().uninstrument()
+        assert_wrapped(self.assertFalse)
+