Skip to content

Commit 7448d74

Browse files
authored
Fix chicken & egg problem when function enable_loglevel_in_test_module is called (#27)
1 parent fae7d70 commit 7448d74

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

squape/report.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,11 @@ def set_level(level) -> None:
6060

6161
LOGLEVEL = set_level(LogLevel.LOG)
6262

63+
_test_log = test.log
64+
_test_warning = test.warning
65+
_test_fail = test.fail
66+
_test_fatal = test.fatal
67+
6368

6469
def __is_level_enabled(level: LogLevel) -> bool:
6570
"""Checks the given log level against the currently set LOGLEVEL
@@ -94,7 +99,7 @@ def debug(msg: str, details: str = "") -> None:
9499
if __is_level_enabled(LogLevel.DEBUG):
95100
test.fixateResultContext(1)
96101
try:
97-
test.log(f"[DEBUG] {msg}", details)
102+
_test_log(f"[DEBUG] {msg}", details)
98103
finally:
99104
test.restoreResultContext()
100105

@@ -119,7 +124,7 @@ def log(msg: str, details: str = "") -> None:
119124
if __is_level_enabled(LogLevel.LOG):
120125
test.fixateResultContext(1)
121126
try:
122-
test.log(msg, details)
127+
_test_log(msg, details)
123128
finally:
124129
test.restoreResultContext()
125130

@@ -145,7 +150,7 @@ def warning(msg: str, details: str = "") -> None:
145150
if __is_level_enabled(LogLevel.WARNING):
146151
test.fixateResultContext(1)
147152
try:
148-
test.warning(msg, details)
153+
_test_warning(msg, details)
149154
finally:
150155
test.restoreResultContext()
151156

@@ -170,7 +175,7 @@ def fail(msg: str, details: str = "") -> None:
170175
if __is_level_enabled(LogLevel.FAIL):
171176
test.fixateResultContext(1)
172177
try:
173-
test.fail(msg, details)
178+
_test_fail(msg, details)
174179
finally:
175180
test.restoreResultContext()
176181

@@ -199,7 +204,7 @@ def fatal(msg: str, details: str = "") -> None:
199204
test.fixateResultContext(1)
200205
try:
201206
squish.testSettings.throwOnFailure = True
202-
test.fatal(msg, details)
207+
_test_fatal(msg, details)
203208
finally:
204209
test.restoreResultContext()
205210

0 commit comments

Comments
 (0)