77import re_assert
88
99from pre_commit import error_handler
10+ from pre_commit .errors import FatalError
1011from pre_commit .store import Store
1112from pre_commit .util import CalledProcessError
1213from testing .util import cmd_output_mocked_pre_commit_home
@@ -26,7 +27,7 @@ def test_error_handler_no_exception(mocked_log_and_exit):
2627
2728
2829def test_error_handler_fatal_error (mocked_log_and_exit ):
29- exc = error_handler . FatalError ('just a test' )
30+ exc = FatalError ('just a test' )
3031 with error_handler .error_handler ():
3132 raise exc
3233
@@ -44,7 +45,7 @@ def test_error_handler_fatal_error(mocked_log_and_exit):
4445 r' File ".+tests.error_handler_test.py", line \d+, '
4546 r'in test_error_handler_fatal_error\n'
4647 r' raise exc\n'
47- r'(pre_commit\.error_handler \.)?FatalError: just a test\n' ,
48+ r'(pre_commit\.errors \.)?FatalError: just a test\n' ,
4849 )
4950 pattern .assert_matches (mocked_log_and_exit .call_args [0 ][2 ])
5051
@@ -99,11 +100,11 @@ def test_log_and_exit(cap_out, mock_store_dir):
99100 tb = (
100101 'Traceback (most recent call last):\n '
101102 ' File "<stdin>", line 2, in <module>\n '
102- 'pre_commit.error_handler .FatalError: hai\n '
103+ 'pre_commit.errors .FatalError: hai\n '
103104 )
104105
105106 with pytest .raises (SystemExit ):
106- error_handler ._log_and_exit ('msg' , error_handler . FatalError ('hai' ), tb )
107+ error_handler ._log_and_exit ('msg' , FatalError ('hai' ), tb )
107108
108109 printed = cap_out .get ()
109110 log_file = os .path .join (mock_store_dir , 'pre-commit.log' )
@@ -133,7 +134,7 @@ def test_log_and_exit(cap_out, mock_store_dir):
133134 r'```\n'
134135 r'Traceback \(most recent call last\):\n'
135136 r' File "<stdin>", line 2, in <module>\n'
136- r'pre_commit\.error_handler \.FatalError: hai\n'
137+ r'pre_commit\.errors \.FatalError: hai\n'
137138 r'```\n' ,
138139 )
139140 pattern .assert_matches (logged )
0 commit comments