Skip to content

Commit c3a5982

Browse files
committed
Fix: removed useless mocking on test code
1 parent fd48489 commit c3a5982

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

Diff for: tests/integrations/spark/test_spark.py

+3-6
Original file line numberDiff line numberDiff 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")
5959
def 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")
7271
def 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

0 commit comments

Comments
 (0)