Non puoi selezionare più di 25 argomenti Gli argomenti devono iniziare con una lettera o un numero, possono includere trattini ('-') e possono essere lunghi fino a 35 caratteri.

1405 righe
74 KiB

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