iframe-proxy | Sunbelt Computer Software

History

119 lines (97 loc) · 3.42 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

# -*- coding: utf-8 -*-

from __future__ import absolute_import

from __future__ import unicode_literals

import io

import os.path

import re

import sys

import mock

import pytest

from pre_commit import error_handler

from testing.util import cmd_output_mocked_pre_commit_home

@pytest.fixture

def mocked_log_and_exit():

with mock.patch.object(error_handler, '_log_and_exit') as log_and_exit:

yield log_and_exit

def test_error_handler_no_exception(mocked_log_and_exit):

with error_handler.error_handler():

pass

assert mocked_log_and_exit.call_count == 0

def test_error_handler_fatal_error(mocked_log_and_exit):

exc = error_handler.FatalError('just a test')

with error_handler.error_handler():

raise exc

mocked_log_and_exit.assert_called_once_with(

'An error has occurred',

exc,

# Tested below

mock.ANY,

)

assert re.match(

r'Traceback \(most recent call last\):\n'

r' File ".+pre_commit.error_handler.py", line \d+, in error_handler\n'

r' yield\n'

r' File ".+tests.error_handler_test.py", line \d+, '

r'in test_error_handler_fatal_error\n'

r' raise exc\n'

r'(pre_commit\.error_handler\.)?FatalError: just a test\n',

mocked_log_and_exit.call_args[0][2],

)

def test_error_handler_uncaught_error(mocked_log_and_exit):

exc = ValueError('another test')

with error_handler.error_handler():

raise exc

mocked_log_and_exit.assert_called_once_with(

'An unexpected error has occurred',

exc,

# Tested below

mock.ANY,

)

assert re.match(

r'Traceback \(most recent call last\):\n'

r' File ".+pre_commit.error_handler.py", line \d+, in error_handler\n'

r' yield\n'

r' File ".+tests.error_handler_test.py", line \d+, '

r'in test_error_handler_uncaught_error\n'

r' raise exc\n'

r'ValueError: another test\n',

mocked_log_and_exit.call_args[0][2],

)

def test_log_and_exit(cap_out, mock_store_dir):

with pytest.raises(SystemExit):

error_handler._log_and_exit(

'msg', error_handler.FatalError('hai'), "I'm a stacktrace",

)

printed = cap_out.get()

log_file = os.path.join(mock_store_dir, 'pre-commit.log')

assert printed == (

'msg: FatalError: hai\n'

'Check the log at {}\n'.format(log_file)

)

assert os.path.exists(log_file)

with io.open(log_file) as f:

assert f.read() == (

'msg: FatalError: hai\n'

"I'm a stacktrace\n"

)

def test_error_handler_non_ascii_exception(mock_store_dir):

with pytest.raises(SystemExit):

with error_handler.error_handler():

raise ValueError('☃')

def test_error_handler_no_tty(tempdir_factory):

pre_commit_home = tempdir_factory.get()

output = cmd_output_mocked_pre_commit_home(

sys.executable, '-c',

'from __future__ import unicode_literals\n'

'from pre_commit.error_handler import error_handler\n'

'with error_handler():\n'

' raise ValueError("\\u2603")\n',

retcode=1,

tempdir_factory=tempdir_factory,

pre_commit_home=pre_commit_home,

)

log_file = os.path.join(pre_commit_home, 'pre-commit.log')

assert output[1].replace('\r', '') == (

'An unexpected error has occurred: ValueError: ☃\n'

'Check the log at {}\n'.format(log_file)

)

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

error_handler_test.py

Latest commit

History

error_handler_test.py

File metadata and controls