diff options
author | abodrov <[email protected]> | 2022-02-10 16:47:18 +0300 |
---|---|---|
committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:47:18 +0300 |
commit | 1b8c043abfc689b3d27384c4202b20031fc4ae31 (patch) | |
tree | 43ffe3a7cfe51a083e22d9eabb71c4f5321f6b08 /library/python/runtime_py3/test/test_traceback.py | |
parent | e7a025f6f4917a3eb54227db101e58bb26ee2cfa (diff) |
Restoring authorship annotation for <[email protected]>. Commit 1 of 2.
Diffstat (limited to 'library/python/runtime_py3/test/test_traceback.py')
-rw-r--r-- | library/python/runtime_py3/test/test_traceback.py | 124 |
1 files changed, 62 insertions, 62 deletions
diff --git a/library/python/runtime_py3/test/test_traceback.py b/library/python/runtime_py3/test/test_traceback.py index 82087ce98a6..0cc4dec75a3 100644 --- a/library/python/runtime_py3/test/test_traceback.py +++ b/library/python/runtime_py3/test/test_traceback.py @@ -1,63 +1,63 @@ -# -*- coding: utf-8 -*- - -from __future__ import print_function, absolute_import, division - -import os -import re - -import pytest - -import yatest.common as yc - - -def clean_traceback(traceback): - traceback = re.sub(br'\033\[(\d|;)+?m', b'', traceback) # strip ANSI codes - traceback = re.sub(br' at 0x[0-9a-fA-F]+', b'', traceback) # remove object ids - return traceback - - [email protected]('mode', [ - 'default', - 'ultratb_color', - 'ultratb_verbose', -]) [email protected]('entry_point', [ - 'main', - 'custom', -]) -def test_traceback(mode, entry_point): - tb_tool = yc.build_path('library/python/runtime_py3/test/traceback/traceback') - stdout_path = yc.test_output_path('stdout_raw.txt') - stderr_path = yc.test_output_path('stderr_raw.txt') - filtered_stdout_path = yc.test_output_path('stdout.txt') - filtered_stderr_path = yc.test_output_path('stderr.txt') - - env = os.environ.copy() +# -*- coding: utf-8 -*- + +from __future__ import print_function, absolute_import, division + +import os +import re + +import pytest + +import yatest.common as yc + + +def clean_traceback(traceback): + traceback = re.sub(br'\033\[(\d|;)+?m', b'', traceback) # strip ANSI codes + traceback = re.sub(br' at 0x[0-9a-fA-F]+', b'', traceback) # remove object ids + return traceback + + [email protected]('mode', [ + 'default', + 'ultratb_color', + 'ultratb_verbose', +]) [email protected]('entry_point', [ + 'main', + 'custom', +]) +def test_traceback(mode, entry_point): + tb_tool = yc.build_path('library/python/runtime_py3/test/traceback/traceback') + stdout_path = yc.test_output_path('stdout_raw.txt') + stderr_path = yc.test_output_path('stderr_raw.txt') + filtered_stdout_path = yc.test_output_path('stdout.txt') + filtered_stderr_path = yc.test_output_path('stderr.txt') + + env = os.environ.copy() env.pop('PYTHONPATH', None) # Do not let program peek into its sources on filesystem - if entry_point == 'custom': - env['Y_PYTHON_ENTRY_POINT'] = 'library.python.runtime_py3.test.traceback.crash:main' - - proc = yc.execute( - command=[tb_tool, mode], - env=env, - stdout=stdout_path, - stderr=stderr_path, - check_exit_code=False, - ) - - with open(filtered_stdout_path, 'wb') as f: - f.write(clean_traceback(proc.std_out)) - - with open(filtered_stderr_path, 'wb') as f: - f.write(clean_traceback(proc.std_err)) - - return { - 'stdout': yc.canonical_file( - filtered_stdout_path, - local=True, - ), - 'stderr': yc.canonical_file( - filtered_stderr_path, - local=True, - ), - } + if entry_point == 'custom': + env['Y_PYTHON_ENTRY_POINT'] = 'library.python.runtime_py3.test.traceback.crash:main' + + proc = yc.execute( + command=[tb_tool, mode], + env=env, + stdout=stdout_path, + stderr=stderr_path, + check_exit_code=False, + ) + + with open(filtered_stdout_path, 'wb') as f: + f.write(clean_traceback(proc.std_out)) + + with open(filtered_stderr_path, 'wb') as f: + f.write(clean_traceback(proc.std_err)) + + return { + 'stdout': yc.canonical_file( + filtered_stdout_path, + local=True, + ), + 'stderr': yc.canonical_file( + filtered_stderr_path, + local=True, + ), + } |