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

1405 line
74 KiB

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