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

1361 lines
72 KiB

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