您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

1511 行
79 KiB

14 年前
14 年前
14 年前
14 年前
14 年前
14 年前
14 年前
14 年前
  1. %% -------------------------------------------------------------------
  2. %%
  3. %% Copyright (c) 2011-2015 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. -include("lager.hrl").
  22. -behaviour(gen_event).
  23. -export([init/1, handle_call/2, handle_event/2, handle_info/2, terminate/2,
  24. code_change/3]).
  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, {level :: list(), buffer :: list(), ignored :: term()}).
  28. -compile({parse_transform, lager_transform}).
  29. -ifdef(TEST).
  30. -include_lib("eunit/include/eunit.hrl").
  31. -record(test, {attrs :: list(), format :: list(), args :: list()}).
  32. -export([pop/0, count/0, count_ignored/0, flush/0, print_state/0]).
  33. -endif.
  34. init(Level) ->
  35. {ok, #state{level=lager_util:config_to_mask(Level), buffer=[], ignored=[]}}.
  36. handle_call(count, #state{buffer=Buffer} = State) ->
  37. {ok, length(Buffer), State};
  38. handle_call(count_ignored, #state{ignored=Ignored} = State) ->
  39. {ok, length(Ignored), State};
  40. handle_call(flush, State) ->
  41. {ok, ok, State#state{buffer=[], ignored=[]}};
  42. handle_call(pop, #state{buffer=Buffer} = State) ->
  43. case Buffer of
  44. [] ->
  45. {ok, undefined, State};
  46. [H|T] ->
  47. {ok, H, State#state{buffer=T}}
  48. end;
  49. handle_call(get_loglevel, #state{level=Level} = State) ->
  50. {ok, Level, State};
  51. handle_call({set_loglevel, Level}, State) ->
  52. {ok, ok, State#state{level=lager_util:config_to_mask(Level)}};
  53. handle_call(print_state, State) ->
  54. spawn(fun() -> lager:info("State ~p", [lager:pr(State, ?MODULE)]) end),
  55. timer:sleep(100),
  56. {ok, ok, State};
  57. handle_call(print_bad_state, State) ->
  58. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, ?MODULE)]) end),
  59. timer:sleep(100),
  60. {ok, ok, State};
  61. handle_call(_Request, State) ->
  62. {ok, ok, State}.
  63. handle_event({log, Msg},
  64. #state{level=LogLevel,buffer=Buffer,ignored=Ignored} = State) ->
  65. case lager_util:is_loggable(Msg, LogLevel, ?MODULE) of
  66. true ->
  67. {ok, State#state{buffer=Buffer ++
  68. [{lager_msg:severity_as_int(Msg),
  69. lager_msg:datetime(Msg),
  70. lager_msg:message(Msg), lager_msg:metadata(Msg)}]}};
  71. _ ->
  72. {ok, State#state{ignored=Ignored ++ [ignored]}}
  73. end;
  74. handle_event(_Event, State) ->
  75. {ok, State}.
  76. handle_info(_Info, State) ->
  77. {ok, State}.
  78. terminate(_Reason, _State) ->
  79. ok.
  80. code_change(_OldVsn, State, _Extra) ->
  81. {ok, State}.
  82. -ifdef(TEST).
  83. pop() ->
  84. pop(lager_event).
  85. count() ->
  86. count(lager_event).
  87. count_ignored() ->
  88. count_ignored(lager_event).
  89. flush() ->
  90. flush(lager_event).
  91. print_state() ->
  92. print_state(lager_event).
  93. print_bad_state() ->
  94. print_bad_state(lager_event).
  95. pop(Sink) ->
  96. gen_event:call(Sink, ?MODULE, pop).
  97. count(Sink) ->
  98. gen_event:call(Sink, ?MODULE, count).
  99. count_ignored(Sink) ->
  100. gen_event:call(Sink, ?MODULE, count_ignored).
  101. flush(Sink) ->
  102. gen_event:call(Sink, ?MODULE, flush).
  103. print_state(Sink) ->
  104. gen_event:call(Sink, ?MODULE, print_state).
  105. print_bad_state(Sink) ->
  106. gen_event:call(Sink, ?MODULE, print_bad_state).
  107. has_line_numbers() ->
  108. %% are we R15 or greater
  109. % this gets called a LOT - cache the answer
  110. case erlang:get({?MODULE, has_line_numbers}) of
  111. undefined ->
  112. R = otp_version() >= 15,
  113. erlang:put({?MODULE, has_line_numbers}, R),
  114. R;
  115. Bool ->
  116. Bool
  117. end.
  118. otp_version() ->
  119. otp_version(erlang:system_info(otp_release)).
  120. otp_version([$R | Rel]) ->
  121. {Ver, _} = string:to_integer(Rel),
  122. Ver;
  123. otp_version(Rel) ->
  124. {Ver, _} = string:to_integer(Rel),
  125. Ver.
  126. not_running_test() ->
  127. ?assertEqual({error, lager_not_running}, lager:log(info, self(), "not running")).
  128. lager_test_() ->
  129. {foreach,
  130. fun setup/0,
  131. fun cleanup/1,
  132. [
  133. {"observe that there is nothing up my sleeve",
  134. fun() ->
  135. ?assertEqual(undefined, pop()),
  136. ?assertEqual(0, count())
  137. end
  138. },
  139. {"test sink not running",
  140. fun() ->
  141. ?assertEqual({error, {sink_not_configured, test}}, lager:log(test, info, self(), "~p", "not running"))
  142. end
  143. },
  144. {"logging works",
  145. fun() ->
  146. lager:warning("test message"),
  147. ?assertEqual(1, count()),
  148. {Level, _Time, Message, _Metadata} = pop(),
  149. ?assertMatch(Level, lager_util:level_to_num(warning)),
  150. ?assertEqual("test message", Message),
  151. ok
  152. end
  153. },
  154. {"unsafe logging works",
  155. fun() ->
  156. lager:warning_unsafe("test message"),
  157. ?assertEqual(1, count()),
  158. {Level, _Time, Message, _Metadata} = pop(),
  159. ?assertMatch(Level, lager_util:level_to_num(warning)),
  160. ?assertEqual("test message", Message),
  161. ok
  162. end
  163. },
  164. {"logging with arguments works",
  165. fun() ->
  166. lager:warning("test message ~p", [self()]),
  167. ?assertEqual(1, count()),
  168. {Level, _Time, Message,_Metadata} = pop(),
  169. ?assertMatch(Level, lager_util:level_to_num(warning)),
  170. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  171. ok
  172. end
  173. },
  174. {"unsafe logging with args works",
  175. fun() ->
  176. lager:warning("test message ~p", [self()]),
  177. ?assertEqual(1, count()),
  178. {Level, _Time, Message,_Metadata} = pop(),
  179. ?assertMatch(Level, lager_util:level_to_num(warning)),
  180. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  181. ok
  182. end
  183. },
  184. {"logging works from inside a begin/end block",
  185. fun() ->
  186. ?assertEqual(0, count()),
  187. begin
  188. lager:warning("test message 2")
  189. end,
  190. ?assertEqual(1, count()),
  191. ok
  192. end
  193. },
  194. {"logging works from inside a list comprehension",
  195. fun() ->
  196. ?assertEqual(0, count()),
  197. [lager:warning("test message") || _N <- lists:seq(1, 10)],
  198. ?assertEqual(10, count()),
  199. ok
  200. end
  201. },
  202. {"logging works from a begin/end block inside a list comprehension",
  203. fun() ->
  204. ?assertEqual(0, count()),
  205. [ begin lager:warning("test message") end || _N <- lists:seq(1, 10)],
  206. ?assertEqual(10, count()),
  207. ok
  208. end
  209. },
  210. {"logging works from a nested list comprehension",
  211. fun() ->
  212. ?assertEqual(0, count()),
  213. [ [lager:warning("test message") || _N <- lists:seq(1, 10)] ||
  214. _I <- lists:seq(1, 10)],
  215. ?assertEqual(100, count()),
  216. ok
  217. end
  218. },
  219. {"variables inplace of literals in logging statements work",
  220. fun() ->
  221. ?assertEqual(0, count()),
  222. Attr = [{a, alpha}, {b, beta}],
  223. Fmt = "format ~p",
  224. Args = [world],
  225. lager:info(Attr, "hello"),
  226. lager:info(Attr, "hello ~p", [world]),
  227. lager:info(Fmt, [world]),
  228. lager:info("hello ~p", Args),
  229. lager:info(Attr, "hello ~p", Args),
  230. lager:info([{d, delta}, {g, gamma}], Fmt, Args),
  231. ?assertEqual(6, count()),
  232. {_Level, _Time, Message, Metadata} = pop(),
  233. ?assertMatch([{a, alpha}, {b, beta}|_], Metadata),
  234. ?assertEqual("hello", lists:flatten(Message)),
  235. {_Level, _Time2, Message2, _Metadata2} = pop(),
  236. ?assertEqual("hello world", lists:flatten(Message2)),
  237. {_Level, _Time3, Message3, _Metadata3} = pop(),
  238. ?assertEqual("format world", lists:flatten(Message3)),
  239. {_Level, _Time4, Message4, _Metadata4} = pop(),
  240. ?assertEqual("hello world", lists:flatten(Message4)),
  241. {_Level, _Time5, Message5, _Metadata5} = pop(),
  242. ?assertEqual("hello world", lists:flatten(Message5)),
  243. {_Level, _Time6, Message6, Metadata6} = pop(),
  244. ?assertMatch([{d, delta}, {g, gamma}|_], Metadata6),
  245. ?assertEqual("format world", lists:flatten(Message6)),
  246. ok
  247. end
  248. },
  249. {"list comprehension inplace of literals in logging statements work",
  250. fun() ->
  251. ?assertEqual(0, count()),
  252. Attr = [{a, alpha}, {b, beta}],
  253. Fmt = "format ~p",
  254. Args = [world],
  255. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello"),
  256. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello ~p", [{atom, X} || X <- Args]),
  257. lager:info([X || X <- Fmt], [world]),
  258. lager:info("hello ~p", [{atom, X} || X <- Args]),
  259. lager:info([{K, atom_to_list(V)} || {K, V} <- Attr], "hello ~p", [{atom, X} || X <- Args]),
  260. lager:info([{d, delta}, {g, gamma}], Fmt, [{atom, X} || X <- Args]),
  261. ?assertEqual(6, count()),
  262. {_Level, _Time, Message, Metadata} = pop(),
  263. ?assertMatch([{a, "alpha"}, {b, "beta"}|_], Metadata),
  264. ?assertEqual("hello", lists:flatten(Message)),
  265. {_Level, _Time2, Message2, _Metadata2} = pop(),
  266. ?assertEqual("hello {atom,world}", lists:flatten(Message2)),
  267. {_Level, _Time3, Message3, _Metadata3} = pop(),
  268. ?assertEqual("format world", lists:flatten(Message3)),
  269. {_Level, _Time4, Message4, _Metadata4} = pop(),
  270. ?assertEqual("hello {atom,world}", lists:flatten(Message4)),
  271. {_Level, _Time5, Message5, _Metadata5} = pop(),
  272. ?assertEqual("hello {atom,world}", lists:flatten(Message5)),
  273. {_Level, _Time6, Message6, Metadata6} = pop(),
  274. ?assertMatch([{d, delta}, {g, gamma}|_], Metadata6),
  275. ?assertEqual("format {atom,world}", lists:flatten(Message6)),
  276. ok
  277. end
  278. },
  279. {"function calls inplace of literals in logging statements work",
  280. fun() ->
  281. ?assertEqual(0, count()),
  282. put(attrs, [{a, alpha}, {b, beta}]),
  283. put(format, "format ~p"),
  284. put(args, [world]),
  285. lager:info(get(attrs), "hello"),
  286. lager:info(get(attrs), "hello ~p", get(args)),
  287. lager:info(get(format), [world]),
  288. lager:info("hello ~p", erlang:get(args)),
  289. lager:info(fun() -> get(attrs) end(), "hello ~p", get(args)),
  290. lager:info([{d, delta}, {g, gamma}], get(format), get(args)),
  291. ?assertEqual(6, count()),
  292. {_Level, _Time, Message, Metadata} = pop(),
  293. ?assertMatch([{a, alpha}, {b, beta}|_], Metadata),
  294. ?assertEqual("hello", lists:flatten(Message)),
  295. {_Level, _Time2, Message2, _Metadata2} = pop(),
  296. ?assertEqual("hello world", lists:flatten(Message2)),
  297. {_Level, _Time3, Message3, _Metadata3} = pop(),
  298. ?assertEqual("format world", lists:flatten(Message3)),
  299. {_Level, _Time4, Message4, _Metadata4} = pop(),
  300. ?assertEqual("hello world", lists:flatten(Message4)),
  301. {_Level, _Time5, Message5, _Metadata5} = pop(),
  302. ?assertEqual("hello world", lists:flatten(Message5)),
  303. {_Level, _Time6, Message6, Metadata6} = pop(),
  304. ?assertMatch([{d, delta}, {g, gamma}|_], Metadata6),
  305. ?assertEqual("format world", lists:flatten(Message6)),
  306. ok
  307. end
  308. },
  309. {"record fields inplace of literals in logging statements work",
  310. fun() ->
  311. ?assertEqual(0, count()),
  312. Test = #test{attrs=[{a, alpha}, {b, beta}], format="format ~p", args=[world]},
  313. lager:info(Test#test.attrs, "hello"),
  314. lager:info(Test#test.attrs, "hello ~p", Test#test.args),
  315. lager:info(Test#test.format, [world]),
  316. lager:info("hello ~p", Test#test.args),
  317. lager:info(Test#test.attrs, "hello ~p", Test#test.args),
  318. lager:info([{d, delta}, {g, gamma}], Test#test.format, Test#test.args),
  319. ?assertEqual(6, count()),
  320. {_Level, _Time, Message, Metadata} = pop(),
  321. ?assertMatch([{a, alpha}, {b, beta}|_], Metadata),
  322. ?assertEqual("hello", lists:flatten(Message)),
  323. {_Level, _Time2, Message2, _Metadata2} = pop(),
  324. ?assertEqual("hello world", lists:flatten(Message2)),
  325. {_Level, _Time3, Message3, _Metadata3} = pop(),
  326. ?assertEqual("format world", lists:flatten(Message3)),
  327. {_Level, _Time4, Message4, _Metadata4} = pop(),
  328. ?assertEqual("hello world", lists:flatten(Message4)),
  329. {_Level, _Time5, Message5, _Metadata5} = pop(),
  330. ?assertEqual("hello world", lists:flatten(Message5)),
  331. {_Level, _Time6, Message6, Metadata6} = pop(),
  332. ?assertMatch([{d, delta}, {g, gamma}|_], Metadata6),
  333. ?assertEqual("format world", lists:flatten(Message6)),
  334. ok
  335. end
  336. },
  337. {"log messages below the threshold are ignored",
  338. fun() ->
  339. ?assertEqual(0, count()),
  340. lager:debug("this message will be ignored"),
  341. ?assertEqual(0, count()),
  342. ?assertEqual(0, count_ignored()),
  343. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(debug)), []}),
  344. lager:debug("this message should be ignored"),
  345. ?assertEqual(0, count()),
  346. ?assertEqual(1, count_ignored()),
  347. lager:set_loglevel(?MODULE, debug),
  348. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  349. lager:debug("this message should be logged"),
  350. ?assertEqual(1, count()),
  351. ?assertEqual(1, count_ignored()),
  352. ?assertEqual(debug, lager:get_loglevel(?MODULE)),
  353. ok
  354. end
  355. },
  356. {"tracing works",
  357. fun() ->
  358. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  359. ok = lager:info("hello world"),
  360. ?assertEqual(0, count()),
  361. lager:trace(?MODULE, [{module, ?MODULE}], debug),
  362. ?assertMatch({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, _}, lager_config:get(loglevel)),
  363. %% elegible for tracing
  364. ok = lager:info("hello world"),
  365. %% NOT elegible for tracing
  366. ok = lager:log(info, [{pid, self()}], "hello world"),
  367. ?assertEqual(1, count()),
  368. ok
  369. end
  370. },
  371. {"tracing works with custom attributes",
  372. fun() ->
  373. lager:set_loglevel(?MODULE, error),
  374. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  375. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  376. lager:info([{requestid, 6}], "hello world"),
  377. ?assertEqual(0, count()),
  378. lager:trace(?MODULE, [{requestid, 6}, {foo, bar}], debug),
  379. lager:info([{requestid, 6}, {foo, bar}], "hello world"),
  380. ?assertEqual(1, count()),
  381. lager:trace(?MODULE, [{requestid, '*'}], debug),
  382. lager:info([{requestid, 6}], "hello world"),
  383. ?assertEqual(2, count()),
  384. lager:clear_all_traces(),
  385. lager:info([{requestid, 6}], "hello world"),
  386. ?assertEqual(2, count()),
  387. ok
  388. end
  389. },
  390. {"tracing works with custom attributes and event stream processing",
  391. fun() ->
  392. lager:set_loglevel(?MODULE, error),
  393. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  394. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  395. lager:info([{requestid, 6}], "hello world"),
  396. ?assertEqual(0, count()),
  397. lager:trace(?MODULE, [{requestid, '>', 5}, {requestid, '<', 7}, {foo, bar}], debug),
  398. lager:info([{requestid, 5}, {foo, bar}], "hello world"),
  399. lager:info([{requestid, 6}, {foo, bar}], "hello world"),
  400. ?assertEqual(1, count()),
  401. lager:clear_all_traces(),
  402. lager:trace(?MODULE, [{requestid, '>', 8}, {foo, bar}]),
  403. lager:info([{foo, bar}], "hello world"),
  404. lager:info([{requestid, 6}], "hello world"),
  405. lager:info([{requestid, 7}], "hello world"),
  406. lager:info([{requestid, 8}], "hello world"),
  407. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  408. lager:info([{requestid, 10}], "hello world"),
  409. ?assertEqual(2, count()),
  410. lager:trace(?MODULE, [{requestid, '>', 8}]),
  411. lager:info([{foo, bar}], "hello world"),
  412. lager:info([{requestid, 6}], "hello world"),
  413. lager:info([{requestid, 7}], "hello world"),
  414. lager:info([{requestid, 8}], "hello world"),
  415. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  416. lager:info([{requestid, 10}], "hello world"),
  417. ?assertEqual(4, count()),
  418. lager:trace(?MODULE, [{foo, '=', bar}]),
  419. lager:info([{foo, bar}], "hello world"),
  420. lager:info([{requestid, 6}], "hello world"),
  421. lager:info([{requestid, 7}], "hello world"),
  422. lager:info([{requestid, 8}], "hello world"),
  423. lager:info([{requestid, 9}, {foo, bar}], "hello world"),
  424. lager:info([{requestid, 10}], "hello world"),
  425. lager:trace(?MODULE, [{fu, '!'}]),
  426. lager:info([{foo, bar}], "hello world"),
  427. lager:info([{ooh, car}], "hello world"),
  428. lager:info([{fu, bar}], "hello world"),
  429. lager:trace(?MODULE, [{fu, '*'}]),
  430. lager:info([{fu, bar}], "hello world"),
  431. ?assertEqual(10, count()),
  432. lager:clear_all_traces(),
  433. lager:info([{requestid, 6}], "hello world"),
  434. ?assertEqual(10, count()),
  435. ok
  436. end
  437. },
  438. {"tracing custom attributes works with event stream processing statistics and reductions",
  439. fun() ->
  440. lager:set_loglevel(?MODULE, error),
  441. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  442. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(error)), []}),
  443. lager:info([{requestid, 6}], "hello world"),
  444. ?assertEqual(0, count()),
  445. lager:trace(?MODULE, [{beta, '*'}]),
  446. lager:trace(?MODULE, [{meta, "data"}]),
  447. lager:info([{meta, "data"}], "hello world"),
  448. lager:info([{beta, 2}], "hello world"),
  449. lager:info([{beta, 2.1}, {foo, bar}], "hello world"),
  450. lager:info([{meta, <<"data">>}], "hello world"),
  451. ?assertEqual(8, ?DEFAULT_TRACER:info(input)),
  452. ?assertEqual(6, ?DEFAULT_TRACER:info(output)),
  453. ?assertEqual(2, ?DEFAULT_TRACER:info(filter)),
  454. lager:clear_all_traces(),
  455. lager:trace(?MODULE, [{meta, "data"}]),
  456. lager:trace(?MODULE, [{beta, '>', 2}, {beta, '<', 2.12}]),
  457. lager:info([{meta, "data"}], "hello world"),
  458. lager:info([{beta, 2}], "hello world"),
  459. lager:info([{beta, 2.1}, {foo, bar}], "hello world"),
  460. lager:info([{meta, <<"data">>}], "hello world"),
  461. ?assertEqual(8, ?DEFAULT_TRACER:info(input)),
  462. ?assertEqual(4, ?DEFAULT_TRACER:info(output)),
  463. ?assertEqual(4, ?DEFAULT_TRACER:info(filter)),
  464. lager:clear_all_traces(),
  465. lager:trace_console([{beta, '>', 2}, {meta, "data"}]),
  466. lager:trace_console([{beta, '>', 2}, {beta, '<', 2.12}]),
  467. Reduced = {all,[{any,[{beta,'<',2.12},{meta,'=',"data"}]},
  468. {beta,'>',2}]},
  469. ?assertEqual(Reduced, ?DEFAULT_TRACER:info('query')),
  470. lager:clear_all_traces(),
  471. lager:info([{requestid, 6}], "hello world"),
  472. ?assertEqual(5, count()),
  473. ok
  474. end
  475. },
  476. {"persistent traces work",
  477. fun() ->
  478. ?assertEqual(0, count()),
  479. lager:debug([{foo, bar}], "hello world"),
  480. ?assertEqual(0, count()),
  481. application:stop(lager),
  482. application:set_env(lager, traces, [{lager_test_backend, [{foo, bar}], debug}]),
  483. lager:start(),
  484. lager:debug([{foo, bar}], "hello world"),
  485. ?assertEqual(1, count()),
  486. application:unset_env(lager, traces),
  487. ok
  488. end
  489. },
  490. {"tracing honors loglevel",
  491. fun() ->
  492. lager:set_loglevel(?MODULE, error),
  493. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  494. {ok, T} = lager:trace(?MODULE, [{module, ?MODULE}], notice),
  495. ok = lager:info("hello world"),
  496. ?assertEqual(0, count()),
  497. ok = lager:notice("hello world"),
  498. ?assertEqual(1, count()),
  499. lager:stop_trace(T),
  500. ok = lager:notice("hello world"),
  501. ?assertEqual(1, count()),
  502. ok
  503. end
  504. },
  505. {"record printing works",
  506. fun() ->
  507. print_state(),
  508. {Level, _Time, Message, _Metadata} = pop(),
  509. ?assertMatch(Level, lager_util:level_to_num(info)),
  510. {mask, Mask} = lager_util:config_to_mask(info),
  511. ?assertEqual("State #state{level={mask,"++integer_to_list(Mask)++"},buffer=[],ignored=[]}", lists:flatten(Message)),
  512. ok
  513. end
  514. },
  515. {"record printing fails gracefully",
  516. fun() ->
  517. print_bad_state(),
  518. {Level, _Time, Message, _Metadata} = pop(),
  519. ?assertMatch(Level, lager_util:level_to_num(info)),
  520. ?assertEqual("State {state,1}", lists:flatten(Message)),
  521. ok
  522. end
  523. },
  524. {"record printing fails gracefully when no lager_record attribute",
  525. fun() ->
  526. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, lager)]) end),
  527. timer:sleep(100),
  528. {Level, _Time, Message, _Metadata} = pop(),
  529. ?assertMatch(Level, lager_util:level_to_num(info)),
  530. ?assertEqual("State {state,1}", lists:flatten(Message)),
  531. ok
  532. end
  533. },
  534. {"record printing fails gracefully when input is not a tuple",
  535. fun() ->
  536. spawn(fun() -> lager:info("State ~p", [lager:pr(ok, lager)]) end),
  537. timer:sleep(100),
  538. {Level, _Time, Message, _Metadata} = pop(),
  539. ?assertMatch(Level, lager_util:level_to_num(info)),
  540. ?assertEqual("State ok", lists:flatten(Message)),
  541. ok
  542. end
  543. },
  544. {"record printing fails gracefully when module is invalid",
  545. fun() ->
  546. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, not_a_module)]) end),
  547. timer:sleep(1000),
  548. {Level, _Time, Message, _Metadata} = pop(),
  549. ?assertMatch(Level, lager_util:level_to_num(info)),
  550. ?assertEqual("State {state,1}", lists:flatten(Message)),
  551. ok
  552. end
  553. },
  554. {"installing a new handler adjusts the global loglevel if necessary",
  555. fun() ->
  556. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  557. supervisor:start_child(lager_handler_watcher_sup, [lager_event, {?MODULE, foo}, debug]),
  558. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  559. ok
  560. end
  561. },
  562. {"metadata in the process dictionary works",
  563. fun() ->
  564. lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}]),
  565. lager:info("I sing the animal kingdom electric!"),
  566. {_Level, _Time, _Message, Metadata} = pop(),
  567. ?assertEqual(gravid, proplists:get_value(platypus, Metadata)),
  568. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata)),
  569. ?assertEqual(erroneous, proplists:get_value(duck, Metadata)),
  570. ?assertEqual(undefined, proplists:get_value(eagle, Metadata)),
  571. lager:md([{platypus, gravid}, {sloth, hirsute}, {eagle, superincumbent}]),
  572. lager:info("I sing the animal kingdom dielectric!"),
  573. {_Level2, _Time2, _Message2, Metadata2} = pop(),
  574. ?assertEqual(gravid, proplists:get_value(platypus, Metadata2)),
  575. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata2)),
  576. ?assertEqual(undefined, proplists:get_value(duck, Metadata2)),
  577. ?assertEqual(superincumbent, proplists:get_value(eagle, Metadata2)),
  578. ok
  579. end
  580. },
  581. {"unsafe messages really are not truncated",
  582. fun() ->
  583. lager:info_unsafe("doom, doom has come upon you all ~p", [string:copies("doom", 1500)]),
  584. {_, _, Msg,_Metadata} = pop(),
  585. ?assert(length(lists:flatten(Msg)) == 6035)
  586. end
  587. },
  588. {"can't store invalid metadata",
  589. fun() ->
  590. ?assertEqual(ok, lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}])),
  591. ?assertError(badarg, lager:md({flamboyant, flamingo})),
  592. ?assertError(badarg, lager:md("zookeeper zephyr")),
  593. ok
  594. end
  595. }
  596. ]
  597. }.
  598. extra_sinks_test_() ->
  599. {foreach,
  600. fun setup_sink/0,
  601. fun cleanup/1,
  602. [
  603. {"observe that there is nothing up my sleeve",
  604. fun() ->
  605. ?assertEqual(undefined, pop(?TEST_SINK_EVENT)),
  606. ?assertEqual(0, count(?TEST_SINK_EVENT))
  607. end
  608. },
  609. {"logging works",
  610. fun() ->
  611. ?TEST_SINK_NAME:warning("test message"),
  612. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  613. {Level, _Time, Message, _Metadata} = pop(?TEST_SINK_EVENT),
  614. ?assertMatch(Level, lager_util:level_to_num(warning)),
  615. ?assertEqual("test message", Message),
  616. ok
  617. end
  618. },
  619. {"logging with arguments works",
  620. fun() ->
  621. ?TEST_SINK_NAME:warning("test message ~p", [self()]),
  622. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  623. {Level, _Time, Message,_Metadata} = pop(?TEST_SINK_EVENT),
  624. ?assertMatch(Level, lager_util:level_to_num(warning)),
  625. ?assertEqual(lists:flatten(io_lib:format("test message ~p", [self()])), lists:flatten(Message)),
  626. ok
  627. end
  628. },
  629. {"variables inplace of literals in logging statements work",
  630. fun() ->
  631. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  632. Attr = [{a, alpha}, {b, beta}],
  633. Fmt = "format ~p",
  634. Args = [world],
  635. ?TEST_SINK_NAME:info(Attr, "hello"),
  636. ?TEST_SINK_NAME:info(Attr, "hello ~p", [world]),
  637. ?TEST_SINK_NAME:info(Fmt, [world]),
  638. ?TEST_SINK_NAME:info("hello ~p", Args),
  639. ?TEST_SINK_NAME:info(Attr, "hello ~p", Args),
  640. ?TEST_SINK_NAME:info([{d, delta}, {g, gamma}], Fmt, Args),
  641. ?assertEqual(6, count(?TEST_SINK_EVENT)),
  642. {_Level, _Time, Message, Metadata} = pop(?TEST_SINK_EVENT),
  643. ?assertMatch([{a, alpha}, {b, beta}|_], Metadata),
  644. ?assertEqual("hello", lists:flatten(Message)),
  645. {_Level, _Time2, Message2, _Metadata2} = pop(?TEST_SINK_EVENT),
  646. ?assertEqual("hello world", lists:flatten(Message2)),
  647. {_Level, _Time3, Message3, _Metadata3} = pop(?TEST_SINK_EVENT),
  648. ?assertEqual("format world", lists:flatten(Message3)),
  649. {_Level, _Time4, Message4, _Metadata4} = pop(?TEST_SINK_EVENT),
  650. ?assertEqual("hello world", lists:flatten(Message4)),
  651. {_Level, _Time5, Message5, _Metadata5} = pop(?TEST_SINK_EVENT),
  652. ?assertEqual("hello world", lists:flatten(Message5)),
  653. {_Level, _Time6, Message6, Metadata6} = pop(?TEST_SINK_EVENT),
  654. ?assertMatch([{d, delta}, {g, gamma}|_], Metadata6),
  655. ?assertEqual("format world", lists:flatten(Message6)),
  656. ok
  657. end
  658. },
  659. {"log messages below the threshold are ignored",
  660. fun() ->
  661. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  662. ?TEST_SINK_NAME:debug("this message will be ignored"),
  663. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  664. ?assertEqual(0, count_ignored(?TEST_SINK_EVENT)),
  665. lager_config:set({?TEST_SINK_EVENT, loglevel}, {element(2, lager_util:config_to_mask(debug)), []}),
  666. ?TEST_SINK_NAME:debug("this message should be ignored"),
  667. ?assertEqual(0, count(?TEST_SINK_EVENT)),
  668. ?assertEqual(1, count_ignored(?TEST_SINK_EVENT)),
  669. lager:set_loglevel(?TEST_SINK_EVENT, ?MODULE, undefined, debug),
  670. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get({?TEST_SINK_EVENT, loglevel})),
  671. ?TEST_SINK_NAME:debug("this message should be logged"),
  672. ?assertEqual(1, count(?TEST_SINK_EVENT)),
  673. ?assertEqual(1, count_ignored(?TEST_SINK_EVENT)),
  674. ?assertEqual(debug, lager:get_loglevel(?TEST_SINK_EVENT, ?MODULE)),
  675. ok
  676. end
  677. }
  678. ]
  679. }.
  680. setup_sink() ->
  681. error_logger:tty(false),
  682. application:load(lager),
  683. application:set_env(lager, handlers, []),
  684. application:set_env(lager, error_logger_redirect, false),
  685. application:set_env(lager, extra_sinks, [{?TEST_SINK_EVENT, [{handlers, [{?MODULE, info}]}]}]),
  686. lager:start(),
  687. gen_event:call(lager_event, ?MODULE, flush),
  688. gen_event:call(?TEST_SINK_EVENT, ?MODULE, flush).
  689. setup() ->
  690. error_logger:tty(false),
  691. application:load(lager),
  692. application:set_env(lager, handlers, [{?MODULE, info}]),
  693. application:set_env(lager, error_logger_redirect, false),
  694. lager:start(),
  695. gen_event:call(lager_event, ?MODULE, flush).
  696. cleanup(_) ->
  697. application:stop(lager),
  698. application:stop(goldrush),
  699. error_logger:tty(true).
  700. crash(Type) ->
  701. spawn(fun() -> gen_server:call(crash, Type) end),
  702. timer:sleep(100),
  703. _ = gen_event:which_handlers(error_logger),
  704. ok.
  705. test_body(Expected, Actual) ->
  706. case has_line_numbers() of
  707. true ->
  708. ExLen = length(Expected),
  709. {Body, Rest} = case length(Actual) > ExLen of
  710. true ->
  711. {string:substr(Actual, 1, ExLen),
  712. string:substr(Actual, (ExLen + 1))};
  713. _ ->
  714. {Actual, []}
  715. end,
  716. ?assertEqual(Expected, Body),
  717. % OTP-17 (and maybe later releases) may tack on additional info
  718. % about the failure, so if Actual starts with Expected (already
  719. % confirmed by having gotten past assertEqual above) and ends
  720. % with " line NNN" we can ignore what's in-between. By extension,
  721. % since there may not be line information appended at all, any
  722. % text we DO find is reportable, but not a test failure.
  723. case Rest of
  724. [] ->
  725. ok;
  726. _ ->
  727. % isolate the extra data and report it if it's not just
  728. % a line number indicator
  729. case re:run(Rest, "^.*( line \\d+)$", [{capture, [1]}]) of
  730. nomatch ->
  731. ?debugFmt(
  732. "Trailing data \"~s\" following \"~s\"",
  733. [Rest, Expected]);
  734. {match, [{0, _}]} ->
  735. % the whole sting is " line NNN"
  736. ok;
  737. {match, [{Off, _}]} ->
  738. ?debugFmt(
  739. "Trailing data \"~s\" following \"~s\"",
  740. [string:substr(Rest, 1, Off), Expected])
  741. end
  742. end;
  743. _ ->
  744. ?assertEqual(Expected, Actual)
  745. end.
  746. error_logger_redirect_crash_test_() ->
  747. TestBody=fun(Name,CrashReason,Expected) -> {Name,
  748. fun() ->
  749. Pid = whereis(crash),
  750. crash(CrashReason),
  751. {Level, _, Msg,Metadata} = pop(),
  752. test_body(Expected, lists:flatten(Msg)),
  753. ?assertEqual(Pid,proplists:get_value(pid,Metadata)),
  754. ?assertEqual(lager_util:level_to_num(error),Level)
  755. end
  756. }
  757. end,
  758. {foreach,
  759. fun() ->
  760. error_logger:tty(false),
  761. application:load(lager),
  762. application:set_env(lager, error_logger_redirect, true),
  763. application:set_env(lager, handlers, [{?MODULE, error}]),
  764. lager:start(),
  765. crash:start()
  766. end,
  767. fun(_) ->
  768. application:stop(lager),
  769. application:stop(goldrush),
  770. case whereis(crash) of
  771. undefined -> ok;
  772. Pid -> exit(Pid, kill)
  773. end,
  774. error_logger:tty(true)
  775. end,
  776. [
  777. {"again, there is nothing up my sleeve",
  778. fun() ->
  779. ?assertEqual(undefined, pop()),
  780. ?assertEqual(0, count())
  781. end
  782. },
  783. TestBody("bad return value",bad_return,"gen_server crash terminated with reason: bad return value: bleh"),
  784. TestBody("bad return value with string",bad_return_string,"gen_server crash terminated with reason: bad return value: {tuple,{tuple,\"string\"}}"),
  785. TestBody("bad return uncaught throw",throw,"gen_server crash terminated with reason: bad return value: a_ball"),
  786. TestBody("case clause",case_clause,"gen_server crash terminated with reason: no case clause matching {} in crash:handle_call/3"),
  787. TestBody("case clause string",case_clause_string,"gen_server crash terminated with reason: no case clause matching \"crash\" in crash:handle_call/3"),
  788. TestBody("function clause",function_clause,"gen_server crash terminated with reason: no function clause matching crash:function({})"),
  789. TestBody("if clause",if_clause,"gen_server crash terminated with reason: no true branch found while evaluating if expression in crash:handle_call/3"),
  790. TestBody("try clause",try_clause,"gen_server crash terminated with reason: no try clause matching [] in crash:handle_call/3"),
  791. TestBody("undefined function",undef,"gen_server crash terminated with reason: call to undefined function crash:booger/0 from crash:handle_call/3"),
  792. TestBody("bad math",badarith,"gen_server crash terminated with reason: bad arithmetic expression in crash:handle_call/3"),
  793. TestBody("bad match",badmatch,"gen_server crash terminated with reason: no match of right hand value {} in crash:handle_call/3"),
  794. 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"),
  795. TestBody("bad arg1",badarg1,"gen_server crash terminated with reason: bad argument in crash:handle_call/3"),
  796. 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"),
  797. TestBody("bad record",badrecord,"gen_server crash terminated with reason: bad record state in crash:handle_call/3"),
  798. TestBody("noproc",noproc,"gen_server crash terminated with reason: no such process or port in call to gen_event:call(foo, bar, baz)"),
  799. TestBody("badfun",badfun,"gen_server crash terminated with reason: bad function booger in crash:handle_call/3")
  800. ]
  801. }.
  802. error_logger_redirect_test_() ->
  803. {foreach,
  804. fun() ->
  805. error_logger:tty(false),
  806. application:load(lager),
  807. application:set_env(lager, error_logger_redirect, true),
  808. application:set_env(lager, handlers, [{?MODULE, info}]),
  809. lager:start(),
  810. lager:log(error, self(), "flush flush"),
  811. timer:sleep(100),
  812. gen_event:call(lager_event, ?MODULE, flush)
  813. end,
  814. fun(_) ->
  815. application:stop(lager),
  816. application:stop(goldrush),
  817. error_logger:tty(true)
  818. end,
  819. [
  820. {"error reports are printed",
  821. fun() ->
  822. sync_error_logger:error_report([{this, is}, a, {silly, format}]),
  823. _ = gen_event:which_handlers(error_logger),
  824. {Level, _, Msg,Metadata} = pop(),
  825. ?assertEqual(lager_util:level_to_num(error),Level),
  826. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  827. Expected = "this: is, a, silly: format",
  828. ?assertEqual(Expected, lists:flatten(Msg))
  829. end
  830. },
  831. {"string error reports are printed",
  832. fun() ->
  833. sync_error_logger:error_report("this is less silly"),
  834. _ = gen_event:which_handlers(error_logger),
  835. {Level, _, Msg,Metadata} = pop(),
  836. ?assertEqual(lager_util:level_to_num(error),Level),
  837. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  838. Expected = "this is less silly",
  839. ?assertEqual(Expected, lists:flatten(Msg))
  840. end
  841. },
  842. {"error messages are printed",
  843. fun() ->
  844. sync_error_logger:error_msg("doom, doom has come upon you all"),
  845. _ = gen_event:which_handlers(error_logger),
  846. {Level, _, Msg,Metadata} = pop(),
  847. ?assertEqual(lager_util:level_to_num(error),Level),
  848. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  849. Expected = "doom, doom has come upon you all",
  850. ?assertEqual(Expected, lists:flatten(Msg))
  851. end
  852. },
  853. {"error messages with unicode characters in Args are printed",
  854. fun() ->
  855. sync_error_logger:error_msg("~ts", ["Привет!"]),
  856. _ = gen_event:which_handlers(error_logger),
  857. {Level, _, Msg,Metadata} = pop(),
  858. ?assertEqual(lager_util:level_to_num(error),Level),
  859. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  860. ?assertEqual("Привет!", lists:flatten(Msg))
  861. end
  862. },
  863. {"error messages are truncated at 4096 characters",
  864. fun() ->
  865. sync_error_logger:error_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  866. _ = gen_event:which_handlers(error_logger),
  867. {_, _, Msg,_Metadata} = pop(),
  868. ?assert(length(lists:flatten(Msg)) < 5100)
  869. end
  870. },
  871. {"info reports are printed",
  872. fun() ->
  873. sync_error_logger:info_report([{this, is}, a, {silly, format}]),
  874. _ = gen_event:which_handlers(error_logger),
  875. {Level, _, Msg,Metadata} = pop(),
  876. ?assertEqual(lager_util:level_to_num(info),Level),
  877. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  878. Expected = "this: is, a, silly: format",
  879. ?assertEqual(Expected, lists:flatten(Msg))
  880. end
  881. },
  882. {"info reports are truncated at 4096 characters",
  883. fun() ->
  884. sync_error_logger:info_report([[{this, is}, a, {silly, format}] || _ <- lists:seq(0, 600)]),
  885. _ = gen_event:which_handlers(error_logger),
  886. {Level, _, Msg,Metadata} = pop(),
  887. ?assertEqual(lager_util:level_to_num(info),Level),
  888. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  889. ?assert(length(lists:flatten(Msg)) < 5000)
  890. end
  891. },
  892. {"single term info reports are printed",
  893. fun() ->
  894. sync_error_logger:info_report({foolish, bees}),
  895. _ = gen_event:which_handlers(error_logger),
  896. {Level, _, Msg,Metadata} = pop(),
  897. ?assertEqual(lager_util:level_to_num(info),Level),
  898. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  899. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  900. end
  901. },
  902. {"single term error reports are printed",
  903. fun() ->
  904. sync_error_logger:error_report({foolish, bees}),
  905. _ = gen_event:which_handlers(error_logger),
  906. {Level, _, Msg,Metadata} = pop(),
  907. ?assertEqual(lager_util:level_to_num(error),Level),
  908. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  909. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  910. end
  911. },
  912. {"string info reports are printed",
  913. fun() ->
  914. sync_error_logger:info_report("this is less silly"),
  915. _ = gen_event:which_handlers(error_logger),
  916. {Level, _, Msg,Metadata} = pop(),
  917. ?assertEqual(lager_util:level_to_num(info),Level),
  918. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  919. ?assertEqual("this is less silly", lists:flatten(Msg))
  920. end
  921. },
  922. {"string info reports are truncated at 4096 characters",
  923. fun() ->
  924. sync_error_logger:info_report(string:copies("this is less silly", 1000)),
  925. _ = gen_event:which_handlers(error_logger),
  926. {Level, _, Msg,Metadata} = pop(),
  927. ?assertEqual(lager_util:level_to_num(info),Level),
  928. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  929. ?assert(length(lists:flatten(Msg)) < 5100)
  930. end
  931. },
  932. {"strings in a mixed report are printed as strings",
  933. fun() ->
  934. sync_error_logger:info_report(["this is less silly", {than, "this"}]),
  935. _ = gen_event:which_handlers(error_logger),
  936. {Level, _, Msg,Metadata} = pop(),
  937. ?assertEqual(lager_util:level_to_num(info),Level),
  938. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  939. ?assertEqual("\"this is less silly\", than: \"this\"", lists:flatten(Msg))
  940. end
  941. },
  942. {"info messages are printed",
  943. fun() ->
  944. sync_error_logger:info_msg("doom, doom has come upon you all"),
  945. _ = gen_event:which_handlers(error_logger),
  946. {Level, _, Msg,Metadata} = pop(),
  947. ?assertEqual(lager_util:level_to_num(info),Level),
  948. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  949. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  950. end
  951. },
  952. {"info messages are truncated at 4096 characters",
  953. fun() ->
  954. sync_error_logger:info_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  955. _ = gen_event:which_handlers(error_logger),
  956. {Level, _, Msg,Metadata} = pop(),
  957. ?assertEqual(lager_util:level_to_num(info),Level),
  958. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  959. ?assert(length(lists:flatten(Msg)) < 5100)
  960. end
  961. },
  962. {"info messages with unicode characters in Args are printed",
  963. fun() ->
  964. sync_error_logger:info_msg("~ts", ["Привет!"]),
  965. _ = gen_event:which_handlers(error_logger),
  966. {Level, _, Msg,Metadata} = pop(),
  967. ?assertEqual(lager_util:level_to_num(info),Level),
  968. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  969. ?assertEqual("Привет!", lists:flatten(Msg))
  970. end
  971. },
  972. {"warning messages with unicode characters in Args are printed",
  973. fun() ->
  974. sync_error_logger:warning_msg("~ts", ["Привет!"]),
  975. Map = error_logger:warning_map(),
  976. _ = gen_event:which_handlers(error_logger),
  977. {Level, _, Msg,Metadata} = pop(),
  978. ?assertEqual(lager_util:level_to_num(Map),Level),
  979. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  980. ?assertEqual("Привет!", lists:flatten(Msg))
  981. end
  982. },
  983. {"warning messages are printed at the correct level",
  984. fun() ->
  985. sync_error_logger:warning_msg("doom, doom has come upon you all"),
  986. Map = error_logger:warning_map(),
  987. _ = gen_event:which_handlers(error_logger),
  988. {Level, _, Msg,Metadata} = pop(),
  989. ?assertEqual(lager_util:level_to_num(Map),Level),
  990. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  991. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  992. end
  993. },
  994. {"warning reports are printed at the correct level",
  995. fun() ->
  996. sync_error_logger:warning_report([{i, like}, pie]),
  997. Map = error_logger:warning_map(),
  998. _ = gen_event:which_handlers(error_logger),
  999. {Level, _, Msg,Metadata} = pop(),
  1000. ?assertEqual(lager_util:level_to_num(Map),Level),
  1001. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1002. ?assertEqual("i: like, pie", lists:flatten(Msg))
  1003. end
  1004. },
  1005. {"single term warning reports are printed at the correct level",
  1006. fun() ->
  1007. sync_error_logger:warning_report({foolish, bees}),
  1008. Map = error_logger:warning_map(),
  1009. _ = gen_event:which_handlers(error_logger),
  1010. {Level, _, Msg,Metadata} = pop(),
  1011. ?assertEqual(lager_util:level_to_num(Map),Level),
  1012. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1013. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  1014. end
  1015. },
  1016. {"application stop reports",
  1017. fun() ->
  1018. sync_error_logger:info_report([{application, foo}, {exited, quittin_time}, {type, lazy}]),
  1019. _ = gen_event:which_handlers(error_logger),
  1020. {Level, _, Msg,Metadata} = pop(),
  1021. ?assertEqual(lager_util:level_to_num(info),Level),
  1022. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1023. ?assertEqual("Application foo exited with reason: quittin_time", lists:flatten(Msg))
  1024. end
  1025. },
  1026. {"supervisor reports",
  1027. fun() ->
  1028. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  1029. _ = gen_event:which_handlers(error_logger),
  1030. {Level, _, Msg,Metadata} = pop(),
  1031. ?assertEqual(lager_util:level_to_num(error),Level),
  1032. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1033. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason fired in context france", lists:flatten(Msg))
  1034. end
  1035. },
  1036. {"supervisor reports with real error",
  1037. fun() ->
  1038. 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}}]),
  1039. _ = gen_event:which_handlers(error_logger),
  1040. {Level, _, Msg,Metadata} = pop(),
  1041. ?assertEqual(lager_util:level_to_num(error),Level),
  1042. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1043. ?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))
  1044. end
  1045. },
  1046. {"supervisor reports with real error and pid",
  1047. fun() ->
  1048. 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}]),
  1049. _ = gen_event:which_handlers(error_logger),
  1050. {Level, _, Msg,Metadata} = pop(),
  1051. ?assertEqual(lager_util:level_to_num(error),Level),
  1052. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1053. ?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))
  1054. end
  1055. },
  1056. {"supervisor_bridge reports",
  1057. fun() ->
  1058. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{mod, mini_steve}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  1059. _ = gen_event:which_handlers(error_logger),
  1060. {Level, _, Msg,Metadata} = pop(),
  1061. ?assertEqual(lager_util:level_to_num(error),Level),
  1062. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1063. ?assertEqual("Supervisor steve had child at module mini_steve at bleh exit with reason fired in context france", lists:flatten(Msg))
  1064. end
  1065. },
  1066. {"application progress report",
  1067. fun() ->
  1068. sync_error_logger:info_report(progress, [{application, foo}, {started_at, node()}]),
  1069. _ = gen_event:which_handlers(error_logger),
  1070. {Level, _, Msg,Metadata} = pop(),
  1071. ?assertEqual(lager_util:level_to_num(info),Level),
  1072. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1073. Expected = lists:flatten(io_lib:format("Application foo started on node ~w", [node()])),
  1074. ?assertEqual(Expected, lists:flatten(Msg))
  1075. end
  1076. },
  1077. {"supervisor progress report",
  1078. fun() ->
  1079. lager:set_loglevel(?MODULE, debug),
  1080. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1081. sync_error_logger:info_report(progress, [{supervisor, {local, foo}}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1082. _ = gen_event:which_handlers(error_logger),
  1083. {Level, _, Msg,Metadata} = pop(),
  1084. ?assertEqual(lager_util:level_to_num(debug),Level),
  1085. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1086. ?assertEqual("Supervisor foo started foo:bar/1 at pid baz", lists:flatten(Msg))
  1087. end
  1088. },
  1089. {"supervisor progress report with pid",
  1090. fun() ->
  1091. lager:set_loglevel(?MODULE, debug),
  1092. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1093. sync_error_logger:info_report(progress, [{supervisor, somepid}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1094. _ = gen_event:which_handlers(error_logger),
  1095. {Level, _, Msg,Metadata} = pop(),
  1096. ?assertEqual(lager_util:level_to_num(debug),Level),
  1097. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1098. ?assertEqual("Supervisor somepid started foo:bar/1 at pid baz", lists:flatten(Msg))
  1099. end
  1100. },
  1101. {"crash report for emfile",
  1102. fun() ->
  1103. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, emfile, [{stack, trace, 1}]}}], []]),
  1104. _ = gen_event:which_handlers(error_logger),
  1105. {Level, _, Msg,Metadata} = pop(),
  1106. ?assertEqual(lager_util:level_to_num(error),Level),
  1107. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1108. 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()])),
  1109. ?assertEqual(Expected, lists:flatten(Msg))
  1110. end
  1111. },
  1112. {"crash report for system process limit",
  1113. fun() ->
  1114. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn, 1}]}}], []]),
  1115. _ = gen_event:which_handlers(error_logger),
  1116. {Level, _, Msg,Metadata} = pop(),
  1117. ?assertEqual(lager_util:level_to_num(error),Level),
  1118. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1119. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1120. ?assertEqual(Expected, lists:flatten(Msg))
  1121. end
  1122. },
  1123. {"crash report for system process limit2",
  1124. fun() ->
  1125. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn_opt, 1}]}}], []]),
  1126. _ = gen_event:which_handlers(error_logger),
  1127. {Level, _, Msg,Metadata} = pop(),
  1128. ?assertEqual(lager_util:level_to_num(error),Level),
  1129. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1130. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1131. ?assertEqual(Expected, lists:flatten(Msg))
  1132. end
  1133. },
  1134. {"crash report for system port limit",
  1135. fun() ->
  1136. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, open_port, 1}]}}], []]),
  1137. _ = gen_event:which_handlers(error_logger),
  1138. {Level, _, Msg,Metadata} = pop(),
  1139. ?assertEqual(lager_util:level_to_num(error),Level),
  1140. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1141. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ports exceeded", [self()])),
  1142. ?assertEqual(Expected, lists:flatten(Msg))
  1143. end
  1144. },
  1145. {"crash report for system port limit",
  1146. fun() ->
  1147. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, list_to_atom, 1}]}}], []]),
  1148. _ = gen_event:which_handlers(error_logger),
  1149. {Level, _, Msg,Metadata} = pop(),
  1150. ?assertEqual(lager_util:level_to_num(error),Level),
  1151. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1152. 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()])),
  1153. ?assertEqual(Expected, lists:flatten(Msg))
  1154. end
  1155. },
  1156. {"crash report for system ets table limit",
  1157. fun() ->
  1158. 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}]}}], []]),
  1159. _ = gen_event:which_handlers(error_logger),
  1160. {Level, _, Msg,Metadata} = pop(),
  1161. ?assertEqual(lager_util:level_to_num(error),Level),
  1162. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1163. 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])),
  1164. ?assertEqual(Expected, lists:flatten(Msg))
  1165. end
  1166. },
  1167. {"crash report for unknown system limit should be truncated at 500 characters",
  1168. fun() ->
  1169. sync_error_logger:error_report(crash_report, [[{pid, self()}, {error_info, {error, system_limit, [{wtf,boom,[string:copies("aaaa", 4096)]}]}}], []]),
  1170. _ = gen_event:which_handlers(error_logger),
  1171. {_, _, Msg,_Metadata} = pop(),
  1172. ?assert(length(lists:flatten(Msg)) > 550),
  1173. ?assert(length(lists:flatten(Msg)) < 600)
  1174. end
  1175. },
  1176. {"crash reports for 'special processes' should be handled right - function_clause",
  1177. fun() ->
  1178. {ok, Pid} = special_process:start(),
  1179. unlink(Pid),
  1180. Pid ! function_clause,
  1181. timer:sleep(500),
  1182. _ = gen_event:which_handlers(error_logger),
  1183. {_, _, Msg, _Metadata} = pop(),
  1184. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no function clause matching special_process:foo(bar)",
  1185. [Pid])),
  1186. test_body(Expected, lists:flatten(Msg))
  1187. end
  1188. },
  1189. {"crash reports for 'special processes' should be handled right - case_clause",
  1190. fun() ->
  1191. {ok, Pid} = special_process:start(),
  1192. unlink(Pid),
  1193. Pid ! {case_clause, wtf},
  1194. timer:sleep(500),
  1195. _ = gen_event:which_handlers(error_logger),
  1196. {_, _, Msg, _Metadata} = pop(),
  1197. 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",
  1198. [Pid])),
  1199. test_body(Expected, lists:flatten(Msg))
  1200. end
  1201. },
  1202. {"crash reports for 'special processes' should be handled right - exit",
  1203. fun() ->
  1204. {ok, Pid} = special_process:start(),
  1205. unlink(Pid),
  1206. Pid ! exit,
  1207. timer:sleep(500),
  1208. _ = gen_event:which_handlers(error_logger),
  1209. {_, _, Msg, _Metadata} = pop(),
  1210. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours exited with reason: byebye in special_process:loop/0",
  1211. [Pid])),
  1212. test_body(Expected, lists:flatten(Msg))
  1213. end
  1214. },
  1215. {"crash reports for 'special processes' should be handled right - error",
  1216. fun() ->
  1217. {ok, Pid} = special_process:start(),
  1218. unlink(Pid),
  1219. Pid ! error,
  1220. timer:sleep(500),
  1221. _ = gen_event:which_handlers(error_logger),
  1222. {_, _, Msg, _Metadata} = pop(),
  1223. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: mybad in special_process:loop/0",
  1224. [Pid])),
  1225. test_body(Expected, lists:flatten(Msg))
  1226. end
  1227. },
  1228. {"webmachine error reports",
  1229. fun() ->
  1230. Path = "/cgi-bin/phpmyadmin",
  1231. Reason = {error,{error,{badmatch,{error,timeout}},
  1232. [{myapp,dostuff,2,[{file,"src/myapp.erl"},{line,123}]},
  1233. {webmachine_resource,resource_call,3,[{file,"src/webmachine_resource.erl"},{line,169}]}]}},
  1234. sync_error_logger:error_msg("webmachine error: path=~p~n~p~n", [Path, Reason]),
  1235. _ = gen_event:which_handlers(error_logger),
  1236. {Level, _, Msg,Metadata} = pop(),
  1237. ?assertEqual(lager_util:level_to_num(error),Level),
  1238. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1239. ?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))
  1240. end
  1241. },
  1242. {"Cowboy error reports, 8 arg version",
  1243. fun() ->
  1244. Stack = [{my_handler,init, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1245. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1246. sync_error_logger:error_msg(
  1247. "** Cowboy handler ~p terminating in ~p/~p~n"
  1248. " for the reason ~p:~p~n"
  1249. "** Options were ~p~n"
  1250. "** Request was ~p~n"
  1251. "** Stacktrace: ~p~n~n",
  1252. [my_handler, init, 3, error, {badmatch, {error, timeout}}, [],
  1253. "Request", Stack]),
  1254. _ = gen_event:which_handlers(error_logger),
  1255. {Level, _, Msg,Metadata} = pop(),
  1256. ?assertEqual(lager_util:level_to_num(error),Level),
  1257. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1258. ?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))
  1259. end
  1260. },
  1261. {"Cowboy error reports, 10 arg version",
  1262. fun() ->
  1263. Stack = [{my_handler,somecallback, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1264. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1265. sync_error_logger:error_msg(
  1266. "** Cowboy handler ~p terminating in ~p/~p~n"
  1267. " for the reason ~p:~p~n** Message was ~p~n"
  1268. "** Options were ~p~n** Handler state was ~p~n"
  1269. "** Request was ~p~n** Stacktrace: ~p~n~n",
  1270. [my_handler, somecallback, 3, error, {badmatch, {error, timeout}}, hello, [],
  1271. {}, "Request", Stack]),
  1272. _ = gen_event:which_handlers(error_logger),
  1273. {Level, _, Msg,Metadata} = pop(),
  1274. ?assertEqual(lager_util:level_to_num(error),Level),
  1275. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1276. ?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))
  1277. end
  1278. },
  1279. {"Cowboy error reports, 5 arg version",
  1280. fun() ->
  1281. sync_error_logger:error_msg(
  1282. "** Cowboy handler ~p terminating; "
  1283. "function ~p/~p was not exported~n"
  1284. "** Request was ~p~n** State was ~p~n~n",
  1285. [my_handler, to_json, 2, "Request", {}]),
  1286. _ = gen_event:which_handlers(error_logger),
  1287. {Level, _, Msg,Metadata} = pop(),
  1288. ?assertEqual(lager_util:level_to_num(error),Level),
  1289. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1290. ?assertEqual("Cowboy handler my_handler terminated with reason: call to undefined function my_handler:to_json/2", lists:flatten(Msg))
  1291. end
  1292. },
  1293. {"messages should not be generated if they don't satisfy the threshold",
  1294. fun() ->
  1295. lager:set_loglevel(?MODULE, error),
  1296. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1297. sync_error_logger:info_report([hello, world]),
  1298. _ = gen_event:which_handlers(error_logger),
  1299. ?assertEqual(0, count()),
  1300. ?assertEqual(0, count_ignored()),
  1301. lager:set_loglevel(?MODULE, info),
  1302. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1303. sync_error_logger:info_report([hello, world]),
  1304. _ = gen_event:which_handlers(error_logger),
  1305. ?assertEqual(1, count()),
  1306. ?assertEqual(0, count_ignored()),
  1307. lager:set_loglevel(?MODULE, error),
  1308. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1309. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(debug)), []}),
  1310. sync_error_logger:info_report([hello, world]),
  1311. _ = gen_event:which_handlers(error_logger),
  1312. ?assertEqual(1, count()),
  1313. ?assertEqual(1, count_ignored())
  1314. end
  1315. }
  1316. ]
  1317. }.
  1318. safe_format_test() ->
  1319. ?assertEqual("foo bar", lists:flatten(lager:safe_format("~p ~p", [foo, bar], 1024))),
  1320. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:safe_format("~p ~p ~p", [foo, bar], 1024))),
  1321. ok.
  1322. unsafe_format_test() ->
  1323. ?assertEqual("foo bar", lists:flatten(lager:unsafe_format("~p ~p", [foo, bar]))),
  1324. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:unsafe_format("~p ~p ~p", [foo, bar]))),
  1325. ok.
  1326. async_threshold_test_() ->
  1327. {foreach,
  1328. fun() ->
  1329. error_logger:tty(false),
  1330. ets:new(async_threshold_test, [set, named_table, public]),
  1331. ets:insert_new(async_threshold_test, {sync_toggled, 0}),
  1332. ets:insert_new(async_threshold_test, {async_toggled, 0}),
  1333. application:load(lager),
  1334. application:set_env(lager, error_logger_redirect, false),
  1335. application:set_env(lager, async_threshold, 2),
  1336. application:set_env(lager, async_threshold_window, 1),
  1337. application:set_env(lager, handlers, [{?MODULE, info}]),
  1338. lager:start()
  1339. end,
  1340. fun(_) ->
  1341. application:unset_env(lager, async_threshold),
  1342. application:stop(lager),
  1343. application:stop(goldrush),
  1344. ets:delete(async_threshold_test),
  1345. error_logger:tty(true)
  1346. end,
  1347. [
  1348. {"async threshold works",
  1349. fun() ->
  1350. %% we start out async
  1351. ?assertEqual(true, lager_config:get(async)),
  1352. %% put a ton of things in the queue
  1353. Workers = [spawn_monitor(fun() -> [lager:info("hello world") || _ <- lists:seq(1, 1000)] end) || _ <- lists:seq(1, 15)],
  1354. %% serialize on mailbox
  1355. _ = gen_event:which_handlers(lager_event),
  1356. timer:sleep(500),
  1357. %% By now the flood of messages will have
  1358. %% forced the backend throttle to turn off
  1359. %% async mode, but it's possible all
  1360. %% outstanding requests have been processed,
  1361. %% so checking the current status (sync or
  1362. %% async) is an exercise in race control.
  1363. %% Instead, we'll see whether the backend
  1364. %% throttle has toggled into sync mode at any
  1365. %% point in the past
  1366. ?assertMatch([{sync_toggled, N}] when N > 0,
  1367. ets:lookup(async_threshold_test, sync_toggled)),
  1368. %% wait for all the workers to return, meaning that all the messages have been logged (since we're definitely in sync mode at the end of the run)
  1369. collect_workers(Workers),
  1370. %% serialize on the mailbox again
  1371. _ = gen_event:which_handlers(lager_event),
  1372. %% just in case...
  1373. timer:sleep(1000),
  1374. lager:info("hello world"),
  1375. _ = gen_event:which_handlers(lager_event),
  1376. %% async is true again now that the mailbox has drained
  1377. ?assertEqual(true, lager_config:get(async)),
  1378. ok
  1379. end
  1380. }
  1381. ]
  1382. }.
  1383. collect_workers([]) ->
  1384. ok;
  1385. collect_workers(Workers) ->
  1386. receive
  1387. {'DOWN', Ref, _, _, _} ->
  1388. collect_workers(lists:keydelete(Ref, 2, Workers))
  1389. end.
  1390. produce_n_error_logger_msgs(N) ->
  1391. lists:foreach(fun (K) ->
  1392. error_logger:error_msg("Foo ~p!", [K])
  1393. end,
  1394. lists:seq(0, N-1)
  1395. ).
  1396. high_watermark_test_() ->
  1397. {foreach,
  1398. fun() ->
  1399. error_logger:tty(false),
  1400. application:load(lager),
  1401. application:set_env(lager, error_logger_redirect, true),
  1402. application:set_env(lager, handlers, [{lager_test_backend, info}]),
  1403. application:set_env(lager, async_threshold, undefined),
  1404. lager:start()
  1405. end,
  1406. fun(_) ->
  1407. application:stop(lager),
  1408. error_logger:tty(true)
  1409. end,
  1410. [
  1411. {"Nothing dropped when error_logger high watermark is undefined",
  1412. fun () ->
  1413. ok = error_logger_lager_h:set_high_water(undefined),
  1414. timer:sleep(100),
  1415. produce_n_error_logger_msgs(10),
  1416. timer:sleep(500),
  1417. ?assert(count() >= 10)
  1418. end
  1419. },
  1420. {"Mostly dropped according to error_logger high watermark",
  1421. fun () ->
  1422. ok = error_logger_lager_h:set_high_water(5),
  1423. timer:sleep(100),
  1424. produce_n_error_logger_msgs(50),
  1425. timer:sleep(1000),
  1426. ?assert(count() < 20)
  1427. end
  1428. },
  1429. {"Non-notifications are not dropped",
  1430. fun () ->
  1431. ok = error_logger_lager_h:set_high_water(2),
  1432. timer:sleep(100),
  1433. spawn(fun () -> produce_n_error_logger_msgs(300) end),
  1434. timer:sleep(50),
  1435. %% if everything were dropped, this call would be dropped
  1436. %% too, so lets hope it's not
  1437. ?assert(is_integer(count())),
  1438. timer:sleep(1000),
  1439. ?assert(count() < 10)
  1440. end
  1441. }
  1442. ]
  1443. }.
  1444. -endif.