1
0
forked from VimPlug/jedi

Make it possible to be able to test errors for refactorings

This commit is contained in:
Dave Halter
2020-02-14 14:15:57 +01:00
parent 0a3ff6bd70
commit dbf88f2750
3 changed files with 22 additions and 7 deletions

View File

@@ -4,6 +4,7 @@ import pytest
from . import helpers
from jedi.inference.utils import indent_block
from jedi import RefactoringError
def assert_case_equal(case, actual, desired):
@@ -61,5 +62,10 @@ def test_refactor(refactor_case):
:type refactor_case: :class:`.refactor.RefactoringCase`
"""
diff = refactor_case.calculate_diff()
assert_case_equal(refactor_case, diff, refactor_case.desired_diff)
if refactor_case.is_error:
with pytest.raises(RefactoringError) as e:
refactor_case.calculate_diff()
assert e.value.args[0] == refactor_case.desired_diff.strip()
else:
diff = refactor_case.calculate_diff()
assert_case_equal(refactor_case, diff, refactor_case.desired_diff)