Allow to wait for logs in log_helper

esl · Apr 23, 2024 · 14f3b46 · 14f3b46
1 parent c66c8fc
commit 14f3b46
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/test/log_helper.hrl b/test/log_helper.hrl
@@ -2,25 +2,34 @@
 %% For set-up and tear-down, see log_helper.erl
 
 -define(assertNoLog(LevelPattern, MsgPattern),
-        case ?receiveLog(LevelPattern, MsgPattern) of
+        ?assertNoLog(LevelPattern, MsgPattern, 0)).
+
+-define(assertNoLog(LevelPattern, MsgPattern, Timeout),
+        case ?receiveLog(LevelPattern, MsgPattern, Timeout) of
             no_log -> ok;
             _ -> ct:fail("Received unexpected log")
         end).
 
 -define(assertLog(LevelPattern, MsgPattern),
-        case ?receiveLog(LevelPattern, MsgPattern) of
+        ?assertLog(LevelPattern, MsgPattern, 0)).
+
+-define(assertLog(LevelPattern, MsgPattern, Timeout),
+        case ?receiveLog(LevelPattern, MsgPattern, Timeout) of
             no_log -> ct:fail("Expected log not received");
             _ -> ok
         end).
 
 -define(receiveLog(LevelPattern, MsgPattern),
+        ?receiveLog(LevelPattern, MsgPattern, 0)).
+
+-define(receiveLog(LevelPattern, MsgPattern, Timeout),
         ?wrap(receive
                   {log, #{level := Level = LevelPattern,
                           msg := {report, Msg = MsgPattern}}} ->
                       ct:log("Received ~p log: ~p", [Level, Msg]),
                       {Level, Msg}
               after
-                  0 -> no_log
+                  Timeout -> no_log
               end)).
 
 %% Wrap in a fun to avoid unsafe variables