25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1407 lines
74 KiB

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([{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.