Skip to content

Commit 858a28d

Browse files
fix: apply ruff formatting to simplified traceback test
- Split long AirbyteLogMessage constructor across multiple lines - Wrap long assert statements with parentheses for readability - Addresses Ruff Format Check CI failure Co-Authored-By: AJ Steers <[email protected]>
1 parent c78206f commit 858a28d

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

unit_tests/test_exception_handler.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,9 @@ def test_uncaught_exception_handler():
5656

5757
expected_log_message = AirbyteMessage(
5858
type=MessageType.LOG,
59-
log=AirbyteLogMessage(level=Level.FATAL, message=f"{exception_message}\n{base_exception_trace}"),
59+
log=AirbyteLogMessage(
60+
level=Level.FATAL, message=f"{exception_message}\n{base_exception_trace}"
61+
),
6062
)
6163

6264
expected_trace_message = AirbyteMessage(
@@ -84,10 +86,18 @@ def test_uncaught_exception_handler():
8486
log_output, trace_output = stdout_lines
8587

8688
out_log_message = AirbyteMessageSerializer.load(json.loads(log_output))
87-
assert base_exception_trace in out_log_message.log.message, "Log message should contain expected traceback format"
88-
assert exception_message in out_log_message.log.message, "Log message should contain expected exception message"
89+
assert base_exception_trace in out_log_message.log.message, (
90+
"Log message should contain expected traceback format"
91+
)
92+
assert exception_message in out_log_message.log.message, (
93+
"Log message should contain expected exception message"
94+
)
8995

9096
out_trace_message = AirbyteMessageSerializer.load(json.loads(trace_output))
9197
assert out_trace_message.trace.emitted_at > 0
92-
assert base_exception_trace in out_trace_message.trace.error.stack_trace, "Trace message should contain expected traceback format"
93-
assert out_trace_message.trace.error.internal_message == exception_message, "Trace message should contain expected exception message"
98+
assert base_exception_trace in out_trace_message.trace.error.stack_trace, (
99+
"Trace message should contain expected traceback format"
100+
)
101+
assert out_trace_message.trace.error.internal_message == exception_message, (
102+
"Trace message should contain expected exception message"
103+
)

0 commit comments

Comments
 (0)