Removed hub from session tests

Author: antonpirker

tests/test_sessions.py

@@ -1,7 +1,6 @@
 from unittest import mock
 
 import sentry_sdk
-from sentry_sdk import Hub
 from sentry_sdk.sessions import auto_session_tracking
 
 
@@ -15,17 +14,17 @@ def test_basic(sentry_init, capture_envelopes):
     sentry_init(release="fun-release", environment="not-fun-env")
     envelopes = capture_envelopes()
 
-    hub = Hub.current
-    hub.start_session()
+    sentry_sdk.Scope.get_isolation_scope().start_session()
 
     try:
-        with hub.configure_scope() as scope:
-            scope.set_user({"id": "42"})
-            raise Exception("all is wrong")
+        scope = sentry_sdk.Scope.get_current_scope()
+        scope.set_user({"id": "42"})
+        raise Exception("all is wrong")
     except Exception:
-        hub.capture_exception()
-    hub.end_session()
-    hub.flush()
+        sentry_sdk.capture_exception()
+
+    sentry_sdk.Scope.get_isolation_scope().end_session()
+    sentry_sdk.flush()
 
     assert len(envelopes) == 2
     assert envelopes[0].get_event() is not None
@@ -51,22 +50,20 @@ def test_aggregates(sentry_init, capture_envelopes):
     )
     envelopes = capture_envelopes()
 
-    hub = Hub.current
-
     with auto_session_tracking(session_mode="request"):
         with sentry_sdk.push_scope():
             try:
-                with sentry_sdk.configure_scope() as scope:
-                    scope.set_user({"id": "42"})
-                    raise Exception("all is wrong")
+                scope = sentry_sdk.Scope.get_current_scope()
+                scope.set_user({"id": "42"})
+                raise Exception("all is wrong")
             except Exception:
                 sentry_sdk.capture_exception()
 
     with auto_session_tracking(session_mode="request"):
         pass
 
-    hub.start_session(session_mode="request")
-    hub.end_session()
+    sentry_sdk.Scope.get_isolation_scope().start_session(session_mode="request")
+    sentry_sdk.Scope.get_isolation_scope().end_session()
 
     sentry_sdk.flush()
 
@@ -95,8 +92,6 @@ def test_aggregates_explicitly_disabled_session_tracking_request_mode(
     )
     envelopes = capture_envelopes()
 
-    hub = Hub.current
-
     with auto_session_tracking(session_mode="request"):
         with sentry_sdk.push_scope():
             try:
@@ -107,8 +102,8 @@ def test_aggregates_explicitly_disabled_session_tracking_request_mode(
     with auto_session_tracking(session_mode="request"):
         pass
 
-    hub.start_session(session_mode="request")
-    hub.end_session()
+    sentry_sdk.Scope.get_isolation_scope().start_session(session_mode="request")
+    sentry_sdk.Scope.get_isolation_scope().end_session()
 
     sentry_sdk.flush()
 
@@ -128,8 +123,6 @@ def test_no_thread_on_shutdown_no_errors(sentry_init):
         environment="not-fun-env",
     )
 
-    hub = Hub.current
-
     # make it seem like the interpreter is shutting down
     with mock.patch(
         "threading.Thread.start",
@@ -145,7 +138,6 @@ def test_no_thread_on_shutdown_no_errors(sentry_init):
         with auto_session_tracking(session_mode="request"):
             pass
 
-        hub.start_session(session_mode="request")
-        hub.end_session()
-
+        sentry_sdk.Scope.get_isolation_scope().start_session(session_mode="request")
+        sentry_sdk.Scope.get_isolation_scope().end_session()
         sentry_sdk.flush()