diff --git a/tests/test_basics.py b/tests/test_basics.py index 0fdf9f811f..2eeba78216 100644 --- a/tests/test_basics.py +++ b/tests/test_basics.py @@ -515,6 +515,66 @@ def test_attachments_graceful_failure( assert envelope.items[1].payload.get_bytes() == b"" +def test_attachments_exceptions(sentry_init): + sentry_init() + + scope = sentry_sdk.get_isolation_scope() + + # bytes and path are None + with pytest.raises(TypeError) as e: + scope.add_attachment() + + assert str(e.value) == "path or raw bytes required for attachment" + + # filename is None + with pytest.raises(TypeError) as e: + scope.add_attachment(bytes=b"Hello World!") + + assert str(e.value) == "filename is required for attachment" + + +def test_attachments_content_type_is_none(sentry_init, capture_envelopes): + sentry_init() + envelopes = capture_envelopes() + + scope = sentry_sdk.get_isolation_scope() + + scope.add_attachment( + bytes=b"Hello World!", filename="message.txt", content_type="foo/bar" + ) + capture_exception(ValueError()) + + (envelope,) = envelopes + attachments = [x for x in envelope.items if x.type == "attachment"] + (message,) = attachments + + assert message.headers["filename"] == "message.txt" + assert message.headers["content_type"] == "foo/bar" + + +def test_attachments_repr(sentry_init): + sentry_init() + + scope = sentry_sdk.get_isolation_scope() + + scope.add_attachment(bytes=b"Hello World!", filename="message.txt") + + assert repr(scope._attachments[0]) == "" + + +def test_attachments_bytes_callable_payload(sentry_init): + sentry_init() + + scope = sentry_sdk.get_isolation_scope() + + scope.add_attachment(bytes=bytes, filename="message.txt") + + attachment = scope._attachments[0] + item = attachment.to_envelope_item() + + assert item.payload.bytes == b"" + + def test_integration_scoping(sentry_init, capture_events): logger = logging.getLogger("test_basics") diff --git a/tests/test_utils.py b/tests/test_utils.py index b731c3e3ab..efa2e7c068 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -32,6 +32,10 @@ _get_installed_modules, _generate_installed_modules, ensure_integration_enabled, + to_string, + exc_info_from_error, + get_lines_from_file, + package_version, ) @@ -61,55 +65,72 @@ def _normalize_distribution_name(name): return re.sub(r"[-_.]+", "-", name).lower() -@pytest.mark.parametrize( - ("input_str", "expected_output"), +isoformat_inputs_and_datetime_outputs = ( ( - ( - "2021-01-01T00:00:00.000000Z", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), # UTC time - ( - "2021-01-01T00:00:00.000000", - datetime(2021, 1, 1).astimezone(timezone.utc), - ), # No TZ -- assume local but convert to UTC - ( - "2021-01-01T00:00:00Z", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), # UTC - No milliseconds - ( - "2021-01-01T00:00:00.000000+00:00", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), - ( - "2021-01-01T00:00:00.000000-00:00", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), - ( - "2021-01-01T00:00:00.000000+0000", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), - ( - "2021-01-01T00:00:00.000000-0000", - datetime(2021, 1, 1, tzinfo=timezone.utc), - ), - ( - "2020-12-31T00:00:00.000000+02:00", - datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=2))), - ), # UTC+2 time - ( - "2020-12-31T00:00:00.000000-0200", - datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=-2))), - ), # UTC-2 time - ( - "2020-12-31T00:00:00-0200", - datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=-2))), - ), # UTC-2 time - no milliseconds + "2021-01-01T00:00:00.000000Z", + datetime(2021, 1, 1, tzinfo=timezone.utc), + ), # UTC time + ( + "2021-01-01T00:00:00.000000", + datetime(2021, 1, 1).astimezone(timezone.utc), + ), # No TZ -- assume local but convert to UTC + ( + "2021-01-01T00:00:00Z", + datetime(2021, 1, 1, tzinfo=timezone.utc), + ), # UTC - No milliseconds + ( + "2021-01-01T00:00:00.000000+00:00", + datetime(2021, 1, 1, tzinfo=timezone.utc), + ), + ( + "2021-01-01T00:00:00.000000-00:00", + datetime(2021, 1, 1, tzinfo=timezone.utc), + ), + ( + "2021-01-01T00:00:00.000000+0000", + datetime(2021, 1, 1, tzinfo=timezone.utc), + ), + ( + "2021-01-01T00:00:00.000000-0000", + datetime(2021, 1, 1, tzinfo=timezone.utc), ), + ( + "2020-12-31T00:00:00.000000+02:00", + datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=2))), + ), # UTC+2 time + ( + "2020-12-31T00:00:00.000000-0200", + datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=-2))), + ), # UTC-2 time + ( + "2020-12-31T00:00:00-0200", + datetime(2020, 12, 31, tzinfo=timezone(timedelta(hours=-2))), + ), # UTC-2 time - no milliseconds +) + + +@pytest.mark.parametrize( + ("input_str", "expected_output"), + isoformat_inputs_and_datetime_outputs, ) def test_datetime_from_isoformat(input_str, expected_output): assert datetime_from_isoformat(input_str) == expected_output, input_str +@pytest.mark.parametrize( + ("input_str", "expected_output"), + isoformat_inputs_and_datetime_outputs, +) +def test_datetime_from_isoformat_with_py_36_or_lower(input_str, expected_output): + """ + `fromisoformat` was added in Python version 3.7 + """ + with mock.patch("sentry_sdk.utils.datetime") as datetime_mocked: + datetime_mocked.fromisoformat.side_effect = AttributeError() + datetime_mocked.strptime = datetime.strptime + assert datetime_from_isoformat(input_str) == expected_output, input_str + + @pytest.mark.parametrize( "env_var_value,strict,expected", [ @@ -346,6 +367,12 @@ def test_sanitize_url_and_split(url, expected_result): assert sanitized_url.fragment == expected_result.fragment +def test_sanitize_url_remove_authority_is_false(): + url = "https://usr:pwd@example.com" + sanitized_url = sanitize_url(url, remove_authority=False) + assert sanitized_url == url + + @pytest.mark.parametrize( ("url", "sanitize", "expected_url", "expected_query", "expected_fragment"), [ @@ -629,6 +656,17 @@ def test_get_error_message(error, expected_result): assert get_error_message(exc_value) == expected_result(exc_value) +def test_safe_str_fails(): + class ExplodingStr: + def __str__(self): + raise Exception + + obj = ExplodingStr() + result = safe_str(obj) + + assert result == repr(obj) + + def test_installed_modules(): try: from importlib.metadata import distributions, version @@ -712,6 +750,20 @@ def test_default_release_empty_string(): assert release is None +def test_get_default_release_sentry_release_env(monkeypatch): + monkeypatch.setenv("SENTRY_RELEASE", "sentry-env-release") + assert get_default_release() == "sentry-env-release" + + +def test_get_default_release_other_release_env(monkeypatch): + monkeypatch.setenv("SOURCE_VERSION", "other-env-release") + + with mock.patch("sentry_sdk.utils.get_git_revision", return_value=""): + release = get_default_release() + + assert release == "other-env-release" + + def test_ensure_integration_enabled_integration_enabled(sentry_init): def original_function(): return "original" @@ -973,3 +1025,55 @@ def test_function(): ... sentry_sdk.utils.qualname_from_function(test_function) == "test_qualname_from_function_none_name..test_function" ) + + +def test_to_string_unicode_decode_error(): + class BadStr: + def __str__(self): + raise UnicodeDecodeError("utf-8", b"", 0, 1, "reason") + + obj = BadStr() + result = to_string(obj) + assert result == repr(obj)[1:-1] + + +def test_exc_info_from_error_dont_get_an_exc(): + class NotAnException: + pass + + with pytest.raises(ValueError) as exc: + exc_info_from_error(NotAnException()) + + assert "Expected Exception object to report, got