diff --git a/sentry_sdk_extensions/tests.py b/sentry_sdk_extensions/tests.py new file mode 100644 index 0000000..17e87fa --- /dev/null +++ b/sentry_sdk_extensions/tests.py @@ -0,0 +1,20 @@ +from unittest import TestCase +import logging + +from sentry_sdk_extensions import capture_or_log_exception + +logger = logging.getLogger("test_logger") + + +class SentrySDKExtensionsTest(TestCase): + + def test_capture_or_log_exception(self): + with self.assertLogs(logger='test_logger', level='ERROR') as test_logger_cm: + try: + raise Exception("I failed") + except Exception as e: + # in tests, the sentry SDK is off, so this tests the "or log exception" part of the test. + capture_or_log_exception(e, logger) + + self.assertTrue('ERROR:test_logger: raise Exception("I failed")' in test_logger_cm.output) + self.assertTrue('ERROR:test_logger:Exception: I failed' in test_logger_cm.output)