Add mysql docker tests

Author: lrafeei

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

@@ -177,9 +177,10 @@
 
 try:
     # wrapt 2.0.0+
-    from wrapt import BaseObjectProxy  # pylint: disable=no-name-in-module
+    from wrapt import BaseObjectProxy, ObjectProxy  # pylint: disable=no-name-in-module
 except ImportError:
     from wrapt import ObjectProxy as BaseObjectProxy
+    from wrapt import ObjectProxy
 
 from opentelemetry import trace as trace_api
 from opentelemetry.instrumentation.dbapi.version import __version__
@@ -805,14 +806,14 @@ async def traced_execution_async(
 
 
 # pylint: disable=abstract-method,no-member
-class TracedCursorProxy(BaseObjectProxy, Generic[CursorT]):
+class TracedCursorProxy(ObjectProxy, Generic[CursorT]):
     # pylint: disable=unused-argument
     def __init__(
         self,
         cursor: CursorT,
         db_api_integration: DatabaseApiIntegration,
     ):
-        BaseObjectProxy.__init__(self, cursor)
+        ObjectProxy.__init__(self, cursor)
         self._self_cursor_tracer = CursorTracer[CursorT](db_api_integration)
 
     def execute(self, *args: Any, **kwargs: Any):
@@ -837,9 +838,6 @@ def __enter__(self):
     def __exit__(self, *args, **kwargs):
         self.__wrapped__.__exit__(*args, **kwargs)
 
-    def __iter__(self):
-        return iter(self.__wrapped__)
-
 
 def get_traced_cursor_proxy(
     cursor: CursorT,

tests/opentelemetry-docker-tests/tests/mysql/test_mysql_functional.py

@@ -111,6 +111,16 @@ def test_executemany(self):
             self._cursor.executemany(stmt, data)
         self.validate_spans("INSERT")
 
+    def test_executemany_with_cursor_iteration(self):
+        """Should create a child span for executemany while iterating over the cursor"""
+        stmt = "SELECT * FROM test"
+        with self._tracer.start_as_current_span("rootSpan"):
+            with self._connection.cursor() as cursor:
+                cursor.execute(stmt)
+                for _row in cursor:
+                    pass
+        self.validate_spans("SELECT")
+
     def test_callproc(self):
         """Should create a child span for callproc"""
         with (

tests/opentelemetry-docker-tests/tests/pymysql/test_pymysql_functional.py

@@ -102,6 +102,16 @@ def test_executemany(self):
             self._cursor.executemany(stmt, data)
         self.validate_spans("INSERT")
 
+    def test_executemany_with_cursor_iteration(self):
+        """Should create a child span for executemany while iterating over the cursor"""
+        stmt = "SELECT * FROM test"
+        with self._tracer.start_as_current_span("rootSpan"):
+            with self._connection.cursor() as cursor:
+                cursor.execute(stmt)
+                for _row in cursor:
+                    pass
+        self.validate_spans("SELECT")
+
     def test_callproc(self):
         """Should create a child span for callproc"""
         with (