Nie możesz wybrać więcej, niż 25 tematów Tematy muszą się zaczynać od litery lub cyfry, mogą zawierać myślniki ('-') i mogą mieć do 35 znaków.

1407 wiersze
74 KiB

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