1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
from __future__ import annotations
import inspect
from cleo.ui.exception_trace.frame import Frame
from tests.ui.exception_trace.helpers import nested_exception
from tests.ui.exception_trace.helpers import simple_exception
def test_frame() -> None:
try:
simple_exception()
except ValueError as e:
assert e.__traceback__ is not None
frame_info = inspect.getinnerframes(e.__traceback__)[0]
frame = Frame(frame_info)
same_frame = Frame(frame_info)
assert frame_info.frame == frame.frame
assert frame.lineno == 12
assert frame.filename == __file__
assert frame.function == "test_frame"
assert frame.line == " simple_exception()\n"
with open(__file__) as f:
assert f.read() == frame.file_content
assert repr(frame) == f"<Frame {__file__}, test_frame, 12>"
try:
nested_exception()
except Exception as e:
assert e.__traceback__ is not None
frame_info = inspect.getinnerframes(e.__traceback__)[0]
other_frame = Frame(frame_info)
assert same_frame == frame
assert other_frame != frame
assert hash(same_frame) == hash(frame)
assert hash(other_frame) != hash(frame)
def test_frame_with_no_context_should_return_empty_line() -> None:
frame = Frame(
inspect.FrameInfo(None, "filename.py", 123, "function", None, 3) # type: ignore[arg-type]
)
assert frame.line == ""
|
