File tree 1 file changed +3
-6
lines changed
1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change @@ -55,22 +55,20 @@ def test_initialize_spark_integration(
55
55
create_spark_context ()
56
56
57
57
58
- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init " )
58
+ @patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration " )
59
59
def test_initialize_spark_integration_before_spark_context_init (
60
- mock_patch_spark_context_init ,
60
+ mock_activate_integration ,
61
61
sentry_init ,
62
62
create_spark_context ,
63
63
):
64
64
sentry_init (integrations = [SparkIntegration ()])
65
65
create_spark_context ()
66
66
67
- mock_patch_spark_context_init .assert_called_once ()
67
+ mock_activate_integration .assert_called_once ()
68
68
69
69
70
70
@patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration" )
71
- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init" )
72
71
def test_initialize_spark_integration_after_spark_context_init (
73
- mock_patch_spark_context_init ,
74
72
mock_activate_integration ,
75
73
create_spark_context ,
76
74
sentry_init ,
@@ -79,7 +77,6 @@ def test_initialize_spark_integration_after_spark_context_init(
79
77
sentry_init (integrations = [SparkIntegration ()])
80
78
81
79
mock_activate_integration .assert_called_once ()
82
- mock_patch_spark_context_init .assert_not_called ()
83
80
84
81
85
82
@pytest .fixture
You can’t perform that action at this time.
0 commit comments