Você não pode selecionar mais de 25 tópicos Os tópicos devem começar com uma letra ou um número, podem incluir traços ('-') e podem ter até 35 caracteres.

1379 linhas
73 KiB

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