rewrite from lager
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1927 lines
87 KiB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
  1. %% -------------------------------------------------------------------
  2. %%
  3. %% Copyright (c) 2011-2017 Basho Technologies, Inc.
  4. %%
  5. %% This file is provided to you under the Apache License,
  6. %% Version 2.0 (the "License"); you may not use this file
  7. %% except in compliance with the License. You may obtain
  8. %% a copy of the License at
  9. %%
  10. %% http://www.apache.org/licenses/LICENSE-2.0
  11. %%
  12. %% Unless required by applicable law or agreed to in writing,
  13. %% software distributed under the License is distributed on an
  14. %% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. %% KIND, either express or implied. See the License for the
  16. %% specific language governing permissions and limitations
  17. %% under the License.
  18. %%
  19. %% -------------------------------------------------------------------
  20. -module(lager_test_backend).
  21. -behaviour(gen_event).
  22. -export([init/1, handle_call/2, handle_event/2, handle_info/2, terminate/2,
  23. code_change/3]).
  24. -include("lager.hrl").
  25. -define(TEST_SINK_NAME, '__lager_test_sink'). %% <-- used by parse transform
  26. -define(TEST_SINK_EVENT, '__lager_test_sink_lager_event'). %% <-- used by lager API calls and internals for gen_event
  27. -record(state, {
  28. level :: list(),
  29. buffer :: list(),
  30. ignored :: term()
  31. }).
  32. -compile({parse_transform, lager_transform}).
  33. -ifdef(TEST).
  34. -include_lib("eunit/include/eunit.hrl").
  35. -record(test, {
  36. attrs :: list(),
  37. format :: list(),
  38. args :: list()
  39. }).
  40. -export([
  41. count/0,
  42. count_ignored/0,
  43. flush/0,
  44. message_stuffer/3,
  45. pop/0,
  46. pop_ignored/0,
  47. print_state/0,
  48. get_buffer/0
  49. ]).
  50. -endif.
  51. init(Level) ->
  52. {ok, #state{level = lager_util:config_to_mask(Level), buffer = [], ignored = []}}.
  53. handle_call(count, #state{buffer = Buffer} = State) ->
  54. {ok, length(Buffer), State};
  55. handle_call(count_ignored, #state{ignored = Ignored} = State) ->
  56. {ok, length(Ignored), State};
  57. handle_call(flush, State) ->
  58. {ok, ok, State#state{buffer = [], ignored = []}};
  59. handle_call(pop, #state{buffer = Buffer} = State) ->
  60. case Buffer of
  61. [] ->
  62. {ok, undefined, State};
  63. [H | T] ->
  64. {ok, H, State#state{buffer = T}}
  65. end;
  66. handle_call(pop_ignored, #state{ignored = Ignored} = State) ->
  67. case Ignored of
  68. [] ->
  69. {ok, undefined, State};
  70. [H | T] ->
  71. {ok, H, State#state{ignored = T}}
  72. end;
  73. handle_call(get_buffer, #state{buffer = Buffer} = State) ->
  74. {ok, Buffer, State};
  75. handle_call(get_loglevel, #state{level = Level} = State) ->
  76. {ok, Level, State};
  77. handle_call({set_loglevel, Level}, State) ->
  78. {ok, ok, State#state{level = lager_util:config_to_mask(Level)}};
  79. handle_call(print_state, State) ->
  80. spawn(fun() -> lager:info("State ~p", [lager:pr(State, ?MODULE)]) end),
  81. timer:sleep(100),
  82. {ok, ok, State};
  83. handle_call(print_bad_state, State) ->
  84. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, ?MODULE)]) end),
  85. timer:sleep(100),
  86. {ok, ok, State};
  87. handle_call(_Request, State) ->
  88. {ok, ok, State}.
  89. handle_event({log, Msg},
  90. #state{level = LogLevel, buffer = Buffer, ignored = Ignored} = State) ->
  91. case lager_util:is_loggable(Msg, LogLevel, ?MODULE) of
  92. true ->
  93. {ok, State#state{buffer = Buffer ++
  94. [{lager_msg:severity_as_int(Msg),
  95. lager_msg:datetime(Msg),
  96. lager_msg:message(Msg), lager_msg:metadata(Msg)}]}};
  97. _ ->
  98. {ok, State#state{ignored = Ignored ++ [Msg]}}
  99. end;
  100. handle_event(_Event, State) ->
  101. {ok, State}.
  102. handle_info(_Info, State) ->
  103. {ok, State}.
  104. terminate(_Reason, _State) ->
  105. ok.
  106. code_change(_OldVsn, State, _Extra) ->
  107. {ok, State}.
  108. -ifdef(TEST).
  109. pop() ->
  110. pop(lager_event).
  111. pop_ignored() ->
  112. pop_ignored(lager_event).
  113. get_buffer() ->
  114. get_buffer(lager_event).
  115. count() ->
  116. count(lager_event).
  117. count_ignored() ->
  118. count_ignored(lager_event).
  119. flush() ->
  120. flush(lager_event).
  121. print_state() ->
  122. print_state(lager_event).
  123. print_bad_state() ->
  124. print_bad_state(lager_event).
  125. pop(Sink) ->
  126. gen_event:call(Sink, ?MODULE, pop).
  127. pop_ignored(Sink) ->
  128. gen_event:call(Sink, ?MODULE, pop_ignored).
  129. get_buffer(Sink) ->
  130. gen_event:call(Sink, ?MODULE, get_buffer).
  131. count(Sink) ->
  132. gen_event:call(Sink, ?MODULE, count).
  133. count_ignored(Sink) ->
  134. gen_event:call(Sink, ?MODULE, count_ignored).
  135. flush(Sink) ->
  136. gen_event:call(Sink, ?MODULE, flush).
  137. print_state(Sink) ->
  138. gen_event:call(Sink, ?MODULE, print_state).
  139. print_bad_state(Sink) ->
  140. gen_event:call(Sink, ?MODULE, print_bad_state).
  141. not_running_test() ->
  142. ?assertEqual({error, lager_not_running}, lager:log(info, self(), "not running")).
  143. lager_test_() ->
  144. {foreach,
  145. fun setup/0,
  146. fun cleanup/1,
  147. [
  148. {"observe that there is nothing up my sleeve",
  149. fun() ->
  150. ?assertEqual(undefined, pop()),
  151. ?assertEqual(0, count())
  152. end
  153. },
  154. {"test sink not running",
  155. fun() ->
  156. ?assertEqual({error, {sink_not_configured, test}}, lager:log(test, info, self(), "~p", "not running"))
  157. end
  158. },
  159. {"logging works",
  160. fun() ->
  161. lager:warning("test message"),
  162. ?assertEqual(1, count()),
  163. {Level, _Time, Message, _Metadata} = pop(),
  164. ?assertMatch(Level, lager_util:level_to_num(warning)),
  165. ?assertEqual("test message", Message),
  166. ok
  167. end
  168. },
  169. {"logging with macro works",
  170. fun() ->
  171. ?lager_warning("test message", []),
  172. ?assertEqual(1, count()),
  173. {Level, _Time, Message, _Metadata} = pop(),
  174. ?assertMatch(Level, lager_util:level_to_num(warning)),
  175. ?assertEqual("test message", Message),
  176. ok
  177. end
  178. },
  179. {"unsafe logging works",
  180. fun() ->
  181. lager:warning_unsafe("test message"),
  182. ?assertEqual(1, count()),
  183. {Level, _Time, Message, _Metadata} = pop(),
  184. ?assertMatch(Level, lager_util:level_to_num(warning)),
  185. ?assertEqual("test message", Message),
  186. ok
  187. end
  188. },
  189. {"logging with arguments works",
  190. fun() ->
  191. lager:warning("test message ~p", [self()]),
  192. ?assertEqual(1, count()),
  193. {Level, _Time, Message, _Metadata} = pop(),
  194. ?assertMatch(Level, lager_util:level_to_num(warning)),
  195. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  196. ok
  197. end
  198. },
  199. {"logging with macro and arguments works",
  200. fun() ->
  201. ?lager_warning("test message ~p", [self()]),
  202. ?assertEqual(1, count()),
  203. {Level, _Time, Message, _Metadata} = pop(),
  204. ?assertMatch(Level, lager_util:level_to_num(warning)),
  205. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  206. ok
  207. end
  208. },
  209. {"unsafe logging with args works",
  210. fun() ->
  211. lager:warning_unsafe("test message ~p", [self()]),
  212. ?assertEqual(1, count()),
  213. {Level, _Time, Message, _Metadata} = pop(),
  214. ?assertMatch(Level, lager_util:level_to_num(warning)),
  215. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  216. ok
  217. end
  218. },
  219. {"logging works from inside a begin/end block",
  220. fun() ->
  221. ?assertEqual(0, count()),
  222. begin
  223. lager:warning("test message 2")
  224. end,
  225. ?assertEqual(1, count()),
  226. ok
  227. end
  228. },
  229. {"logging works from inside a list comprehension",
  230. fun() ->
  231. ?assertEqual(0, count()),
  232. [lager:warning("test message") || _N <- lists:seq(1, 10)],
  233. ?assertEqual(10, count()),
  234. ok
  235. end
  236. },
  237. {"logging works from a begin/end block inside a list comprehension",
  238. fun() ->
  239. ?assertEqual(0, count()),
  240. [begin lager:warning("test message") end || _N <- lists:seq(1, 10)],
  241. ?assertEqual(10, count()),
  242. ok
  243. end
  244. },
  245. {"logging works from a nested list comprehension",
  246. fun() ->
  247. ?assertEqual(0, count()),
  248. [[lager:warning("test message") || _N <- lists:seq(1, 10)] ||
  249. _I <- lists:seq(1, 10)],
  250. ?assertEqual(100, count()),
  251. ok
  252. end
  253. },
  254. {"logging with only metadata works",
  255. fun() ->
  256. ?assertEqual(0, count()),
  257. lager:warning([{just, metadata}]),
  258. lager:warning([{just, metadata}, {foo, bar}]),
  259. ?assertEqual(2, count()),
  260. ok
  261. end
  262. },
  263. {"variables inplace of literals in logging statements work",
  264. fun() ->
  265. ?assertEqual(0, count()),
  266. Attr = [{a, alpha}, {b, beta}],
  267. Fmt = "format ~p",
  268. Args = [world],
  269. lager:info(Attr, "hello"),
  270. lager:info(Attr, "hello ~p", [world]),
  271. lager:info(Fmt, [world]),
  272. lager:info("hello ~p", Args),
  273. lager:info(Attr, "hello ~p", Args),
  274. lager:info([{d, delta}, {g, gamma}], Fmt, Args),
  275. ?assertEqual(6, count()),
  276. {Level, _Time, Message, Metadata} = pop(),
  277. ?assertMatch([{a, alpha}, {b, beta} | _], Metadata),
  278. ?assertEqual("hello", lists:flatten(Message)),
  279. {Level, _Time2, Message2, _Metadata2} = pop(),
  280. ?assertEqual("hello world", lists:flatten(Message2)),
  281. {Level, _Time3, Message3, _Metadata3} = pop(),
  282. ?assertEqual("format world", lists:flatten(Message3)),
  283. {Level, _Time4, Message4, _Metadata4} = pop(),
  284. ?assertEqual("hello world", lists:flatten(Message4)),
  285. {Level, _Time5, Message5, _Metadata5} = pop(),
  286. ?assertEqual("hello world", lists:flatten(Message5)),
  287. {Level, _Time6, Message6, Metadata6} = pop(),
  288. ?assertMatch([{d, delta}, {g, gamma} | _], Metadata6),
  289. ?assertEqual("format world", lists:flatten(Message6)),
  290. ok
  291. end
  292. },
  293. {"list comprehension inplace of literals in logging statements work",
  294. fun() ->
  295. ?assertEqual(0, count()),
  296. Attr = [{a, alpha}, {b, beta}],
  297. Fmt = "format ~p",
  298. Args = [world],
  299. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello"),
  300. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello ~p", [{atom, X} || X <- Args]),
  301. lager:info([X || X <- Fmt], [world]),
  302. lager:info("hello ~p", [{atom, X} || X <- Args]),
  303. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello ~p", [{atom, X} || X <- Args]),
  304. lager:info([{d, delta}, {g, gamma}], Fmt, [{atom, X} || X <- Args]),
  305. ?assertEqual(6, count()),
  306. {Level, _Time, Message, Metadata} = pop(),
  307. ?assertMatch([{a, "alpha"}, {b, "beta"} | _], Metadata),
  308. ?assertEqual("hello", lists:flatten(Message)),
  309. {Level, _Time2, Message2, _Metadata2} = pop(),
  310. ?assertEqual("hello {atom,world}", lists:flatten(Message2)),
  311. {Level, _Time3, Message3, _Metadata3} = pop(),
  312. ?assertEqual("format world", lists:flatten(Message3)),
  313. {Level, _Time4, Message4, _Metadata4} = pop(),
  314. ?assertEqual("hello {atom,world}", lists:flatten(Message4)),
  315. {Level, _Time5, Message5, _Metadata5} = pop(),
  316. ?assertEqual("hello {atom,world}", lists:flatten(Message5)),
  317. {Level, _Time6, Message6, Metadata6} = pop(),
  318. ?assertMatch([{d, delta}, {g, gamma} | _], Metadata6),
  319. ?assertEqual("format {atom,world}", lists:flatten(Message6)),
  320. ok
  321. end
  322. },
  323. {"function calls inplace of literals in logging statements work",
  324. fun() ->
  325. ?assertEqual(0, count()),
  326. put(attrs, [{a, alpha}, {b, beta}]),
  327. put(format, "format ~p"),
  328. put(args, [world]),
  329. lager:info(get(attrs), "hello"),
  330. lager:info(get(attrs), "hello ~p", get(args)),
  331. lager:info(get(format), [world]),
  332. lager:info("hello ~p", erlang:get(args)),
  333. lager:info(fun() -> get(attrs) end(), "hello ~p", get(args)),
  334. lager:info([{d, delta}, {g, gamma}], get(format), get(args)),
  335. ?assertEqual(6, count()),
  336. {Level, _Time, Message, Metadata} = pop(),
  337. ?assertMatch([{a, alpha}, {b, beta} | _], Metadata),
  338. ?assertEqual("hello", lists:flatten(Message)),
  339. {Level, _Time2, Message2, _Metadata2} = pop(),
  340. ?assertEqual("hello world", lists:flatten(Message2)),
  341. {Level, _Time3, Message3, _Metadata3} = pop(),
  342. ?assertEqual("format world", lists:flatten(Message3)),
  343. {Level, _Time4, Message4, _Metadata4} = pop(),
  344. ?assertEqual("hello world", lists:flatten(Message4)),
  345. {Level, _Time5, Message5, _Metadata5} = pop(),
  346. ?assertEqual("hello world", lists:flatten(Message5)),
  347. {Level, _Time6, Message6, Metadata6} = pop(),
  348. ?assertMatch([{d, delta}, {g, gamma} | _], Metadata6),
  349. ?assertEqual("format world", lists:flatten(Message6)),
  350. ok
  351. end
  352. },
  353. {"record fields inplace of literals in logging statements work",
  354. fun() ->
  355. ?assertEqual(0, count()),
  356. Test = #test{attrs = [{a, alpha}, {b, beta}], format = "format ~p", args = [world]},
  357. lager:info(Test#test.attrs, "hello"),
  358. lager:info(Test#test.attrs, "hello ~p", Test#test.args),
  359. lager:info(Test#test.format, [world]),
  360. lager:info("hello ~p", Test#test.args),
  361. lager:info(Test#test.attrs, "hello ~p", Test#test.args),
  362. lager:info([{d, delta}, {g, gamma}], Test#test.format, Test#test.args),
  363. ?assertEqual(6, count()),
  364. {Level, _Time, Message, Metadata} = pop(),
  365. ?assertMatch([{a, alpha}, {b, beta} | _], Metadata),
  366. ?assertEqual("hello", lists:flatten(Message)),
  367. {Level, _Time2, Message2, _Metadata2} = pop(),
  368. ?assertEqual("hello world", lists:flatten(Message2)),
  369. {Level, _Time3, Message3, _Metadata3} = pop(),
  370. ?assertEqual("format world", lists:flatten(Message3)),
  371. {Level, _Time4, Message4, _Metadata4} = pop(),
  372. ?assertEqual("hello world", lists:flatten(Message4)),
  373. {Level, _Time5, Message5, _Metadata5} = pop(),
  374. ?assertEqual("hello world", lists:flatten(Message5)),
  375. {Level, _Time6, Message6, Metadata6} = pop(),
  376. ?assertMatch([{d, delta}, {g, gamma} | _], Metadata6),
  377. ?assertEqual("format world", lists:flatten(Message6)),
  378. ok
  379. end
  380. },
  381. {"log messages below the threshold are ignored",
  382. fun() ->
  383. ?assertEqual(0, count()),
  384. lager:debug("this message will be ignored"),
  385. ?assertEqual(0, count()),
  386. ?assertEqual(0, count_ignored()),
  387. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(debug)), []}),
  388. lager:debug("this message should be ignored"),
  389. ?assertEqual(0, count()),
  390. ?assertEqual(1, count_ignored()),
  391. lager:set_loglevel(?MODULE, debug),
  392. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  393. lager:debug("this message should be logged"),
  394. ?assertEqual(1, count()),
  395. ?assertEqual(1, count_ignored()),
  396. ?assertEqual(debug, lager:get_loglevel(?MODULE)),
  397. ok
  398. end
  399. },
  400. {"tracing works",
  401. fun() ->
  402. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  403. ok = lager:info("hello world"),
  404. ?assertEqual(0, count()),
  405. lager:trace(?MODULE, [{module, ?MODULE}], debug),
  406. ?assertMatch({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, _}, lager_config:get(loglevel)),
  407. %% elegible for tracing
  408. ok = lager:info("hello world"),
  409. %% NOT elegible for tracing
  410. ok = lager:log(info, [{pid, self()}], "hello world"),
  411. ?assertEqual(1, count()),
  412. ok
  413. end
  414. },
  415. {"tracing works with custom attributes",
  416. fun() ->
  417. lager:set_loglevel(?MODULE, error),
  418. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  419. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  420. lager:info([{requestid, 6}], "hello world"),
  421. ?assertEqual(0, count()),
  422. lager:trace(?MODULE, [{requestid, 6}, {foo, bar}], debug),
  423. lager:info([{requestid, 6}, {foo, bar}], "hello world"),
  424. ?assertEqual(1, count()),
  425. lager:trace(?MODULE, [{requestid, '*'}], debug),
  426. lager:info([{requestid, 6}], "hello world"),
  427. ?assertEqual(2, count()),
  428. lager:clear_all_traces(),
  429. lager:info([{requestid, 6}], "hello world"),
  430. ?assertEqual(2, count()),
  431. ok
  432. end
  433. },
  434. {"tracing works with custom attributes and event stream processing",
  435. fun() ->
  436. lager:set_loglevel(?MODULE, error),
  437. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  438. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  439. lager:info([{requestid, 6}], "hello world"),
  440. ?assertEqual(0, count()),
  441. lager:trace(?MODULE, [{requestid, '>', 5}, {requestid, '<', 7}, {foo, bar}], debug),
  442. lager:info([{requestid, 5}, {foo, bar}], "hello world"),
  443. lager:info([{requestid, 6}, {foo, bar}], "hello world"),
  444. ?assertEqual(1, count()),
  445. lager:clear_all_traces(),
  446. lager:trace(?MODULE, [{requestid, '>', 8}, {foo, bar}]),
  447. lager:info([{foo, bar}], "hello world"),
  448. lager:info([{requestid, 6}], "hello world"),
  449. lager:info([{requestid, 7}], "hello world"),
  450. lager:info([{requestid, 8}], "hello world"),
  451. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  452. lager:info([{requestid, 10}], "hello world"),
  453. ?assertEqual(2, count()),
  454. lager:trace(?MODULE, [{requestid, '>', 8}]),
  455. lager:info([{foo, bar}], "hello world"),
  456. lager:info([{requestid, 6}], "hello world"),
  457. lager:info([{requestid, 7}], "hello world"),
  458. lager:info([{requestid, 8}], "hello world"),
  459. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  460. lager:info([{requestid, 10}], "hello world"),
  461. ?assertEqual(4, count()),
  462. lager:trace(?MODULE, [{foo, '=', bar}]),
  463. lager:info([{foo, bar}], "hello world"),
  464. lager:info([{requestid, 6}], "hello world"),
  465. lager:info([{requestid, 7}], "hello world"),
  466. lager:info([{requestid, 8}], "hello world"),
  467. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  468. lager:info([{requestid, 10}], "hello world"),
  469. lager:trace(?MODULE, [{fu, '!'}]),
  470. lager:info([{foo, bar}], "hello world"),
  471. lager:info([{ooh, car}], "hello world"),
  472. lager:info([{fu, bar}], "hello world"),
  473. lager:trace(?MODULE, [{fu, '*'}]),
  474. lager:info([{fu, bar}], "hello world"),
  475. ?assertEqual(10, count()),
  476. lager:clear_all_traces(),
  477. lager:info([{requestid, 6}], "hello world"),
  478. ?assertEqual(10, count()),
  479. lager:clear_all_traces(),
  480. lager:trace(?MODULE, [{requestid, '>=', 5}, {requestid, '=<', 7}], debug),
  481. lager:info([{requestid, 4}], "nope!"),
  482. lager:info([{requestid, 5}], "hello world"),
  483. lager:info([{requestid, 7}], "hello world again"),
  484. ?assertEqual(12, count()),
  485. lager:clear_all_traces(),
  486. lager:trace(?MODULE, [{foo, '!=', bar}]),
  487. lager:info([{foo, bar}], "hello world"),
  488. ?assertEqual(12, count()),
  489. lager:info([{foo, baz}], "blarg"),
  490. ?assertEqual(13, count()),
  491. lager:clear_all_traces(),
  492. lager:trace(?MODULE, [{all, [{foo, '=', bar}, {null, false}]}]),
  493. lager:info([{foo, bar}], "should not be logged"),
  494. ?assertEqual(13, count()),
  495. lager:clear_all_traces(),
  496. lager:trace(?MODULE, [{any, [{foo, '=', bar}, {null, true}]}]),
  497. lager:info([{foo, qux}], "should be logged"),
  498. ?assertEqual(14, count()),
  499. ok
  500. end
  501. },
  502. {"tracing custom attributes works with event stream processing statistics and reductions",
  503. fun() ->
  504. lager:set_loglevel(?MODULE, error),
  505. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  506. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  507. lager:info([{requestid, 6}], "hello world"),
  508. ?assertEqual(0, count()),
  509. lager:trace(?MODULE, [{beta, '*'}]),
  510. lager:trace(?MODULE, [{meta, "data"}]),
  511. lager:info([{meta, "data"}], "hello world"),
  512. lager:info([{beta, 2}], "hello world"),
  513. lager:info([{beta, 2.1}, {foo, bar}], "hello world"),
  514. lager:info([{meta, <<"data">>}], "hello world"),
  515. ?assertEqual(8, ?DEFAULT_TRACER:info(input)),
  516. ?assertEqual(6, ?DEFAULT_TRACER:info(output)),
  517. ?assertEqual(2, ?DEFAULT_TRACER:info(filter)),
  518. lager:clear_all_traces(),
  519. lager:trace(?MODULE, [{meta, "data"}]),
  520. lager:trace(?MODULE, [{beta, '>', 2}, {beta, '<', 2.12}]),
  521. lager:info([{meta, "data"}], "hello world"),
  522. lager:info([{beta, 2}], "hello world"),
  523. lager:info([{beta, 2.1}, {foo, bar}], "hello world"),
  524. lager:info([{meta, <<"data">>}], "hello world"),
  525. ?assertEqual(8, ?DEFAULT_TRACER:info(input)),
  526. ?assertEqual(4, ?DEFAULT_TRACER:info(output)),
  527. ?assertEqual(4, ?DEFAULT_TRACER:info(filter)),
  528. lager:clear_all_traces(),
  529. lager:trace_console([{beta, '>', 2}, {meta, "data"}]),
  530. lager:trace_console([{beta, '>', 2}, {beta, '<', 2.12}]),
  531. Reduced = {all, [{any, [{beta, '<', 2.12}, {meta, '=', "data"}]},
  532. {beta, '>', 2}]},
  533. ?assertEqual(Reduced, ?DEFAULT_TRACER:info('query')),
  534. lager:clear_all_traces(),
  535. lager:info([{requestid, 6}], "hello world"),
  536. ?assertEqual(5, count()),
  537. ok
  538. end
  539. },
  540. {"persistent traces work",
  541. fun() ->
  542. ?assertEqual(0, count()),
  543. lager:debug([{foo, bar}], "hello world"),
  544. ?assertEqual(0, count()),
  545. application:stop(lager),
  546. application:set_env(lager, traces, [{lager_test_backend, [{foo, bar}], debug}]),
  547. lager:start(),
  548. timer:sleep(5),
  549. flush(),
  550. lager:debug([{foo, bar}], "hello world"),
  551. ?assertEqual(1, count()),
  552. application:unset_env(lager, traces),
  553. ok
  554. end
  555. },
  556. {"tracing honors loglevel",
  557. fun() ->
  558. lager:set_loglevel(?MODULE, error),
  559. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  560. {ok, T} = lager:trace(?MODULE, [{module, ?MODULE}], notice),
  561. ok = lager:info("hello world"),
  562. ?assertEqual(0, count()),
  563. ok = lager:notice("hello world"),
  564. ?assertEqual(1, count()),
  565. lager:stop_trace(T),
  566. ok = lager:notice("hello world"),
  567. ?assertEqual(1, count()),
  568. ok
  569. end
  570. },
  571. {"stopped trace stops and removes its event handler - default sink (gh#267)",
  572. {timeout, 10,
  573. fun() ->
  574. Sink = ?DEFAULT_SINK,
  575. StartHandlers = gen_event:which_handlers(Sink),
  576. {_, T0} = lager_config:get({Sink, loglevel}),
  577. StartGlobal = lager_config:global_get(handlers),
  578. ?assertEqual([], T0),
  579. {ok, TestTrace1} = lager:trace_file("/tmp/test", [{a, b}]),
  580. MidHandlers = gen_event:which_handlers(Sink),
  581. {ok, TestTrace2} = lager:trace_file("/tmp/test", [{c, d}]),
  582. MidHandlers = gen_event:which_handlers(Sink),
  583. ?assertEqual(length(StartHandlers) + 1, length(MidHandlers)),
  584. MidGlobal = lager_config:global_get(handlers),
  585. ?assertEqual(length(StartGlobal) + 1, length(MidGlobal)),
  586. {_, T1} = lager_config:get({Sink, loglevel}),
  587. ?assertEqual(2, length(T1)),
  588. ok = lager:stop_trace(TestTrace1),
  589. {_, T2} = lager_config:get({Sink, loglevel}),
  590. ?assertEqual(1, length(T2)),
  591. ?assertEqual(length(StartHandlers) + 1, length(
  592. gen_event:which_handlers(Sink))),
  593. ?assertEqual(length(StartGlobal) + 1, length(lager_config:global_get(handlers))),
  594. ok = lager:stop_trace(TestTrace2),
  595. EndHandlers = gen_event:which_handlers(Sink),
  596. EndGlobal = lager_config:global_get(handlers),
  597. {_, T3} = lager_config:get({Sink, loglevel}),
  598. ?assertEqual([], T3),
  599. ?assertEqual(StartHandlers, EndHandlers),
  600. ?assertEqual(StartGlobal, EndGlobal),
  601. ok
  602. end}
  603. },
  604. {"record printing works",
  605. fun() ->
  606. print_state(),
  607. {Level, _Time, Message, _Metadata} = pop(),
  608. ?assertMatch(Level, lager_util:level_to_num(info)),
  609. {mask, Mask} = lager_util:config_to_mask(info),
  610. ?assertEqual("State #state{level={mask," ++ integer_to_list(Mask) ++ "},buffer=[],ignored=[]}", lists:flatten(Message)),
  611. ok
  612. end
  613. },
  614. {"record printing fails gracefully",
  615. fun() ->
  616. print_bad_state(),
  617. {Level, _Time, Message, _Metadata} = pop(),
  618. ?assertMatch(Level, lager_util:level_to_num(info)),
  619. ?assertEqual("State {state,1}", lists:flatten(Message)),
  620. ok
  621. end
  622. },
  623. {"record printing fails gracefully when no lager_record attribute",
  624. fun() ->
  625. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, lager)]) end),
  626. timer:sleep(100),
  627. {Level, _Time, Message, _Metadata} = pop(),
  628. ?assertMatch(Level, lager_util:level_to_num(info)),
  629. ?assertEqual("State {state,1}", lists:flatten(Message)),
  630. ok
  631. end
  632. },
  633. {"record printing fails gracefully when input is not a tuple",
  634. fun() ->
  635. spawn(fun() -> lager:info("State ~p", [lager:pr(ok, lager)]) end),
  636. timer:sleep(100),
  637. {Level, _Time, Message, _Metadata} = pop(),
  638. ?assertMatch(Level, lager_util:level_to_num(info)),
  639. ?assertEqual("State ok", lists:flatten(Message)),
  640. ok
  641. end
  642. },
  643. {"record printing fails gracefully when module is invalid",
  644. fun() ->
  645. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, not_a_module)]) end),
  646. timer:sleep(1000),
  647. {Level, _Time, Message, _Metadata} = pop(),
  648. ?assertMatch(Level, lager_util:level_to_num(info)),
  649. ?assertEqual("State {state,1}", lists:flatten(Message)),
  650. ok
  651. end
  652. },
  653. {"installing a new handler adjusts the global loglevel if necessary",
  654. fun() ->
  655. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  656. supervisor:start_child(lager_handler_watcher_sup, [lager_event, {?MODULE, foo}, debug]),
  657. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  658. ok
  659. end
  660. },
  661. {"metadata in the process dictionary works",
  662. fun() ->
  663. lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}]),
  664. lager:info("I sing the animal kingdom electric!"),
  665. {_Level, _Time, _Message, Metadata} = pop(),
  666. ?assertEqual(gravid, proplists:get_value(platypus, Metadata)),
  667. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata)),
  668. ?assertEqual(erroneous, proplists:get_value(duck, Metadata)),
  669. ?assertEqual(undefined, proplists:get_value(eagle, Metadata)),
  670. lager:md([{platypus, gravid}, {sloth, hirsute}, {eagle, superincumbent}]),
  671. lager:info("I sing the animal kingdom dielectric!"),
  672. {_Level2, _Time2, _Message2, Metadata2} = pop(),
  673. ?assertEqual(gravid, proplists:get_value(platypus, Metadata2)),
  674. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata2)),
  675. ?assertEqual(undefined, proplists:get_value(duck, Metadata2)),
  676. ?assertEqual(superincumbent, proplists:get_value(eagle, Metadata2)),
  677. ok
  678. end
  679. },
  680. {"unsafe messages really are not truncated",
  681. fun() ->
  682. lager:info_unsafe("doom, doom has come upon you all ~p", [string:copies("doom", 1500)]),
  683. {_, _, Msg, _Metadata} = pop(),
  684. ?assert(length(lists:flatten(Msg)) == 6035)
  685. end
  686. },
  687. {"can't store invalid metadata",
  688. fun() ->
  689. ?assertEqual(ok, lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}])),
  690. ?assertError(badarg, lager:md({flamboyant, flamingo})),
  691. ?assertError(badarg, lager:md("zookeeper zephyr")),
  692. ok
  693. end
  694. },
  695. {"dates should be local by default",
  696. fun() ->
  697. lager:warning("so long, and thanks for all the fish"),
  698. ?assertEqual(1, count()),
  699. {_Level, {_Date, Time}, _Message, _Metadata} = pop(),
  700. ?assertEqual(nomatch, binary:match(iolist_to_binary(Time), <<"UTC">>)),
  701. ok
  702. end
  703. },
  704. {"dates should be UTC if SASL is configured as UTC",
  705. fun() ->
  706. application:set_env(sasl, utc_log, true),
  707. lager:warning("so long, and thanks for all the fish"),
  708. application:set_env(sasl, utc_log, false),
  709. ?assertEqual(1, count()),
  710. {_Level, {_Date, Time}, _Message, _Metadata} = pop(),
  711. ?assertNotEqual(nomatch, binary:match(iolist_to_binary(Time), <<"UTC">>)),
  712. ok
  713. end
  714. }
  715. ]
  716. }.
  717. extra_sinks_test_() ->
  718. {foreach,
  719. fun setup_sink/0,
  720. fun cleanup/1,
  721. [
  722. {"observe that there is nothing up my sleeve",
  723. fun() ->
  724. ?assertEqual(undefined, pop(?TEST_SINK_EVENT)),
  725. ?assertEqual(0, count(?TEST_SINK_EVENT))
  726. end
  727. },
  728. {"logging works",
  729. fun() ->
  730. ?TEST_SINK_NAME:warning("test message"),
  731. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  732. {Level, _Time, Message, _Metadata} = pop(?TEST_SINK_EVENT),
  733. ?assertMatch(Level, lager_util:level_to_num(warning)),
  734. ?assertEqual("test message", Message),
  735. ok
  736. end
  737. },
  738. {"logging with arguments works",
  739. fun() ->
  740. ?TEST_SINK_NAME:warning("test message ~p", [self()]),
  741. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  742. {Level, _Time, Message, _Metadata} = pop(?TEST_SINK_EVENT),
  743. ?assertMatch(Level, lager_util:level_to_num(warning)),
  744. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  745. ok
  746. end
  747. },
  748. {"variables inplace of literals in logging statements work",
  749. fun() ->
  750. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  751. Attr = [{a, alpha}, {b, beta}],
  752. Fmt = "format ~p",
  753. Args = [world],
  754. ?TEST_SINK_NAME:info(Attr, "hello"),
  755. ?TEST_SINK_NAME:info(Attr, "hello ~p", [world]),
  756. ?TEST_SINK_NAME:info(Fmt, [world]),
  757. ?TEST_SINK_NAME:info("hello ~p", Args),
  758. ?TEST_SINK_NAME:info(Attr, "hello ~p", Args),
  759. ?TEST_SINK_NAME:info([{d, delta}, {g, gamma}], Fmt, Args),
  760. ?assertEqual(6, count(?TEST_SINK_EVENT)),
  761. {Level, _Time, Message, Metadata} = pop(?TEST_SINK_EVENT),
  762. ?assertMatch([{a, alpha}, {b, beta} | _], Metadata),
  763. ?assertEqual("hello", lists:flatten(Message)),
  764. {Level, _Time2, Message2, _Metadata2} = pop(?TEST_SINK_EVENT),
  765. ?assertEqual("hello world", lists:flatten(Message2)),
  766. {Level, _Time3, Message3, _Metadata3} = pop(?TEST_SINK_EVENT),
  767. ?assertEqual("format world", lists:flatten(Message3)),
  768. {Level, _Time4, Message4, _Metadata4} = pop(?TEST_SINK_EVENT),
  769. ?assertEqual("hello world", lists:flatten(Message4)),
  770. {Level, _Time5, Message5, _Metadata5} = pop(?TEST_SINK_EVENT),
  771. ?assertEqual("hello world", lists:flatten(Message5)),
  772. {Level, _Time6, Message6, Metadata6} = pop(?TEST_SINK_EVENT),
  773. ?assertMatch([{d, delta}, {g, gamma} | _], Metadata6),
  774. ?assertEqual("format world", lists:flatten(Message6)),
  775. ok
  776. end
  777. },
  778. {"stopped trace stops and removes its event handler - test sink (gh#267)",
  779. fun() ->
  780. Sink = ?TEST_SINK_EVENT,
  781. StartHandlers = gen_event:which_handlers(Sink),
  782. {_, T0} = lager_config:get({Sink, loglevel}),
  783. StartGlobal = lager_config:global_get(handlers),
  784. ?assertEqual([], T0),
  785. {ok, TestTrace1} = lager:trace_file("/tmp/test", [{sink, Sink}, {a, b}]),
  786. MidHandlers = gen_event:which_handlers(Sink),
  787. {ok, TestTrace2} = lager:trace_file("/tmp/test", [{sink, Sink}, {c, d}]),
  788. MidHandlers = gen_event:which_handlers(Sink),
  789. ?assertEqual(length(StartHandlers) + 1, length(MidHandlers)),
  790. MidGlobal = lager_config:global_get(handlers),
  791. ?assertEqual(length(StartGlobal) + 1, length(MidGlobal)),
  792. {_, T1} = lager_config:get({Sink, loglevel}),
  793. ?assertEqual(2, length(T1)),
  794. ok = lager:stop_trace(TestTrace1),
  795. {_, T2} = lager_config:get({Sink, loglevel}),
  796. ?assertEqual(1, length(T2)),
  797. ?assertEqual(length(StartHandlers) + 1, length(
  798. gen_event:which_handlers(Sink))),
  799. ?assertEqual(length(StartGlobal) + 1, length(lager_config:global_get(handlers))),
  800. ok = lager:stop_trace(TestTrace2),
  801. EndHandlers = gen_event:which_handlers(Sink),
  802. EndGlobal = lager_config:global_get(handlers),
  803. {_, T3} = lager_config:get({Sink, loglevel}),
  804. ?assertEqual([], T3),
  805. ?assertEqual(StartHandlers, EndHandlers),
  806. ?assertEqual(StartGlobal, EndGlobal),
  807. ok
  808. end
  809. },
  810. {"log messages below the threshold are ignored",
  811. fun() ->
  812. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  813. ?TEST_SINK_NAME:debug("this message will be ignored"),
  814. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  815. ?assertEqual(0, count_ignored(?TEST_SINK_EVENT)),
  816. lager_config:set({?TEST_SINK_EVENT, loglevel}, {element(2, lager_util:config_to_mask(debug)), []}),
  817. ?TEST_SINK_NAME:debug("this message should be ignored"),
  818. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  819. ?assertEqual(1, count_ignored(?TEST_SINK_EVENT)),
  820. lager:set_loglevel(?TEST_SINK_EVENT, ?MODULE, undefined, debug),
  821. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({?TEST_SINK_EVENT, loglevel})),
  822. ?TEST_SINK_NAME:debug("this message should be logged"),
  823. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  824. ?assertEqual(1, count_ignored(?TEST_SINK_EVENT)),
  825. ?assertEqual(debug, lager:get_loglevel(?TEST_SINK_EVENT, ?MODULE)),
  826. ok
  827. end
  828. }
  829. ]
  830. }.
  831. setup_sink() ->
  832. error_logger:tty(false),
  833. application:load(lager),
  834. application:set_env(lager, handlers, []),
  835. application:set_env(lager, error_logger_redirect, false),
  836. application:set_env(lager, extra_sinks, [{?TEST_SINK_EVENT, [{handlers, [{?MODULE, info}]}]}]),
  837. lager:start(),
  838. gen_event:call(lager_event, ?MODULE, flush),
  839. gen_event:call(?TEST_SINK_EVENT, ?MODULE, flush).
  840. setup() ->
  841. error_logger:tty(false),
  842. application:load(lager),
  843. application:set_env(lager, handlers, [{?MODULE, info}]),
  844. application:set_env(lager, error_logger_redirect, false),
  845. application:unset_env(lager, traces),
  846. lager:start(),
  847. %% There is a race condition between the application start up, lager logging its own
  848. %% start up condition and several tests that count messages or parse the output of
  849. %% tests. When the lager start up message wins the race, it causes these tests
  850. %% which parse output or count message arrivals to fail.
  851. %%
  852. %% We introduce a sleep here to allow `flush' to arrive *after* the start up
  853. %% message has been received and processed.
  854. %%
  855. %% This race condition was first exposed during the work on
  856. %% 4b5260c4524688b545cc12da6baa2dfa4f2afec9 which introduced the lager
  857. %% manager killer PR.
  858. application:set_env(lager, suppress_supervisor_start_stop, true),
  859. application:set_env(lager, suppress_application_start_stop, true),
  860. timer:sleep(1000),
  861. gen_event:call(lager_event, ?MODULE, flush).
  862. cleanup(_) ->
  863. catch ets:delete(lager_config), %% kill the ets config table with fire
  864. application:stop(lager),
  865. application:stop(goldrush),
  866. error_logger:tty(true).
  867. crash(Type) ->
  868. spawn(fun() -> gen_server:call(crash, Type) end),
  869. timer:sleep(100),
  870. _ = gen_event:which_handlers(error_logger),
  871. ok.
  872. test_body({slice, Expected}, Actual) ->
  873. SlicedActual = string:slice(Actual, 0, length(Expected)),
  874. ?assertEqual(Expected, SlicedActual, {Actual, sliced_to, SlicedActual, is_not_a_member_of, Expected});
  875. test_body(Expected, Actual) ->
  876. ExLen = length(Expected),
  877. {Body, Rest} = case length(Actual) > ExLen of
  878. true ->
  879. {string:substr(Actual, 1, ExLen),
  880. string:substr(Actual, (ExLen + 1))};
  881. _ ->
  882. {Actual, []}
  883. end,
  884. ?assertEqual(Expected, Body),
  885. % OTP-17 (and maybe later releases) may tack on additional info
  886. % about the failure, so if Actual starts with Expected (already
  887. % confirmed by having gotten past assertEqual above) and ends
  888. % with " line NNN" we can ignore what's in-between. By extension,
  889. % since there may not be line information appended at all, any
  890. % text we DO find is reportable, but not a test failure.
  891. case Rest of
  892. [] ->
  893. ok;
  894. _ ->
  895. % isolate the extra data and report it if it's not just
  896. % a line number indicator
  897. case re:run(Rest, "^.*( line \\d+)$", [{capture, [1]}]) of
  898. nomatch ->
  899. ?debugFmt(
  900. "Trailing data \"~s\" following \"~s\"",
  901. [Rest, Expected]);
  902. {match, [{0, _}]} ->
  903. % the whole sting is " line NNN"
  904. ok;
  905. {match, [{Off, _}]} ->
  906. ?debugFmt(
  907. "Trailing data \"~s\" following \"~s\"",
  908. [string:substr(Rest, 1, Off), Expected])
  909. end
  910. end.
  911. error_logger_redirect_crash_setup() ->
  912. error_logger:tty(false),
  913. application:load(lager),
  914. application:set_env(lager, error_logger_redirect, true),
  915. application:set_env(lager, handlers, [{?MODULE, error}]),
  916. lager:start(),
  917. crash:start(),
  918. lager_event.
  919. error_logger_redirect_crash_setup_sink() ->
  920. error_logger:tty(false),
  921. application:load(lager),
  922. application:set_env(lager, error_logger_redirect, true),
  923. application:unset_env(lager, handlers),
  924. application:set_env(lager, extra_sinks, [
  925. {error_logger_lager_event, [
  926. {handlers, [{?MODULE, error}]}]}]),
  927. lager:start(),
  928. crash:start(),
  929. error_logger_lager_event.
  930. error_logger_redirect_crash_cleanup(_Sink) ->
  931. application:stop(lager),
  932. application:stop(goldrush),
  933. application:unset_env(lager, extra_sinks),
  934. case whereis(crash) of
  935. undefined -> ok;
  936. Pid -> exit(Pid, kill)
  937. end,
  938. error_logger:tty(true).
  939. crash_fsm_setup() ->
  940. error_logger:tty(false),
  941. application:load(lager),
  942. application:set_env(lager, error_logger_redirect, true),
  943. application:set_env(lager, handlers, [{?MODULE, error}]),
  944. lager:start(),
  945. crash_fsm:start(),
  946. crash_statem:start(),
  947. lager:log(error, self(), "flush flush"),
  948. timer:sleep(100),
  949. gen_event:call(lager_event, ?MODULE, flush),
  950. lager_event.
  951. crash_fsm_sink_setup() ->
  952. ErrorSink = error_logger_lager_event,
  953. error_logger:tty(false),
  954. application:load(lager),
  955. application:set_env(lager, error_logger_redirect, true),
  956. application:set_env(lager, handlers, []),
  957. application:set_env(lager, extra_sinks, [{ErrorSink, [{handlers, [{?MODULE, error}]}]}]),
  958. lager:start(),
  959. crash_fsm:start(),
  960. crash_statem:start(),
  961. lager:log(ErrorSink, error, self(), "flush flush", []),
  962. timer:sleep(100),
  963. flush(ErrorSink),
  964. ErrorSink.
  965. crash_fsm_cleanup(_Sink) ->
  966. application:stop(lager),
  967. application:stop(goldrush),
  968. application:unset_env(lager, extra_sinks),
  969. lists:foreach(fun(N) -> kill_crasher(N) end, [crash_fsm, crash_statem]),
  970. error_logger:tty(true).
  971. kill_crasher(RegName) ->
  972. case whereis(RegName) of
  973. undefined -> ok;
  974. Pid -> exit(Pid, kill)
  975. end.
  976. spawn_fsm_crash(Module, Function, Args) ->
  977. spawn(fun() -> erlang:apply(Module, Function, Args) end),
  978. timer:sleep(100),
  979. _ = gen_event:which_handlers(error_logger),
  980. ok.
  981. crash_fsm_test_() ->
  982. TestBody = fun(Name, FsmModule, FSMFunc, FSMArgs, Expected) ->
  983. fun(Sink) ->
  984. {Name,
  985. fun() ->
  986. case {FsmModule =:= crash_statem, lager_util:otp_version() < 19} of
  987. {true, true} -> ok;
  988. _ ->
  989. Pid = whereis(FsmModule),
  990. spawn_fsm_crash(FsmModule, FSMFunc, FSMArgs),
  991. {Level, _, Msg, Metadata} = pop(Sink),
  992. test_body(Expected, lists:flatten(Msg)),
  993. ?assertEqual(Pid, proplists:get_value(pid, Metadata)),
  994. ?assertEqual(lager_util:level_to_num(error), Level)
  995. end
  996. end
  997. }
  998. end
  999. end,
  1000. Tests = [
  1001. fun(Sink) ->
  1002. {"again, there is nothing up my sleeve",
  1003. fun() ->
  1004. ?assertEqual(undefined, pop(Sink)),
  1005. ?assertEqual(0, count(Sink))
  1006. end
  1007. }
  1008. end,
  1009. TestBody("gen_statem crash", crash_statem, crash, [], {slice, "gen_statem crash_statem in state state1 terminated with reason"}),
  1010. TestBody("gen_statem stop", crash_statem, stop, [explode], {slice, "gen_statem crash_statem in state state1 terminated with reason"}),
  1011. TestBody("gen_statem timeout", crash_statem, timeout, [], {slice, "gen_statem crash_statem in state state1 terminated with reason"})
  1012. ] ++ test_body_gen_fsm_crash(TestBody),
  1013. {"FSM crash output tests", [
  1014. {"Default sink",
  1015. {foreach,
  1016. fun crash_fsm_setup/0,
  1017. fun crash_fsm_cleanup/1,
  1018. Tests}},
  1019. {"Error logger sink",
  1020. {foreach,
  1021. fun crash_fsm_sink_setup/0,
  1022. fun crash_fsm_cleanup/1,
  1023. Tests}}
  1024. ]}.
  1025. - if (?OTP_RELEASE < 23).
  1026. test_body_gen_fsm_crash(TestBody) ->
  1027. [TestBody("gen_fsm crash", crash_fsm, crash, [], "gen_fsm crash_fsm in state state1 terminated with reason: call to undefined function crash_fsm:state1/3 from gen_fsm:handle_msg/")].
  1028. -else.
  1029. test_body_gen_fsm_crash(_TestBody) ->
  1030. [].
  1031. -endif.
  1032. error_logger_redirect_crash_test_() ->
  1033. TestBody = fun(Name, CrashReason, Expected) ->
  1034. fun(Sink) ->
  1035. {Name,
  1036. fun() ->
  1037. Pid = whereis(crash),
  1038. crash(CrashReason),
  1039. {Level, _, Msg, Metadata} = pop(Sink),
  1040. test_body(Expected, lists:flatten(Msg)),
  1041. ?assertEqual(Pid, proplists:get_value(pid, Metadata)),
  1042. ?assertEqual(lager_util:level_to_num(error), Level)
  1043. end
  1044. }
  1045. end
  1046. end,
  1047. Tests = [
  1048. fun(Sink) ->
  1049. {"again, there is nothing up my sleeve",
  1050. fun() ->
  1051. ?assertEqual(undefined, pop(Sink)),
  1052. ?assertEqual(0, count(Sink))
  1053. end
  1054. }
  1055. end,
  1056. TestBody("bad return value", bad_return, "gen_server crash terminated with reason: bad return value: bleh"),
  1057. TestBody("bad return value with string", bad_return_string, "gen_server crash terminated with reason: bad return value: {tuple,{tuple,\"string\"}}"),
  1058. TestBody("bad return uncaught throw", throw, "gen_server crash terminated with reason: bad return value: a_ball"),
  1059. TestBody("case clause", case_clause, "gen_server crash terminated with reason: no case clause matching {} in crash:handle_call/3"),
  1060. TestBody("case clause string", case_clause_string, "gen_server crash terminated with reason: no case clause matching \"crash\" in crash:handle_call/3"),
  1061. TestBody("function clause", function_clause, "gen_server crash terminated with reason: no function clause matching crash:function({})"),
  1062. TestBody("if clause", if_clause, "gen_server crash terminated with reason: no true branch found while evaluating if expression in crash:handle_call/3"),
  1063. TestBody("try clause", try_clause, "gen_server crash terminated with reason: no try clause matching [] in crash:handle_call/3"),
  1064. TestBody("undefined function", undef, "gen_server crash terminated with reason: call to undefined function crash:booger/0 from crash:handle_call/3"),
  1065. TestBody("bad math", badarith, "gen_server crash terminated with reason: bad arithmetic expression in crash:handle_call/3"),
  1066. TestBody("bad match", badmatch, "gen_server crash terminated with reason: no match of right hand value {} in crash:handle_call/3"),
  1067. TestBody("bad arity", badarity, "gen_server crash terminated with reason: fun called with wrong arity of 1 instead of 3 in crash:handle_call/3"),
  1068. TestBody("bad arg1", badarg1, "gen_server crash terminated with reason: bad argument in crash:handle_call/3"),
  1069. TestBody("bad arg2", badarg2, "gen_server crash terminated with reason: bad argument in call to erlang:iolist_to_binary([\"foo\",bar]) in crash:handle_call/3"),
  1070. TestBody("bad record", badrecord, "gen_server crash terminated with reason: bad record state in crash:handle_call/3"),
  1071. TestBody("noproc", noproc, "gen_server crash terminated with reason: no such process or port in call to gen_event:call(foo, bar, baz)"),
  1072. TestBody("noproc_proc_lib", noproc_proc_lib, "gen_server crash terminated with reason: no such process or port in call to proc_lib:stop/3"),
  1073. TestBody("badfun", badfun, "gen_server crash terminated with reason: bad function booger in crash:handle_call/3")
  1074. ],
  1075. {"Error logger redirect crash", [
  1076. {"Redirect to default sink",
  1077. {foreach,
  1078. fun error_logger_redirect_crash_setup/0,
  1079. fun error_logger_redirect_crash_cleanup/1,
  1080. Tests}},
  1081. {"Redirect to error_logger_lager_event sink",
  1082. {foreach,
  1083. fun error_logger_redirect_crash_setup_sink/0,
  1084. fun error_logger_redirect_crash_cleanup/1,
  1085. Tests}}
  1086. ]}.
  1087. error_logger_redirect_setup() ->
  1088. error_logger:tty(false),
  1089. application:load(lager),
  1090. application:set_env(lager, error_logger_redirect, true),
  1091. application:set_env(lager, handlers, [{?MODULE, info}]),
  1092. application:set_env(lager, suppress_supervisor_start_stop, false),
  1093. application:set_env(lager, suppress_application_start_stop, false),
  1094. lager:start(),
  1095. lager:log(error, self(), "flush flush"),
  1096. timer:sleep(1000),
  1097. gen_event:call(lager_event, ?MODULE, flush),
  1098. lager_event.
  1099. error_logger_redirect_setup_sink() ->
  1100. error_logger:tty(false),
  1101. application:load(lager),
  1102. application:set_env(lager, error_logger_redirect, true),
  1103. application:unset_env(lager, handlers),
  1104. application:set_env(lager, extra_sinks, [
  1105. {error_logger_lager_event, [
  1106. {handlers, [{?MODULE, info}]}]}]),
  1107. application:set_env(lager, suppress_supervisor_start_stop, false),
  1108. application:set_env(lager, suppress_application_start_stop, false),
  1109. lager:start(),
  1110. lager:log(error_logger_lager_event, error, self(), "flush flush", []),
  1111. timer:sleep(1000),
  1112. gen_event:call(error_logger_lager_event, ?MODULE, flush),
  1113. error_logger_lager_event.
  1114. error_logger_redirect_cleanup(_) ->
  1115. application:stop(lager),
  1116. application:stop(goldrush),
  1117. application:unset_env(lager, extra_sinks),
  1118. error_logger:tty(true).
  1119. error_logger_redirect_test_() ->
  1120. Tests = [
  1121. {"error reports are printed",
  1122. fun(Sink) ->
  1123. sync_error_logger:error_report([{this, is}, a, {silly, format}]),
  1124. _ = gen_event:which_handlers(error_logger),
  1125. {Level, _, Msg, Metadata} = pop(Sink),
  1126. ?assertEqual(lager_util:level_to_num(error), Level),
  1127. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1128. Expected = "this: is, a, silly: format",
  1129. ?assertEqual(Expected, lists:flatten(Msg))
  1130. end
  1131. },
  1132. {"string error reports are printed",
  1133. fun(Sink) ->
  1134. sync_error_logger:error_report("this is less silly"),
  1135. _ = gen_event:which_handlers(error_logger),
  1136. {Level, _, Msg, Metadata} = pop(Sink),
  1137. ?assertEqual(lager_util:level_to_num(error), Level),
  1138. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1139. Expected = "this is less silly",
  1140. ?assertEqual(Expected, lists:flatten(Msg))
  1141. end
  1142. },
  1143. {"error messages are printed",
  1144. fun(Sink) ->
  1145. sync_error_logger:error_msg("doom, doom has come upon you all"),
  1146. _ = gen_event:which_handlers(error_logger),
  1147. {Level, _, Msg, Metadata} = pop(Sink),
  1148. ?assertEqual(lager_util:level_to_num(error), Level),
  1149. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1150. Expected = "doom, doom has come upon you all",
  1151. ?assertEqual(Expected, lists:flatten(Msg))
  1152. end
  1153. },
  1154. {"error messages with unicode characters in Args are printed",
  1155. fun(Sink) ->
  1156. sync_error_logger:error_msg("~ts", ["Привет!"]),
  1157. _ = gen_event:which_handlers(error_logger),
  1158. {Level, _, Msg, Metadata} = pop(Sink),
  1159. ?assertEqual(lager_util:level_to_num(error), Level),
  1160. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1161. ?assertEqual("Привет!", lists:flatten(Msg))
  1162. end
  1163. },
  1164. {"error messages are truncated at 4096 characters",
  1165. fun(Sink) ->
  1166. sync_error_logger:error_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  1167. _ = gen_event:which_handlers(error_logger),
  1168. {_, _, Msg, _Metadata} = pop(Sink),
  1169. ?assert(length(lists:flatten(Msg)) < 5100)
  1170. end
  1171. },
  1172. {"info reports are printed",
  1173. fun(Sink) ->
  1174. sync_error_logger:info_report([{this, is}, a, {silly, format}]),
  1175. _ = gen_event:which_handlers(error_logger),
  1176. {Level, _, Msg, Metadata} = pop(Sink),
  1177. ?assertEqual(lager_util:level_to_num(info), Level),
  1178. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1179. Expected = "this: is, a, silly: format",
  1180. ?assertEqual(Expected, lists:flatten(Msg))
  1181. end
  1182. },
  1183. {"info reports are truncated at 4096 characters",
  1184. fun(Sink) ->
  1185. sync_error_logger:info_report([[{this, is}, a, {silly, format}] || _ <- lists:seq(0, 600)]),
  1186. _ = gen_event:which_handlers(error_logger),
  1187. {Level, _, Msg, Metadata} = pop(Sink),
  1188. ?assertEqual(lager_util:level_to_num(info), Level),
  1189. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1190. ?assert(length(lists:flatten(Msg)) < 5000)
  1191. end
  1192. },
  1193. {"single term info reports are printed",
  1194. fun(Sink) ->
  1195. sync_error_logger:info_report({foolish, bees}),
  1196. _ = gen_event:which_handlers(error_logger),
  1197. {Level, _, Msg, Metadata} = pop(Sink),
  1198. ?assertEqual(lager_util:level_to_num(info), Level),
  1199. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1200. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  1201. end
  1202. },
  1203. {"single term error reports are printed",
  1204. fun(Sink) ->
  1205. sync_error_logger:error_report({foolish, bees}),
  1206. _ = gen_event:which_handlers(error_logger),
  1207. {Level, _, Msg, Metadata} = pop(Sink),
  1208. ?assertEqual(lager_util:level_to_num(error), Level),
  1209. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1210. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  1211. end
  1212. },
  1213. {"string info reports are printed",
  1214. fun(Sink) ->
  1215. sync_error_logger:info_report("this is less silly"),
  1216. _ = gen_event:which_handlers(error_logger),
  1217. {Level, _, Msg, Metadata} = pop(Sink),
  1218. ?assertEqual(lager_util:level_to_num(info), Level),
  1219. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1220. ?assertEqual("this is less silly", lists:flatten(Msg))
  1221. end
  1222. },
  1223. {"string info reports are truncated at 4096 characters",
  1224. fun(Sink) ->
  1225. sync_error_logger:info_report(string:copies("this is less silly", 1000)),
  1226. _ = gen_event:which_handlers(error_logger),
  1227. {Level, _, Msg, Metadata} = pop(Sink),
  1228. ?assertEqual(lager_util:level_to_num(info), Level),
  1229. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1230. ?assert(length(lists:flatten(Msg)) < 5100)
  1231. end
  1232. },
  1233. {"strings in a mixed report are printed as strings",
  1234. fun(Sink) ->
  1235. sync_error_logger:info_report(["this is less silly", {than, "this"}]),
  1236. _ = gen_event:which_handlers(error_logger),
  1237. {Level, _, Msg, Metadata} = pop(Sink),
  1238. ?assertEqual(lager_util:level_to_num(info), Level),
  1239. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1240. ?assertEqual("\"this is less silly\", than: \"this\"", lists:flatten(Msg))
  1241. end
  1242. },
  1243. {"info messages are printed",
  1244. fun(Sink) ->
  1245. sync_error_logger:info_msg("doom, doom has come upon you all"),
  1246. _ = gen_event:which_handlers(error_logger),
  1247. {Level, _, Msg, Metadata} = pop(Sink),
  1248. ?assertEqual(lager_util:level_to_num(info), Level),
  1249. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1250. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  1251. end
  1252. },
  1253. {"info messages are truncated at 4096 characters",
  1254. fun(Sink) ->
  1255. sync_error_logger:info_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  1256. _ = gen_event:which_handlers(error_logger),
  1257. {Level, _, Msg, Metadata} = pop(Sink),
  1258. ?assertEqual(lager_util:level_to_num(info), Level),
  1259. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1260. ?assert(length(lists:flatten(Msg)) < 5100)
  1261. end
  1262. },
  1263. {"info messages with unicode characters in Args are printed",
  1264. fun(Sink) ->
  1265. sync_error_logger:info_msg("~ts", ["Привет!"]),
  1266. _ = gen_event:which_handlers(error_logger),
  1267. {Level, _, Msg, Metadata} = pop(Sink),
  1268. ?assertEqual(lager_util:level_to_num(info), Level),
  1269. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1270. ?assertEqual("Привет!", lists:flatten(Msg))
  1271. end
  1272. },
  1273. {"warning messages with unicode characters in Args are printed",
  1274. %% The next 4 tests need to store the current value of
  1275. %% `error_logger:warning_map/0' into a process dictionary
  1276. %% key `warning_map' so that the error message level used
  1277. %% to process the log messages will match what lager
  1278. %% expects.
  1279. %%
  1280. %% The atom returned by `error_logger:warning_map/0'
  1281. %% changed between OTP 17 and 18 (and later releases)
  1282. %%
  1283. %% `warning_map' is consumed in the `test/sync_error_logger.erl'
  1284. %% module. The default message level used in sync_error_logger
  1285. %% was fine for OTP releases through 17 and then broke
  1286. %% when 18 was released. By storing the expected value
  1287. %% in the process dictionary, sync_error_logger will
  1288. %% use the correct message level to process the
  1289. %% messages and these tests will no longer
  1290. %% break.
  1291. fun(Sink) ->
  1292. Lvl = error_logger:warning_map(),
  1293. put(warning_map, Lvl),
  1294. sync_error_logger:warning_msg("~ts", ["Привет!"]),
  1295. _ = gen_event:which_handlers(error_logger),
  1296. {Level, _, Msg, Metadata} = pop(Sink),
  1297. ?assertEqual(lager_util:level_to_num(Lvl), Level),
  1298. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1299. ?assertEqual("Привет!", lists:flatten(Msg))
  1300. end
  1301. },
  1302. {"warning messages are printed at the correct level",
  1303. fun(Sink) ->
  1304. Lvl = error_logger:warning_map(),
  1305. put(warning_map, Lvl),
  1306. sync_error_logger:warning_msg("doom, doom has come upon you all"),
  1307. _ = gen_event:which_handlers(error_logger),
  1308. {Level, _, Msg, Metadata} = pop(Sink),
  1309. ?assertEqual(lager_util:level_to_num(Lvl), Level),
  1310. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1311. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  1312. end
  1313. },
  1314. {"warning reports are printed at the correct level",
  1315. fun(Sink) ->
  1316. Lvl = error_logger:warning_map(),
  1317. put(warning_map, Lvl),
  1318. sync_error_logger:warning_report([{i, like}, pie]),
  1319. _ = gen_event:which_handlers(error_logger),
  1320. {Level, _, Msg, Metadata} = pop(Sink),
  1321. ?assertEqual(lager_util:level_to_num(Lvl), Level),
  1322. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1323. ?assertEqual("i: like, pie", lists:flatten(Msg))
  1324. end
  1325. },
  1326. {"single term warning reports are printed at the correct level",
  1327. fun(Sink) ->
  1328. Lvl = error_logger:warning_map(),
  1329. put(warning_map, Lvl),
  1330. sync_error_logger:warning_report({foolish, bees}),
  1331. _ = gen_event:which_handlers(error_logger),
  1332. {Level, _, Msg, Metadata} = pop(Sink),
  1333. ?assertEqual(lager_util:level_to_num(Lvl), Level),
  1334. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1335. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  1336. end
  1337. },
  1338. {"application stop reports",
  1339. fun(Sink) ->
  1340. sync_error_logger:info_report([{application, foo}, {exited, quittin_time}, {type, lazy}]),
  1341. _ = gen_event:which_handlers(error_logger),
  1342. {Level, _, Msg, Metadata} = pop(Sink),
  1343. ?assertEqual(lager_util:level_to_num(info), Level),
  1344. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1345. ?assertEqual("Application foo exited with reason: quittin_time", lists:flatten(Msg))
  1346. end
  1347. },
  1348. {"supervisor reports",
  1349. fun(Sink) ->
  1350. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  1351. _ = gen_event:which_handlers(error_logger),
  1352. {Level, _, Msg, Metadata} = pop(Sink),
  1353. ?assertEqual(lager_util:level_to_num(error), Level),
  1354. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1355. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason fired in context france", lists:flatten(Msg))
  1356. end
  1357. },
  1358. {"supervisor reports with real error",
  1359. fun(Sink) ->
  1360. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, {function_clause, [{crash, handle_info, [foo]}]}}, {supervisor, {local, steve}}]),
  1361. _ = gen_event:which_handlers(error_logger),
  1362. {Level, _, Msg, Metadata} = pop(Sink),
  1363. ?assertEqual(lager_util:level_to_num(error), Level),
  1364. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1365. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason no function clause matching crash:handle_info(foo) in context france", lists:flatten(Msg))
  1366. end
  1367. },
  1368. {"supervisor reports with real error and pid",
  1369. fun(Sink) ->
  1370. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, {function_clause, [{crash, handle_info, [foo]}]}}, {supervisor, somepid}]),
  1371. _ = gen_event:which_handlers(error_logger),
  1372. {Level, _, Msg, Metadata} = pop(Sink),
  1373. ?assertEqual(lager_util:level_to_num(error), Level),
  1374. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1375. ?assertEqual("Supervisor somepid had child mini_steve started with a:b(c) at bleh exit with reason no function clause matching crash:handle_info(foo) in context france", lists:flatten(Msg))
  1376. end
  1377. },
  1378. {"supervisor_bridge reports",
  1379. fun(Sink) ->
  1380. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{mod, mini_steve}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  1381. _ = gen_event:which_handlers(error_logger),
  1382. {Level, _, Msg, Metadata} = pop(Sink),
  1383. ?assertEqual(lager_util:level_to_num(error), Level),
  1384. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1385. ?assertEqual("Supervisor steve had child at module mini_steve at bleh exit with reason fired in context france", lists:flatten(Msg))
  1386. end
  1387. },
  1388. {"application progress report",
  1389. fun(Sink) ->
  1390. sync_error_logger:info_report(progress, [{application, foo}, {started_at, node()}]),
  1391. _ = gen_event:which_handlers(error_logger),
  1392. {Level, _, Msg, Metadata} = pop(Sink),
  1393. ?assertEqual(lager_util:level_to_num(info), Level),
  1394. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1395. Expected = lists:flatten(io_lib:format("Application foo started on node ~w", [node()])),
  1396. ?assertEqual(Expected, lists:flatten(Msg))
  1397. end
  1398. },
  1399. {"supervisor progress report",
  1400. fun(Sink) ->
  1401. lager:set_loglevel(Sink, ?MODULE, undefined, debug),
  1402. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({Sink, loglevel})),
  1403. sync_error_logger:info_report(progress, [{supervisor, {local, foo}}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1404. _ = gen_event:which_handlers(error_logger),
  1405. {Level, _, Msg, Metadata} = pop(Sink),
  1406. ?assertEqual(lager_util:level_to_num(debug), Level),
  1407. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1408. ?assertEqual("Supervisor foo started foo:bar/1 at pid baz", lists:flatten(Msg))
  1409. end
  1410. },
  1411. {"supervisor progress report with pid",
  1412. fun(Sink) ->
  1413. lager:set_loglevel(Sink, ?MODULE, undefined, debug),
  1414. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({Sink, loglevel})),
  1415. sync_error_logger:info_report(progress, [{supervisor, somepid}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1416. _ = gen_event:which_handlers(error_logger),
  1417. {Level, _, Msg, Metadata} = pop(Sink),
  1418. ?assertEqual(lager_util:level_to_num(debug), Level),
  1419. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1420. ?assertEqual("Supervisor somepid started foo:bar/1 at pid baz", lists:flatten(Msg))
  1421. end
  1422. },
  1423. {"crash report for emfile",
  1424. fun(Sink) ->
  1425. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, emfile, [{stack, trace, 1}]}}], []]),
  1426. _ = gen_event:which_handlers(error_logger),
  1427. {Level, _, Msg, Metadata} = pop(Sink),
  1428. ?assertEqual(lager_util:level_to_num(error), Level),
  1429. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1430. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: maximum number of file descriptors exhausted, check ulimit -n", [self()])),
  1431. ?assertEqual(Expected, lists:flatten(Msg))
  1432. end
  1433. },
  1434. {"crash report for system process limit",
  1435. fun(Sink) ->
  1436. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn, 1}]}}], []]),
  1437. _ = gen_event:which_handlers(error_logger),
  1438. {Level, _, Msg, Metadata} = pop(Sink),
  1439. ?assertEqual(lager_util:level_to_num(error), Level),
  1440. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1441. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1442. ?assertEqual(Expected, lists:flatten(Msg))
  1443. end
  1444. },
  1445. {"crash report for system process limit2",
  1446. fun(Sink) ->
  1447. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn_opt, 1}]}}], []]),
  1448. _ = gen_event:which_handlers(error_logger),
  1449. {Level, _, Msg, Metadata} = pop(Sink),
  1450. ?assertEqual(lager_util:level_to_num(error), Level),
  1451. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1452. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1453. ?assertEqual(Expected, lists:flatten(Msg))
  1454. end
  1455. },
  1456. {"crash report for system port limit",
  1457. fun(Sink) ->
  1458. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, open_port, 1}]}}], []]),
  1459. _ = gen_event:which_handlers(error_logger),
  1460. {Level, _, Msg, Metadata} = pop(Sink),
  1461. ?assertEqual(lager_util:level_to_num(error), Level),
  1462. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1463. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ports exceeded", [self()])),
  1464. ?assertEqual(Expected, lists:flatten(Msg))
  1465. end
  1466. },
  1467. {"crash report for system port limit",
  1468. fun(Sink) ->
  1469. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, list_to_atom, 1}]}}], []]),
  1470. _ = gen_event:which_handlers(error_logger),
  1471. {Level, _, Msg, Metadata} = pop(Sink),
  1472. ?assertEqual(lager_util:level_to_num(error), Level),
  1473. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1474. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: tried to create an atom larger than 255, or maximum atom count exceeded", [self()])),
  1475. ?assertEqual(Expected, lists:flatten(Msg))
  1476. end
  1477. },
  1478. {"crash report for system ets table limit",
  1479. fun(Sink) ->
  1480. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, test}, {error_info, {error, system_limit, [{ets, new, [segment_offsets, [ordered_set, public]]}, {mi_segment, open_write, 1}, {mi_buffer_converter, handle_cast, 2}, {gen_server, handle_msg, 5}, {proc_lib, init_p_do_apply, 3}]}}], []]),
  1481. _ = gen_event:which_handlers(error_logger),
  1482. {Level, _, Msg, Metadata} = pop(Sink),
  1483. ?assertEqual(lager_util:level_to_num(error), Level),
  1484. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1485. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ETS tables exceeded", [test])),
  1486. ?assertEqual(Expected, lists:flatten(Msg))
  1487. end
  1488. },
  1489. {"crash report for unknown system limit should be truncated at 500 characters",
  1490. fun(Sink) ->
  1491. sync_error_logger:error_report(crash_report, [[{pid, self()}, {error_info, {error, system_limit, [{wtf, boom, [string:copies("aaaa", 4096)]}]}}], []]),
  1492. _ = gen_event:which_handlers(error_logger),
  1493. {_, _, Msg, _Metadata} = pop(Sink),
  1494. ?assert(length(lists:flatten(Msg)) > 550),
  1495. ?assert(length(lists:flatten(Msg)) < 600)
  1496. end
  1497. },
  1498. {"crash reports for 'special processes' should be handled right - function_clause",
  1499. fun(Sink) ->
  1500. {ok, Pid} = special_process:start(),
  1501. unlink(Pid),
  1502. Pid ! function_clause,
  1503. timer:sleep(500),
  1504. _ = gen_event:which_handlers(error_logger),
  1505. {_, _, Msg, _Metadata} = pop(Sink),
  1506. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no function clause matching special_process:foo(bar)",
  1507. [Pid])),
  1508. test_body(Expected, lists:flatten(Msg))
  1509. end
  1510. },
  1511. {"crash reports for 'special processes' should be handled right - case_clause",
  1512. fun(Sink) ->
  1513. {ok, Pid} = special_process:start(),
  1514. unlink(Pid),
  1515. Pid ! {case_clause, wtf},
  1516. timer:sleep(500),
  1517. _ = gen_event:which_handlers(error_logger),
  1518. {_, _, Msg, _Metadata} = pop(Sink),
  1519. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no case clause matching wtf in special_process:loop/0",
  1520. [Pid])),
  1521. test_body(Expected, lists:flatten(Msg))
  1522. end
  1523. },
  1524. {"crash reports for 'special processes' should be handled right - exit",
  1525. fun(Sink) ->
  1526. {ok, Pid} = special_process:start(),
  1527. unlink(Pid),
  1528. Pid ! exit,
  1529. timer:sleep(500),
  1530. _ = gen_event:which_handlers(error_logger),
  1531. {_, _, Msg, _Metadata} = pop(Sink),
  1532. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours exited with reason: byebye in special_process:loop/0",
  1533. [Pid])),
  1534. test_body(Expected, lists:flatten(Msg))
  1535. end
  1536. },
  1537. {"crash reports for 'special processes' should be handled right - error",
  1538. fun(Sink) ->
  1539. {ok, Pid} = special_process:start(),
  1540. unlink(Pid),
  1541. Pid ! error,
  1542. timer:sleep(500),
  1543. _ = gen_event:which_handlers(error_logger),
  1544. {_, _, Msg, _Metadata} = pop(Sink),
  1545. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: mybad in special_process:loop/0",
  1546. [Pid])),
  1547. test_body(Expected, lists:flatten(Msg))
  1548. end
  1549. },
  1550. {"webmachine error reports",
  1551. fun(Sink) ->
  1552. Path = "/cgi-bin/phpmyadmin",
  1553. Reason = {error, {error, {badmatch, {error, timeout}},
  1554. [{myapp, dostuff, 2, [{file, "src/myapp.erl"}, {line, 123}]},
  1555. {webmachine_resource, resource_call, 3, [{file, "src/webmachine_resource.erl"}, {line, 169}]}]}},
  1556. sync_error_logger:error_msg("webmachine error: path=~p~n~p~n", [Path, Reason]),
  1557. _ = gen_event:which_handlers(error_logger),
  1558. {Level, _, Msg, Metadata} = pop(Sink),
  1559. ?assertEqual(lager_util:level_to_num(error), Level),
  1560. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1561. ?assertEqual("Webmachine error at path \"/cgi-bin/phpmyadmin\" : no match of right hand value {error,timeout} in myapp:dostuff/2 line 123", lists:flatten(Msg))
  1562. end
  1563. },
  1564. {"Cowboy error reports, 8 arg version",
  1565. fun(Sink) ->
  1566. Stack = [{my_handler, init, 3, [{file, "src/my_handler.erl"}, {line, 123}]},
  1567. {cowboy_handler, handler_init, 4, [{file, "src/cowboy_handler.erl"}, {line, 169}]}],
  1568. sync_error_logger:error_msg(
  1569. "** Cowboy handler ~p terminating in ~p/~p~n"
  1570. " for the reason ~p:~p~n"
  1571. "** Options were ~p~n"
  1572. "** Request was ~p~n"
  1573. "** Stacktrace: ~p~n~n",
  1574. [my_handler, init, 3, error, {badmatch, {error, timeout}}, [],
  1575. "Request", Stack]),
  1576. _ = gen_event:which_handlers(error_logger),
  1577. {Level, _, Msg, Metadata} = pop(Sink),
  1578. ?assertEqual(lager_util:level_to_num(error), Level),
  1579. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1580. ?assertEqual("Cowboy handler my_handler terminated in my_handler:init/3 with reason: no match of right hand value {error,timeout} in my_handler:init/3 line 123", lists:flatten(Msg))
  1581. end
  1582. },
  1583. {"Cowboy error reports, 10 arg version",
  1584. fun(Sink) ->
  1585. Stack = [{my_handler, somecallback, 3, [{file, "src/my_handler.erl"}, {line, 123}]},
  1586. {cowboy_handler, handler_init, 4, [{file, "src/cowboy_handler.erl"}, {line, 169}]}],
  1587. sync_error_logger:error_msg(
  1588. "** Cowboy handler ~p terminating in ~p/~p~n"
  1589. " for the reason ~p:~p~n** Message was ~p~n"
  1590. "** Options were ~p~n** Handler state was ~p~n"
  1591. "** Request was ~p~n** Stacktrace: ~p~n~n",
  1592. [my_handler, somecallback, 3, error, {badmatch, {error, timeout}}, hello, [],
  1593. {}, "Request", Stack]),
  1594. _ = gen_event:which_handlers(error_logger),
  1595. {Level, _, Msg, Metadata} = pop(Sink),
  1596. ?assertEqual(lager_util:level_to_num(error), Level),
  1597. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1598. ?assertEqual("Cowboy handler my_handler terminated in my_handler:somecallback/3 with reason: no match of right hand value {error,timeout} in my_handler:somecallback/3 line 123", lists:flatten(Msg))
  1599. end
  1600. },
  1601. {"Cowboy error reports, 5 arg version",
  1602. fun(Sink) ->
  1603. sync_error_logger:error_msg(
  1604. "** Cowboy handler ~p terminating; "
  1605. "function ~p/~p was not exported~n"
  1606. "** Request was ~p~n** State was ~p~n~n",
  1607. [my_handler, to_json, 2, "Request", {}]),
  1608. _ = gen_event:which_handlers(error_logger),
  1609. {Level, _, Msg, Metadata} = pop(Sink),
  1610. ?assertEqual(lager_util:level_to_num(error), Level),
  1611. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1612. ?assertEqual("Cowboy handler my_handler terminated with reason: call to undefined function my_handler:to_json/2", lists:flatten(Msg))
  1613. end
  1614. },
  1615. {"Cowboy error reports, 6 arg version",
  1616. fun(Sink) ->
  1617. Stack = [{app_http, init, 2, [{file, "app_http.erl"}, {line, 9}]},
  1618. {cowboy_handler, execute, 2, [{file, "cowboy_handler.erl"}, {line, 41}]}],
  1619. ConnectionPid = list_to_pid("<0.82.0>"),
  1620. sync_error_logger:error_msg(
  1621. "Ranch listener ~p, connection process ~p, stream ~p "
  1622. "had its request process ~p exit with reason "
  1623. "~999999p and stacktrace ~999999p~n",
  1624. [my_listner, ConnectionPid, 1, self(), {badmatch, 2}, Stack]),
  1625. _ = gen_event:which_handlers(error_logger),
  1626. {Level, _, Msg, Metadata} = pop(Sink),
  1627. ?assertEqual(lager_util:level_to_num(error), Level),
  1628. ?assertEqual(self(), proplists:get_value(pid, Metadata)),
  1629. ?assertEqual("Cowboy stream 1 with ranch listener my_listner and "
  1630. "connection process <0.82.0> had its request process exit "
  1631. "with reason: no match of right hand value 2 "
  1632. "in app_http:init/2 line 9", lists:flatten(Msg))
  1633. end
  1634. },
  1635. {"messages should not be generated if they don't satisfy the threshold",
  1636. fun(Sink) ->
  1637. lager:set_loglevel(Sink, ?MODULE, undefined, error),
  1638. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({Sink, loglevel})),
  1639. sync_error_logger:info_report([hello, world]),
  1640. _ = gen_event:which_handlers(error_logger),
  1641. ?assertEqual(0, count(Sink)),
  1642. ?assertEqual(0, count_ignored(Sink)),
  1643. lager:set_loglevel(Sink, ?MODULE, undefined, info),
  1644. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({Sink, loglevel})),
  1645. sync_error_logger:info_report([hello, world]),
  1646. _ = gen_event:which_handlers(error_logger),
  1647. ?assertEqual(1, count(Sink)),
  1648. ?assertEqual(0, count_ignored(Sink)),
  1649. lager:set_loglevel(Sink, ?MODULE, undefined, error),
  1650. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({Sink, loglevel})),
  1651. lager_config:set({Sink, loglevel}, {element(2, lager_util:config_to_mask(debug)), []}),
  1652. sync_error_logger:info_report([hello, world]),
  1653. _ = gen_event:which_handlers(error_logger),
  1654. ?assertEqual(1, count(Sink)),
  1655. ?assertEqual(1, count_ignored(Sink))
  1656. end
  1657. }
  1658. ],
  1659. SinkTests = lists:map(
  1660. fun({Name, F}) ->
  1661. fun(Sink) -> {Name, fun() -> F(Sink) end} end
  1662. end,
  1663. Tests),
  1664. {"Error logger redirect", [
  1665. {"Redirect to default sink",
  1666. {foreach,
  1667. fun error_logger_redirect_setup/0,
  1668. fun error_logger_redirect_cleanup/1,
  1669. SinkTests}},
  1670. {"Redirect to error_logger_lager_event sink",
  1671. {foreach,
  1672. fun error_logger_redirect_setup_sink/0,
  1673. fun error_logger_redirect_cleanup/1,
  1674. SinkTests}}
  1675. ]}.
  1676. safe_format_test() ->
  1677. ?assertEqual("foo bar", lists:flatten(lager:safe_format("~p ~p", [foo, bar], 1024))),
  1678. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:safe_format("~p ~p ~p", [foo, bar], 1024))),
  1679. ok.
  1680. unsafe_format_test() ->
  1681. ?assertEqual("foo bar", lists:flatten(lager:unsafe_format("~p ~p", [foo, bar]))),
  1682. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:unsafe_format("~p ~p ~p", [foo, bar]))),
  1683. ok.
  1684. async_threshold_test_() ->
  1685. Cleanup = fun(Reset) ->
  1686. _ = error_logger:tty(false),
  1687. _ = application:stop(lager),
  1688. _ = application:stop(goldrush),
  1689. _ = application:unset_env(lager, async_threshold),
  1690. if
  1691. Reset ->
  1692. true = ets:delete(async_threshold_test),
  1693. error_logger:tty(true);
  1694. true ->
  1695. _ = (catch ets:delete(async_threshold_test)),
  1696. ok
  1697. end
  1698. end,
  1699. Setup = fun() ->
  1700. % Evidence suggests that previous tests somewhere are leaving some of this stuff
  1701. % loaded, and cleaning it out forcefully to allows the test to succeed.
  1702. _ = Cleanup(false),
  1703. _ = ets:new(async_threshold_test, [set, named_table, public]),
  1704. ?assertEqual(true, ets:insert_new(async_threshold_test, {sync_toggled, 0})),
  1705. ?assertEqual(true, ets:insert_new(async_threshold_test, {async_toggled, 0})),
  1706. _ = application:load(lager),
  1707. ok = application:set_env(lager, error_logger_redirect, false),
  1708. ok = application:set_env(lager, async_threshold, 2),
  1709. ok = application:set_env(lager, async_threshold_window, 1),
  1710. ok = application:set_env(lager, handlers, [{?MODULE, info}]),
  1711. ok = lager:start(),
  1712. true
  1713. end,
  1714. {foreach, Setup, Cleanup, [
  1715. {"async threshold works",
  1716. {timeout, 30, fun() ->
  1717. Sleep = get_long_sleep_value(),
  1718. %% we start out async
  1719. ?assertEqual(true, lager_config:get(async)),
  1720. ?assertEqual([{sync_toggled, 0}],
  1721. ets:lookup(async_threshold_test, sync_toggled)),
  1722. %% put a ton of things in the queue
  1723. WorkCnt = erlang:max(10, (erlang:system_info(schedulers) * 2)),
  1724. OtpVsn = lager_util:otp_version(),
  1725. % newer OTPs *may* handle the messages faster, so we'll send more
  1726. MsgCnt = ((OtpVsn * OtpVsn) div 2),
  1727. Workers = spawn_stuffers(WorkCnt, [MsgCnt, info, "hello world"], []),
  1728. %% serialize on mailbox
  1729. _ = gen_event:which_handlers(lager_event),
  1730. timer:sleep(Sleep),
  1731. %% By now the flood of messages should have forced the backend throttle
  1732. %% to turn off async mode, but it's possible all outstanding requests
  1733. %% have been processed, so checking the current status (sync or async)
  1734. %% is an exercise in race control.
  1735. %% Instead, we'll see whether the backend throttle has toggled into sync
  1736. %% mode at any point in the past.
  1737. ?assertMatch([{sync_toggled, N}] when N > 0,
  1738. ets:lookup(async_threshold_test, sync_toggled)),
  1739. %% Wait for all the workers to return, meaning that all the messages have
  1740. %% been logged (since we're definitely in sync mode at the end of the run).
  1741. collect_workers(Workers),
  1742. %% serialize on the mailbox again
  1743. _ = gen_event:which_handlers(lager_event),
  1744. timer:sleep(Sleep),
  1745. lager:info("hello world"),
  1746. _ = gen_event:which_handlers(lager_event),
  1747. timer:sleep(Sleep),
  1748. %% async is true again now that the mailbox has drained
  1749. ?assertEqual(true, lager_config:get(async)),
  1750. ok
  1751. end}}
  1752. ]}.
  1753. % Fire off the stuffers with minimal resource overhead - speed is of the essence.
  1754. spawn_stuffers(0, _, Refs) ->
  1755. % Attempt to return them in about the order that they'll finish.
  1756. lists:reverse(Refs);
  1757. spawn_stuffers(N, Args, Refs) ->
  1758. {_Pid, Ref} = erlang:spawn_monitor(?MODULE, message_stuffer, Args),
  1759. spawn_stuffers((N - 1), Args, [Ref | Refs]).
  1760. % Spawned process to stuff N copies of Message into lager's message queue as fast as possible.
  1761. % Skip using a list function for speed and low memory footprint - don't want to take the
  1762. % resources to create a sequence (or pass one in).
  1763. message_stuffer(N, Level, Message) ->
  1764. message_stuffer_(N, Level, [{pid, erlang:self()}], Message).
  1765. message_stuffer_(0, _, _, _) ->
  1766. ok;
  1767. message_stuffer_(N, Level, Meta, Message) ->
  1768. lager:log(Level, Meta, Message),
  1769. message_stuffer_((N - 1), Level, Meta, Message).
  1770. collect_workers([]) ->
  1771. ok;
  1772. collect_workers([Ref | Refs]) ->
  1773. receive
  1774. {'DOWN', Ref, _, _, _} ->
  1775. collect_workers(Refs)
  1776. end.
  1777. produce_n_error_logger_msgs(N) ->
  1778. lists:foreach(fun(K) ->
  1779. error_logger:error_msg("Foo ~p!", [K])
  1780. end,
  1781. lists:seq(0, N - 1)
  1782. ).
  1783. high_watermark_test_() ->
  1784. {foreach,
  1785. fun() ->
  1786. error_logger:tty(false),
  1787. application:load(lager),
  1788. application:set_env(lager, error_logger_redirect, true),
  1789. application:set_env(lager, handlers, [{lager_test_backend, info}]),
  1790. application:set_env(lager, async_threshold, undefined),
  1791. lager:start()
  1792. end,
  1793. fun(_) ->
  1794. application:stop(lager),
  1795. error_logger:tty(true)
  1796. end,
  1797. [
  1798. {"Nothing dropped when error_logger high watermark is undefined",
  1799. fun() ->
  1800. ok = error_logger_lager_h:set_high_water(undefined),
  1801. timer:sleep(100),
  1802. produce_n_error_logger_msgs(10),
  1803. timer:sleep(500),
  1804. ?assert(count() >= 10)
  1805. end
  1806. },
  1807. {"Mostly dropped according to error_logger high watermark",
  1808. fun() ->
  1809. ok = error_logger_lager_h:set_high_water(5),
  1810. timer:sleep(100),
  1811. produce_n_error_logger_msgs(50),
  1812. timer:sleep(1000),
  1813. ?assert(count() < 20)
  1814. end
  1815. },
  1816. {"Non-notifications are not dropped",
  1817. fun() ->
  1818. ok = error_logger_lager_h:set_high_water(2),
  1819. timer:sleep(100),
  1820. spawn(fun() -> produce_n_error_logger_msgs(300) end),
  1821. timer:sleep(50),
  1822. %% if everything were dropped, this call would be dropped
  1823. %% too, so lets hope it's not
  1824. ?assert(is_integer(count())),
  1825. timer:sleep(1000),
  1826. ?assert(count() < 10)
  1827. end
  1828. }
  1829. ]
  1830. }.
  1831. get_long_sleep_value() ->
  1832. case os:getenv("CI") of
  1833. false ->
  1834. 500;
  1835. _ ->
  1836. 5000
  1837. end.
  1838. -endif.