From 0cedcf216ac3f2fac6b4890f2c29b6f45deffebd Mon Sep 17 00:00:00 2001 From: Marwan Zaarab Date: Fri, 9 Feb 2024 16:34:49 -0600 Subject: [PATCH] run scripts to format and lint test file --- tests/unit/analytics/test_custom_on_error.py | 23 +++++++++++--------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/tests/unit/analytics/test_custom_on_error.py b/tests/unit/analytics/test_custom_on_error.py index c701d9a4..1b8f8f87 100644 --- a/tests/unit/analytics/test_custom_on_error.py +++ b/tests/unit/analytics/test_custom_on_error.py @@ -1,10 +1,10 @@ -import pytest -from segment import analytics import logging -from unittest.mock import patch, Mock -import requests import uuid +from unittest.mock import Mock, patch +import pytest +import requests +from segment import analytics logger = logging.getLogger(__name__) analytics.max_retries = 0 @@ -13,8 +13,10 @@ def mock_post_failure(*args, **kwargs): response = requests.Response() response.status_code = 500 - response._content = b'{"error": "Simulated failure", "code": "internal_error"}' - response.headers['Content-Type'] = 'application/json' + response._content = ( + b'{"error": "Simulated failure", "code": "internal_error"}' + ) + response.headers["Content-Type"] = "application/json" return response @@ -38,10 +40,11 @@ def reset_analytics_client(): @pytest.mark.usefixtures("reset_analytics_client") def test_segment_custom_on_error_handler_invocation(caplog): - with caplog.at_level(logging.CRITICAL), \ - patch('segment.analytics.request._session.post', side_effect=mock_post_failure), \ - patch('segment.analytics.on_error', Mock()) as mock_on_error: - analytics.track('test_user_id', 'Test Event', {'property': 'value'}) + with caplog.at_level(logging.CRITICAL), patch( + "segment.analytics.request._session.post", + side_effect=mock_post_failure, + ), patch("segment.analytics.on_error", Mock()) as mock_on_error: + analytics.track("test_user_id", "Test Event", {"property": "value"}) analytics.flush() mock_on_error.assert_called()