fix sqlalchemy test

Author: sentrivana

tests/integrations/sqlalchemy/test_sqlalchemy.py

@@ -1080,19 +1080,19 @@ class Person(Base):
 
             class fake_record_sql_queries:  # noqa: N801
                 def __init__(self, *args, **kwargs):
-                    with record_sql_queries_supporting_streaming(
+                    self._ctx_mgr = record_sql_queries_supporting_streaming(
                         *args, **kwargs
-                    ) as span:
-                        self.span = span
+                    )
 
+                def __enter__(self):
+                    self.span = self._ctx_mgr.__enter__()
                     self.span._start_timestamp = datetime(2024, 1, 1, microsecond=0)
                     self.span._end_timestamp = datetime(2024, 1, 1, microsecond=101000)
-
-                def __enter__(self):
                     return self.span
 
                 def __exit__(self, type, value, traceback):
-                    pass
+                    self.span._end_timestamp = None
+                    self._ctx_mgr.__exit__(type, value, traceback)
 
             with mock.patch(
                 "sentry_sdk.integrations.sqlalchemy.record_sql_queries_supporting_streaming",