@@ -48,29 +48,25 @@ def test_start_sentry_listener(create_spark_context):
48
48
assert gateway ._callback_server is not None
49
49
50
50
51
- def test_initialize_spark_integration (
52
- sentry_init , create_spark_context , reset_integrations
53
- ):
51
+ def test_initialize_spark_integration (sentry_init , create_spark_context ):
54
52
sentry_init (integrations = [SparkIntegration ()])
55
53
create_spark_context ()
56
54
57
55
58
- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init " )
56
+ @patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration " )
59
57
def test_initialize_spark_integration_before_spark_context_init (
60
- mock_patch_spark_context_init ,
58
+ mock_activate_integration ,
61
59
sentry_init ,
62
60
create_spark_context ,
63
61
):
64
62
sentry_init (integrations = [SparkIntegration ()])
65
63
create_spark_context ()
66
64
67
- mock_patch_spark_context_init .assert_called_once ()
65
+ mock_activate_integration .assert_called_once ()
68
66
69
67
70
68
@patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration" )
71
- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init" )
72
69
def test_initialize_spark_integration_after_spark_context_init (
73
- mock_patch_spark_context_init ,
74
70
mock_activate_integration ,
75
71
create_spark_context ,
76
72
sentry_init ,
@@ -79,7 +75,6 @@ def test_initialize_spark_integration_after_spark_context_init(
79
75
sentry_init (integrations = [SparkIntegration ()])
80
76
81
77
mock_activate_integration .assert_called_once ()
82
- mock_patch_spark_context_init .assert_not_called ()
83
78
84
79
85
80
@pytest .fixture
0 commit comments