File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Expand file tree Collapse file tree 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(
5555 create_spark_context ()
5656
5757
58- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init " )
58+ @patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration " )
5959def test_initialize_spark_integration_before_spark_context_init (
60- mock_patch_spark_context_init ,
60+ mock_activate_integration ,
6161 sentry_init ,
6262 create_spark_context ,
6363):
6464 sentry_init (integrations = [SparkIntegration ()])
6565 create_spark_context ()
6666
67- mock_patch_spark_context_init .assert_called_once ()
67+ mock_activate_integration .assert_called_once ()
6868
6969
7070@patch ("sentry_sdk.integrations.spark.spark_driver._activate_integration" )
71- @patch ("sentry_sdk.integrations.spark.spark_driver._patch_spark_context_init" )
7271def test_initialize_spark_integration_after_spark_context_init (
73- mock_patch_spark_context_init ,
7472 mock_activate_integration ,
7573 create_spark_context ,
7674 sentry_init ,
@@ -79,7 +77,6 @@ def test_initialize_spark_integration_after_spark_context_init(
7977 sentry_init (integrations = [SparkIntegration ()])
8078
8179 mock_activate_integration .assert_called_once ()
82- mock_patch_spark_context_init .assert_not_called ()
8380
8481
8582@pytest .fixture
You can’t perform that action at this time.
0 commit comments