您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

1379 行
73 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. {"stopped trace stops and removes its event handler (gh#267)",
  467. fun() ->
  468. StartHandlers = gen_event:which_handlers(lager_event),
  469. {_, T0} = lager_config:get(loglevel),
  470. ?assertEqual([], T0),
  471. {ok, Test} = lager:trace_file("/tmp/test", [{vhost, test}]),
  472. MidHandlers = gen_event:which_handlers(lager_event),
  473. ?assertEqual(length(StartHandlers)+1, length(MidHandlers)),
  474. {_, T1} = lager_config:get(loglevel),
  475. ?assertEqual(1, length(T1)),
  476. ok = lager:stop_trace(Test),
  477. {_, T2} = lager_config:get(loglevel),
  478. ?assertEqual([], T2),
  479. EndHandlers = gen_event:which_handlers(lager_event),
  480. ?assertEqual(StartHandlers, EndHandlers),
  481. ok
  482. end
  483. },
  484. {"record printing works",
  485. fun() ->
  486. print_state(),
  487. {Level, _Time, Message, _Metadata} = pop(),
  488. ?assertMatch(Level, lager_util:level_to_num(info)),
  489. {mask, Mask} = lager_util:config_to_mask(info),
  490. ?assertEqual("State #state{level={mask,"++integer_to_list(Mask)++"},buffer=[],ignored=[]}", lists:flatten(Message)),
  491. ok
  492. end
  493. },
  494. {"record printing fails gracefully",
  495. fun() ->
  496. print_bad_state(),
  497. {Level, _Time, Message, _Metadata} = pop(),
  498. ?assertMatch(Level, lager_util:level_to_num(info)),
  499. ?assertEqual("State {state,1}", lists:flatten(Message)),
  500. ok
  501. end
  502. },
  503. {"record printing fails gracefully when no lager_record attribute",
  504. fun() ->
  505. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, lager)]) end),
  506. timer:sleep(100),
  507. {Level, _Time, Message, _Metadata} = pop(),
  508. ?assertMatch(Level, lager_util:level_to_num(info)),
  509. ?assertEqual("State {state,1}", lists:flatten(Message)),
  510. ok
  511. end
  512. },
  513. {"record printing fails gracefully when input is not a tuple",
  514. fun() ->
  515. spawn(fun() -> lager:info("State ~p", [lager:pr(ok, lager)]) end),
  516. timer:sleep(100),
  517. {Level, _Time, Message, _Metadata} = pop(),
  518. ?assertMatch(Level, lager_util:level_to_num(info)),
  519. ?assertEqual("State ok", lists:flatten(Message)),
  520. ok
  521. end
  522. },
  523. {"record printing fails gracefully when module is invalid",
  524. fun() ->
  525. spawn(fun() -> lager:info("State ~p", [lager:pr({state, 1}, not_a_module)]) end),
  526. timer:sleep(1000),
  527. {Level, _Time, Message, _Metadata} = pop(),
  528. ?assertMatch(Level, lager_util:level_to_num(info)),
  529. ?assertEqual("State {state,1}", lists:flatten(Message)),
  530. ok
  531. end
  532. },
  533. {"installing a new handler adjusts the global loglevel if necessary",
  534. fun() ->
  535. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  536. supervisor:start_child(lager_handler_watcher_sup, [lager_event, {?MODULE, foo}, debug]),
  537. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  538. ok
  539. end
  540. },
  541. {"metadata in the process dictionary works",
  542. fun() ->
  543. lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}]),
  544. lager:info("I sing the animal kingdom electric!"),
  545. {_Level, _Time, _Message, Metadata} = pop(),
  546. ?assertEqual(gravid, proplists:get_value(platypus, Metadata)),
  547. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata)),
  548. ?assertEqual(erroneous, proplists:get_value(duck, Metadata)),
  549. ?assertEqual(undefined, proplists:get_value(eagle, Metadata)),
  550. lager:md([{platypus, gravid}, {sloth, hirsute}, {eagle, superincumbent}]),
  551. lager:info("I sing the animal kingdom dielectric!"),
  552. {_Level2, _Time2, _Message2, Metadata2} = pop(),
  553. ?assertEqual(gravid, proplists:get_value(platypus, Metadata2)),
  554. ?assertEqual(hirsute, proplists:get_value(sloth, Metadata2)),
  555. ?assertEqual(undefined, proplists:get_value(duck, Metadata2)),
  556. ?assertEqual(superincumbent, proplists:get_value(eagle, Metadata2)),
  557. ok
  558. end
  559. },
  560. {"can't store invalid metadata",
  561. fun() ->
  562. ?assertEqual(ok, lager:md([{platypus, gravid}, {sloth, hirsute}, {duck, erroneous}])),
  563. ?assertError(badarg, lager:md({flamboyant, flamingo})),
  564. ?assertError(badarg, lager:md("zookeeper zephyr")),
  565. ok
  566. end
  567. }
  568. ]
  569. }.
  570. setup() ->
  571. error_logger:tty(false),
  572. application:load(lager),
  573. application:set_env(lager, handlers, [{?MODULE, info}]),
  574. application:set_env(lager, error_logger_redirect, false),
  575. lager:start(),
  576. gen_event:call(lager_event, ?MODULE, flush).
  577. cleanup(_) ->
  578. application:stop(lager),
  579. application:stop(goldrush),
  580. error_logger:tty(true).
  581. crash(Type) ->
  582. spawn(fun() -> gen_server:call(crash, Type) end),
  583. timer:sleep(100),
  584. _ = gen_event:which_handlers(error_logger),
  585. ok.
  586. test_body(Expected, Actual) ->
  587. case has_line_numbers() of
  588. true ->
  589. ExLen = length(Expected),
  590. {Body, Rest} = case length(Actual) > ExLen of
  591. true ->
  592. {string:substr(Actual, 1, ExLen),
  593. string:substr(Actual, (ExLen + 1))};
  594. _ ->
  595. {Actual, []}
  596. end,
  597. ?assertEqual(Expected, Body),
  598. % OTP-17 (and maybe later releases) may tack on additional info
  599. % about the failure, so if Actual starts with Expected (already
  600. % confirmed by having gotten past assertEqual above) and ends
  601. % with " line NNN" we can ignore what's in-between. By extension,
  602. % since there may not be line information appended at all, any
  603. % text we DO find is reportable, but not a test failure.
  604. case Rest of
  605. [] ->
  606. ok;
  607. _ ->
  608. % isolate the extra data and report it if it's not just
  609. % a line number indicator
  610. case re:run(Rest, "^.*( line \\d+)$", [{capture, [1]}]) of
  611. nomatch ->
  612. ?debugFmt(
  613. "Trailing data \"~s\" following \"~s\"",
  614. [Rest, Expected]);
  615. {match, [{0, _}]} ->
  616. % the whole sting is " line NNN"
  617. ok;
  618. {match, [{Off, _}]} ->
  619. ?debugFmt(
  620. "Trailing data \"~s\" following \"~s\"",
  621. [string:substr(Rest, 1, Off), Expected])
  622. end
  623. end;
  624. _ ->
  625. ?assertEqual(Expected, Actual)
  626. end.
  627. error_logger_redirect_crash_test_() ->
  628. TestBody=fun(Name,CrashReason,Expected) -> {Name,
  629. fun() ->
  630. Pid = whereis(crash),
  631. crash(CrashReason),
  632. {Level, _, Msg,Metadata} = pop(),
  633. test_body(Expected, lists:flatten(Msg)),
  634. ?assertEqual(Pid,proplists:get_value(pid,Metadata)),
  635. ?assertEqual(lager_util:level_to_num(error),Level)
  636. end
  637. }
  638. end,
  639. {foreach,
  640. fun() ->
  641. error_logger:tty(false),
  642. application:load(lager),
  643. application:set_env(lager, error_logger_redirect, true),
  644. application:set_env(lager, handlers, [{?MODULE, error}]),
  645. lager:start(),
  646. crash:start()
  647. end,
  648. fun(_) ->
  649. application:stop(lager),
  650. application:stop(goldrush),
  651. case whereis(crash) of
  652. undefined -> ok;
  653. Pid -> exit(Pid, kill)
  654. end,
  655. error_logger:tty(true)
  656. end,
  657. [
  658. {"again, there is nothing up my sleeve",
  659. fun() ->
  660. ?assertEqual(undefined, pop()),
  661. ?assertEqual(0, count())
  662. end
  663. },
  664. TestBody("bad return value",bad_return,"gen_server crash terminated with reason: bad return value: bleh"),
  665. TestBody("bad return value with string",bad_return_string,"gen_server crash terminated with reason: bad return value: {tuple,{tuple,\"string\"}}"),
  666. TestBody("bad return uncaught throw",throw,"gen_server crash terminated with reason: bad return value: a_ball"),
  667. TestBody("case clause",case_clause,"gen_server crash terminated with reason: no case clause matching {} in crash:handle_call/3"),
  668. TestBody("case clause string",case_clause_string,"gen_server crash terminated with reason: no case clause matching \"crash\" in crash:handle_call/3"),
  669. TestBody("function clause",function_clause,"gen_server crash terminated with reason: no function clause matching crash:function({})"),
  670. TestBody("if clause",if_clause,"gen_server crash terminated with reason: no true branch found while evaluating if expression in crash:handle_call/3"),
  671. TestBody("try clause",try_clause,"gen_server crash terminated with reason: no try clause matching [] in crash:handle_call/3"),
  672. TestBody("undefined function",undef,"gen_server crash terminated with reason: call to undefined function crash:booger/0 from crash:handle_call/3"),
  673. TestBody("bad math",badarith,"gen_server crash terminated with reason: bad arithmetic expression in crash:handle_call/3"),
  674. TestBody("bad match",badmatch,"gen_server crash terminated with reason: no match of right hand value {} in crash:handle_call/3"),
  675. TestBody("bad arity",badarity,"gen_server crash terminated with reason: fun called with wrong arity of 1 instead of 3 in crash:handle_call/3"),
  676. TestBody("bad arg1",badarg1,"gen_server crash terminated with reason: bad argument in crash:handle_call/3"),
  677. TestBody("bad arg2",badarg2,"gen_server crash terminated with reason: bad argument in call to erlang:iolist_to_binary([\"foo\",bar]) in crash:handle_call/3"),
  678. TestBody("bad record",badrecord,"gen_server crash terminated with reason: bad record state in crash:handle_call/3"),
  679. TestBody("noproc",noproc,"gen_server crash terminated with reason: no such process or port in call to gen_event:call(foo, bar, baz)"),
  680. TestBody("badfun",badfun,"gen_server crash terminated with reason: bad function booger in crash:handle_call/3")
  681. ]
  682. }.
  683. error_logger_redirect_test_() ->
  684. {foreach,
  685. fun() ->
  686. error_logger:tty(false),
  687. application:load(lager),
  688. application:set_env(lager, error_logger_redirect, true),
  689. application:set_env(lager, handlers, [{?MODULE, info}]),
  690. lager:start(),
  691. lager:log(error, self(), "flush flush"),
  692. timer:sleep(100),
  693. gen_event:call(lager_event, ?MODULE, flush)
  694. end,
  695. fun(_) ->
  696. application:stop(lager),
  697. application:stop(goldrush),
  698. error_logger:tty(true)
  699. end,
  700. [
  701. {"error reports are printed",
  702. fun() ->
  703. sync_error_logger:error_report([{this, is}, a, {silly, format}]),
  704. _ = gen_event:which_handlers(error_logger),
  705. {Level, _, Msg,Metadata} = pop(),
  706. ?assertEqual(lager_util:level_to_num(error),Level),
  707. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  708. Expected = "this: is, a, silly: format",
  709. ?assertEqual(Expected, lists:flatten(Msg))
  710. end
  711. },
  712. {"string error reports are printed",
  713. fun() ->
  714. sync_error_logger:error_report("this is less silly"),
  715. _ = gen_event:which_handlers(error_logger),
  716. {Level, _, Msg,Metadata} = pop(),
  717. ?assertEqual(lager_util:level_to_num(error),Level),
  718. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  719. Expected = "this is less silly",
  720. ?assertEqual(Expected, lists:flatten(Msg))
  721. end
  722. },
  723. {"error messages are printed",
  724. fun() ->
  725. sync_error_logger:error_msg("doom, doom has come upon you all"),
  726. _ = gen_event:which_handlers(error_logger),
  727. {Level, _, Msg,Metadata} = pop(),
  728. ?assertEqual(lager_util:level_to_num(error),Level),
  729. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  730. Expected = "doom, doom has come upon you all",
  731. ?assertEqual(Expected, lists:flatten(Msg))
  732. end
  733. },
  734. {"error messages with unicode characters in Args are printed",
  735. fun() ->
  736. sync_error_logger:error_msg("~ts", ["Привет!"]),
  737. _ = gen_event:which_handlers(error_logger),
  738. {Level, _, Msg,Metadata} = pop(),
  739. ?assertEqual(lager_util:level_to_num(error),Level),
  740. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  741. ?assertEqual("Привет!", lists:flatten(Msg))
  742. end
  743. },
  744. {"error messages are truncated at 4096 characters",
  745. fun() ->
  746. sync_error_logger:error_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  747. _ = gen_event:which_handlers(error_logger),
  748. {_, _, Msg,_Metadata} = pop(),
  749. ?assert(length(lists:flatten(Msg)) < 5100)
  750. end
  751. },
  752. {"info reports are printed",
  753. fun() ->
  754. sync_error_logger:info_report([{this, is}, a, {silly, format}]),
  755. _ = gen_event:which_handlers(error_logger),
  756. {Level, _, Msg,Metadata} = pop(),
  757. ?assertEqual(lager_util:level_to_num(info),Level),
  758. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  759. Expected = "this: is, a, silly: format",
  760. ?assertEqual(Expected, lists:flatten(Msg))
  761. end
  762. },
  763. {"info reports are truncated at 4096 characters",
  764. fun() ->
  765. sync_error_logger:info_report([[{this, is}, a, {silly, format}] || _ <- lists:seq(0, 600)]),
  766. _ = gen_event:which_handlers(error_logger),
  767. {Level, _, Msg,Metadata} = pop(),
  768. ?assertEqual(lager_util:level_to_num(info),Level),
  769. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  770. ?assert(length(lists:flatten(Msg)) < 5000)
  771. end
  772. },
  773. {"single term info reports are printed",
  774. fun() ->
  775. sync_error_logger:info_report({foolish, bees}),
  776. _ = gen_event:which_handlers(error_logger),
  777. {Level, _, Msg,Metadata} = pop(),
  778. ?assertEqual(lager_util:level_to_num(info),Level),
  779. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  780. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  781. end
  782. },
  783. {"single term error reports are printed",
  784. fun() ->
  785. sync_error_logger:error_report({foolish, bees}),
  786. _ = gen_event:which_handlers(error_logger),
  787. {Level, _, Msg,Metadata} = pop(),
  788. ?assertEqual(lager_util:level_to_num(error),Level),
  789. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  790. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  791. end
  792. },
  793. {"string info reports are printed",
  794. fun() ->
  795. sync_error_logger:info_report("this is less silly"),
  796. _ = gen_event:which_handlers(error_logger),
  797. {Level, _, Msg,Metadata} = pop(),
  798. ?assertEqual(lager_util:level_to_num(info),Level),
  799. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  800. ?assertEqual("this is less silly", lists:flatten(Msg))
  801. end
  802. },
  803. {"string info reports are truncated at 4096 characters",
  804. fun() ->
  805. sync_error_logger:info_report(string:copies("this is less silly", 1000)),
  806. _ = gen_event:which_handlers(error_logger),
  807. {Level, _, Msg,Metadata} = pop(),
  808. ?assertEqual(lager_util:level_to_num(info),Level),
  809. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  810. ?assert(length(lists:flatten(Msg)) < 5100)
  811. end
  812. },
  813. {"strings in a mixed report are printed as strings",
  814. fun() ->
  815. sync_error_logger:info_report(["this is less silly", {than, "this"}]),
  816. _ = gen_event:which_handlers(error_logger),
  817. {Level, _, Msg,Metadata} = pop(),
  818. ?assertEqual(lager_util:level_to_num(info),Level),
  819. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  820. ?assertEqual("\"this is less silly\", than: \"this\"", lists:flatten(Msg))
  821. end
  822. },
  823. {"info messages are printed",
  824. fun() ->
  825. sync_error_logger:info_msg("doom, doom has come upon you all"),
  826. _ = gen_event:which_handlers(error_logger),
  827. {Level, _, Msg,Metadata} = pop(),
  828. ?assertEqual(lager_util:level_to_num(info),Level),
  829. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  830. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  831. end
  832. },
  833. {"info messages are truncated at 4096 characters",
  834. fun() ->
  835. sync_error_logger:info_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  836. _ = gen_event:which_handlers(error_logger),
  837. {Level, _, Msg,Metadata} = pop(),
  838. ?assertEqual(lager_util:level_to_num(info),Level),
  839. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  840. ?assert(length(lists:flatten(Msg)) < 5100)
  841. end
  842. },
  843. {"info messages with unicode characters in Args are printed",
  844. fun() ->
  845. sync_error_logger:info_msg("~ts", ["Привет!"]),
  846. _ = gen_event:which_handlers(error_logger),
  847. {Level, _, Msg,Metadata} = pop(),
  848. ?assertEqual(lager_util:level_to_num(info),Level),
  849. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  850. ?assertEqual("Привет!", lists:flatten(Msg))
  851. end
  852. },
  853. {"warning messages with unicode characters in Args are printed",
  854. fun() ->
  855. sync_error_logger:warning_msg("~ts", ["Привет!"]),
  856. Map = error_logger:warning_map(),
  857. _ = gen_event:which_handlers(error_logger),
  858. {Level, _, Msg,Metadata} = pop(),
  859. ?assertEqual(lager_util:level_to_num(Map),Level),
  860. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  861. ?assertEqual("Привет!", lists:flatten(Msg))
  862. end
  863. },
  864. {"warning messages are printed at the correct level",
  865. fun() ->
  866. sync_error_logger:warning_msg("doom, doom has come upon you all"),
  867. Map = error_logger:warning_map(),
  868. _ = gen_event:which_handlers(error_logger),
  869. {Level, _, Msg,Metadata} = pop(),
  870. ?assertEqual(lager_util:level_to_num(Map),Level),
  871. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  872. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  873. end
  874. },
  875. {"warning reports are printed at the correct level",
  876. fun() ->
  877. sync_error_logger:warning_report([{i, like}, pie]),
  878. Map = error_logger:warning_map(),
  879. _ = gen_event:which_handlers(error_logger),
  880. {Level, _, Msg,Metadata} = pop(),
  881. ?assertEqual(lager_util:level_to_num(Map),Level),
  882. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  883. ?assertEqual("i: like, pie", lists:flatten(Msg))
  884. end
  885. },
  886. {"single term warning reports are printed at the correct level",
  887. fun() ->
  888. sync_error_logger:warning_report({foolish, bees}),
  889. Map = error_logger:warning_map(),
  890. _ = gen_event:which_handlers(error_logger),
  891. {Level, _, Msg,Metadata} = pop(),
  892. ?assertEqual(lager_util:level_to_num(Map),Level),
  893. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  894. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  895. end
  896. },
  897. {"application stop reports",
  898. fun() ->
  899. sync_error_logger:info_report([{application, foo}, {exited, quittin_time}, {type, lazy}]),
  900. _ = gen_event:which_handlers(error_logger),
  901. {Level, _, Msg,Metadata} = pop(),
  902. ?assertEqual(lager_util:level_to_num(info),Level),
  903. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  904. ?assertEqual("Application foo exited with reason: quittin_time", lists:flatten(Msg))
  905. end
  906. },
  907. {"supervisor reports",
  908. fun() ->
  909. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  910. _ = gen_event:which_handlers(error_logger),
  911. {Level, _, Msg,Metadata} = pop(),
  912. ?assertEqual(lager_util:level_to_num(error),Level),
  913. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  914. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason fired in context france", lists:flatten(Msg))
  915. end
  916. },
  917. {"supervisor reports with real error",
  918. fun() ->
  919. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, {function_clause,[{crash,handle_info,[foo]}]}}, {supervisor, {local, steve}}]),
  920. _ = gen_event:which_handlers(error_logger),
  921. {Level, _, Msg,Metadata} = pop(),
  922. ?assertEqual(lager_util:level_to_num(error),Level),
  923. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  924. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason no function clause matching crash:handle_info(foo) in context france", lists:flatten(Msg))
  925. end
  926. },
  927. {"supervisor reports with real error and pid",
  928. fun() ->
  929. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, {function_clause,[{crash,handle_info,[foo]}]}}, {supervisor, somepid}]),
  930. _ = gen_event:which_handlers(error_logger),
  931. {Level, _, Msg,Metadata} = pop(),
  932. ?assertEqual(lager_util:level_to_num(error),Level),
  933. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  934. ?assertEqual("Supervisor somepid had child mini_steve started with a:b(c) at bleh exit with reason no function clause matching crash:handle_info(foo) in context france", lists:flatten(Msg))
  935. end
  936. },
  937. {"supervisor_bridge reports",
  938. fun() ->
  939. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{mod, mini_steve}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  940. _ = gen_event:which_handlers(error_logger),
  941. {Level, _, Msg,Metadata} = pop(),
  942. ?assertEqual(lager_util:level_to_num(error),Level),
  943. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  944. ?assertEqual("Supervisor steve had child at module mini_steve at bleh exit with reason fired in context france", lists:flatten(Msg))
  945. end
  946. },
  947. {"application progress report",
  948. fun() ->
  949. sync_error_logger:info_report(progress, [{application, foo}, {started_at, node()}]),
  950. _ = gen_event:which_handlers(error_logger),
  951. {Level, _, Msg,Metadata} = pop(),
  952. ?assertEqual(lager_util:level_to_num(info),Level),
  953. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  954. Expected = lists:flatten(io_lib:format("Application foo started on node ~w", [node()])),
  955. ?assertEqual(Expected, lists:flatten(Msg))
  956. end
  957. },
  958. {"supervisor progress report",
  959. fun() ->
  960. lager:set_loglevel(?MODULE, debug),
  961. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  962. sync_error_logger:info_report(progress, [{supervisor, {local, foo}}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  963. _ = gen_event:which_handlers(error_logger),
  964. {Level, _, Msg,Metadata} = pop(),
  965. ?assertEqual(lager_util:level_to_num(debug),Level),
  966. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  967. ?assertEqual("Supervisor foo started foo:bar/1 at pid baz", lists:flatten(Msg))
  968. end
  969. },
  970. {"supervisor progress report with pid",
  971. fun() ->
  972. lager:set_loglevel(?MODULE, debug),
  973. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  974. sync_error_logger:info_report(progress, [{supervisor, somepid}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  975. _ = gen_event:which_handlers(error_logger),
  976. {Level, _, Msg,Metadata} = pop(),
  977. ?assertEqual(lager_util:level_to_num(debug),Level),
  978. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  979. ?assertEqual("Supervisor somepid started foo:bar/1 at pid baz", lists:flatten(Msg))
  980. end
  981. },
  982. {"crash report for emfile",
  983. fun() ->
  984. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, emfile, [{stack, trace, 1}]}}], []]),
  985. _ = gen_event:which_handlers(error_logger),
  986. {Level, _, Msg,Metadata} = pop(),
  987. ?assertEqual(lager_util:level_to_num(error),Level),
  988. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  989. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: maximum number of file descriptors exhausted, check ulimit -n", [self()])),
  990. ?assertEqual(Expected, lists:flatten(Msg))
  991. end
  992. },
  993. {"crash report for system process limit",
  994. fun() ->
  995. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn, 1}]}}], []]),
  996. _ = gen_event:which_handlers(error_logger),
  997. {Level, _, Msg,Metadata} = pop(),
  998. ?assertEqual(lager_util:level_to_num(error),Level),
  999. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1000. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1001. ?assertEqual(Expected, lists:flatten(Msg))
  1002. end
  1003. },
  1004. {"crash report for system process limit2",
  1005. fun() ->
  1006. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn_opt, 1}]}}], []]),
  1007. _ = gen_event:which_handlers(error_logger),
  1008. {Level, _, Msg,Metadata} = pop(),
  1009. ?assertEqual(lager_util:level_to_num(error),Level),
  1010. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1011. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1012. ?assertEqual(Expected, lists:flatten(Msg))
  1013. end
  1014. },
  1015. {"crash report for system port limit",
  1016. fun() ->
  1017. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, open_port, 1}]}}], []]),
  1018. _ = gen_event:which_handlers(error_logger),
  1019. {Level, _, Msg,Metadata} = pop(),
  1020. ?assertEqual(lager_util:level_to_num(error),Level),
  1021. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1022. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ports exceeded", [self()])),
  1023. ?assertEqual(Expected, lists:flatten(Msg))
  1024. end
  1025. },
  1026. {"crash report for system port limit",
  1027. fun() ->
  1028. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, list_to_atom, 1}]}}], []]),
  1029. _ = gen_event:which_handlers(error_logger),
  1030. {Level, _, Msg,Metadata} = pop(),
  1031. ?assertEqual(lager_util:level_to_num(error),Level),
  1032. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1033. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: tried to create an atom larger than 255, or maximum atom count exceeded", [self()])),
  1034. ?assertEqual(Expected, lists:flatten(Msg))
  1035. end
  1036. },
  1037. {"crash report for system ets table limit",
  1038. fun() ->
  1039. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, test}, {error_info, {error, system_limit, [{ets,new,[segment_offsets,[ordered_set,public]]},{mi_segment,open_write,1},{mi_buffer_converter,handle_cast,2},{gen_server,handle_msg,5},{proc_lib,init_p_do_apply,3}]}}], []]),
  1040. _ = gen_event:which_handlers(error_logger),
  1041. {Level, _, Msg,Metadata} = pop(),
  1042. ?assertEqual(lager_util:level_to_num(error),Level),
  1043. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1044. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ETS tables exceeded", [test])),
  1045. ?assertEqual(Expected, lists:flatten(Msg))
  1046. end
  1047. },
  1048. {"crash report for unknown system limit should be truncated at 500 characters",
  1049. fun() ->
  1050. sync_error_logger:error_report(crash_report, [[{pid, self()}, {error_info, {error, system_limit, [{wtf,boom,[string:copies("aaaa", 4096)]}]}}], []]),
  1051. _ = gen_event:which_handlers(error_logger),
  1052. {_, _, Msg,_Metadata} = pop(),
  1053. ?assert(length(lists:flatten(Msg)) > 550),
  1054. ?assert(length(lists:flatten(Msg)) < 600)
  1055. end
  1056. },
  1057. {"crash reports for 'special processes' should be handled right - function_clause",
  1058. fun() ->
  1059. {ok, Pid} = special_process:start(),
  1060. unlink(Pid),
  1061. Pid ! function_clause,
  1062. timer:sleep(500),
  1063. _ = gen_event:which_handlers(error_logger),
  1064. {_, _, Msg, _Metadata} = pop(),
  1065. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no function clause matching special_process:foo(bar)",
  1066. [Pid])),
  1067. test_body(Expected, lists:flatten(Msg))
  1068. end
  1069. },
  1070. {"crash reports for 'special processes' should be handled right - case_clause",
  1071. fun() ->
  1072. {ok, Pid} = special_process:start(),
  1073. unlink(Pid),
  1074. Pid ! {case_clause, wtf},
  1075. timer:sleep(500),
  1076. _ = gen_event:which_handlers(error_logger),
  1077. {_, _, Msg, _Metadata} = pop(),
  1078. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no case clause matching wtf in special_process:loop/0",
  1079. [Pid])),
  1080. test_body(Expected, lists:flatten(Msg))
  1081. end
  1082. },
  1083. {"crash reports for 'special processes' should be handled right - exit",
  1084. fun() ->
  1085. {ok, Pid} = special_process:start(),
  1086. unlink(Pid),
  1087. Pid ! exit,
  1088. timer:sleep(500),
  1089. _ = gen_event:which_handlers(error_logger),
  1090. {_, _, Msg, _Metadata} = pop(),
  1091. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours exited with reason: byebye in special_process:loop/0",
  1092. [Pid])),
  1093. test_body(Expected, lists:flatten(Msg))
  1094. end
  1095. },
  1096. {"crash reports for 'special processes' should be handled right - error",
  1097. fun() ->
  1098. {ok, Pid} = special_process:start(),
  1099. unlink(Pid),
  1100. Pid ! error,
  1101. timer:sleep(500),
  1102. _ = gen_event:which_handlers(error_logger),
  1103. {_, _, Msg, _Metadata} = pop(),
  1104. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: mybad in special_process:loop/0",
  1105. [Pid])),
  1106. test_body(Expected, lists:flatten(Msg))
  1107. end
  1108. },
  1109. {"webmachine error reports",
  1110. fun() ->
  1111. Path = "/cgi-bin/phpmyadmin",
  1112. Reason = {error,{error,{badmatch,{error,timeout}},
  1113. [{myapp,dostuff,2,[{file,"src/myapp.erl"},{line,123}]},
  1114. {webmachine_resource,resource_call,3,[{file,"src/webmachine_resource.erl"},{line,169}]}]}},
  1115. sync_error_logger:error_msg("webmachine error: path=~p~n~p~n", [Path, Reason]),
  1116. _ = gen_event:which_handlers(error_logger),
  1117. {Level, _, Msg,Metadata} = pop(),
  1118. ?assertEqual(lager_util:level_to_num(error),Level),
  1119. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1120. ?assertEqual("Webmachine error at path \"/cgi-bin/phpmyadmin\" : no match of right hand value {error,timeout} in myapp:dostuff/2 line 123", lists:flatten(Msg))
  1121. end
  1122. },
  1123. {"Cowboy error reports, 8 arg version",
  1124. fun() ->
  1125. Stack = [{my_handler,init, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1126. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1127. sync_error_logger:error_msg(
  1128. "** Cowboy handler ~p terminating in ~p/~p~n"
  1129. " for the reason ~p:~p~n"
  1130. "** Options were ~p~n"
  1131. "** Request was ~p~n"
  1132. "** Stacktrace: ~p~n~n",
  1133. [my_handler, init, 3, error, {badmatch, {error, timeout}}, [],
  1134. "Request", Stack]),
  1135. _ = gen_event:which_handlers(error_logger),
  1136. {Level, _, Msg,Metadata} = pop(),
  1137. ?assertEqual(lager_util:level_to_num(error),Level),
  1138. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1139. ?assertEqual("Cowboy handler my_handler terminated in my_handler:init/3 with reason: no match of right hand value {error,timeout} in my_handler:init/3 line 123", lists:flatten(Msg))
  1140. end
  1141. },
  1142. {"Cowboy error reports, 10 arg version",
  1143. fun() ->
  1144. Stack = [{my_handler,somecallback, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1145. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1146. sync_error_logger:error_msg(
  1147. "** Cowboy handler ~p terminating in ~p/~p~n"
  1148. " for the reason ~p:~p~n** Message was ~p~n"
  1149. "** Options were ~p~n** Handler state was ~p~n"
  1150. "** Request was ~p~n** Stacktrace: ~p~n~n",
  1151. [my_handler, somecallback, 3, error, {badmatch, {error, timeout}}, hello, [],
  1152. {}, "Request", Stack]),
  1153. _ = gen_event:which_handlers(error_logger),
  1154. {Level, _, Msg,Metadata} = pop(),
  1155. ?assertEqual(lager_util:level_to_num(error),Level),
  1156. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1157. ?assertEqual("Cowboy handler my_handler terminated in my_handler:somecallback/3 with reason: no match of right hand value {error,timeout} in my_handler:somecallback/3 line 123", lists:flatten(Msg))
  1158. end
  1159. },
  1160. {"Cowboy error reports, 5 arg version",
  1161. fun() ->
  1162. sync_error_logger:error_msg(
  1163. "** Cowboy handler ~p terminating; "
  1164. "function ~p/~p was not exported~n"
  1165. "** Request was ~p~n** State was ~p~n~n",
  1166. [my_handler, to_json, 2, "Request", {}]),
  1167. _ = gen_event:which_handlers(error_logger),
  1168. {Level, _, Msg,Metadata} = pop(),
  1169. ?assertEqual(lager_util:level_to_num(error),Level),
  1170. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1171. ?assertEqual("Cowboy handler my_handler terminated with reason: call to undefined function my_handler:to_json/2", lists:flatten(Msg))
  1172. end
  1173. },
  1174. {"messages should not be generated if they don't satisfy the threshold",
  1175. fun() ->
  1176. lager:set_loglevel(?MODULE, error),
  1177. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1178. sync_error_logger:info_report([hello, world]),
  1179. _ = gen_event:which_handlers(error_logger),
  1180. ?assertEqual(0, count()),
  1181. ?assertEqual(0, count_ignored()),
  1182. lager:set_loglevel(?MODULE, info),
  1183. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1184. sync_error_logger:info_report([hello, world]),
  1185. _ = gen_event:which_handlers(error_logger),
  1186. ?assertEqual(1, count()),
  1187. ?assertEqual(0, count_ignored()),
  1188. lager:set_loglevel(?MODULE, error),
  1189. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1190. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(debug)), []}),
  1191. sync_error_logger:info_report([hello, world]),
  1192. _ = gen_event:which_handlers(error_logger),
  1193. ?assertEqual(1, count()),
  1194. ?assertEqual(1, count_ignored())
  1195. end
  1196. }
  1197. ]
  1198. }.
  1199. safe_format_test() ->
  1200. ?assertEqual("foo bar", lists:flatten(lager:safe_format("~p ~p", [foo, bar], 1024))),
  1201. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:safe_format("~p ~p ~p", [foo, bar], 1024))),
  1202. ok.
  1203. async_threshold_test_() ->
  1204. {foreach,
  1205. fun() ->
  1206. error_logger:tty(false),
  1207. ets:new(async_threshold_test, [set, named_table, public]),
  1208. ets:insert_new(async_threshold_test, {sync_toggled, 0}),
  1209. ets:insert_new(async_threshold_test, {async_toggled, 0}),
  1210. application:load(lager),
  1211. application:set_env(lager, error_logger_redirect, false),
  1212. application:set_env(lager, async_threshold, 2),
  1213. application:set_env(lager, async_threshold_window, 1),
  1214. application:set_env(lager, handlers, [{?MODULE, info}]),
  1215. lager:start()
  1216. end,
  1217. fun(_) ->
  1218. application:unset_env(lager, async_threshold),
  1219. application:stop(lager),
  1220. application:stop(goldrush),
  1221. ets:delete(async_threshold_test),
  1222. error_logger:tty(true)
  1223. end,
  1224. [
  1225. {"async threshold works",
  1226. fun() ->
  1227. %% we start out async
  1228. ?assertEqual(true, lager_config:get(async)),
  1229. %% put a ton of things in the queue
  1230. Workers = [spawn_monitor(fun() -> [lager:info("hello world") || _ <- lists:seq(1, 1000)] end) || _ <- lists:seq(1, 15)],
  1231. %% serialize on mailbox
  1232. _ = gen_event:which_handlers(lager_event),
  1233. timer:sleep(500),
  1234. %% By now the flood of messages will have
  1235. %% forced the backend throttle to turn off
  1236. %% async mode, but it's possible all
  1237. %% outstanding requests have been processed,
  1238. %% so checking the current status (sync or
  1239. %% async) is an exercise in race control.
  1240. %% Instead, we'll see whether the backend
  1241. %% throttle has toggled into sync mode at any
  1242. %% point in the past
  1243. ?assertMatch([{sync_toggled, N}] when N > 0,
  1244. ets:lookup(async_threshold_test, sync_toggled)),
  1245. %% wait for all the workers to return, meaning that all the messages have been logged (since we're definitely in sync mode at the end of the run)
  1246. collect_workers(Workers),
  1247. %% serialize on the mailbox again
  1248. _ = gen_event:which_handlers(lager_event),
  1249. %% just in case...
  1250. timer:sleep(1000),
  1251. lager:info("hello world"),
  1252. _ = gen_event:which_handlers(lager_event),
  1253. %% async is true again now that the mailbox has drained
  1254. ?assertEqual(true, lager_config:get(async)),
  1255. ok
  1256. end
  1257. }
  1258. ]
  1259. }.
  1260. collect_workers([]) ->
  1261. ok;
  1262. collect_workers(Workers) ->
  1263. receive
  1264. {'DOWN', Ref, _, _, _} ->
  1265. collect_workers(lists:keydelete(Ref, 2, Workers))
  1266. end.
  1267. produce_n_error_logger_msgs(N) ->
  1268. lists:foreach(fun (K) ->
  1269. error_logger:error_msg("Foo ~p!", [K])
  1270. end,
  1271. lists:seq(0, N-1)
  1272. ).
  1273. high_watermark_test_() ->
  1274. {foreach,
  1275. fun() ->
  1276. error_logger:tty(false),
  1277. application:load(lager),
  1278. application:set_env(lager, error_logger_redirect, true),
  1279. application:set_env(lager, handlers, [{lager_test_backend, info}]),
  1280. application:set_env(lager, async_threshold, undefined),
  1281. lager:start()
  1282. end,
  1283. fun(_) ->
  1284. application:stop(lager),
  1285. error_logger:tty(true)
  1286. end,
  1287. [
  1288. {"Nothing dropped when error_logger high watermark is undefined",
  1289. fun () ->
  1290. ok = error_logger_lager_h:set_high_water(undefined),
  1291. timer:sleep(100),
  1292. produce_n_error_logger_msgs(10),
  1293. timer:sleep(500),
  1294. ?assert(count() >= 10)
  1295. end
  1296. },
  1297. {"Mostly dropped according to error_logger high watermark",
  1298. fun () ->
  1299. ok = error_logger_lager_h:set_high_water(5),
  1300. timer:sleep(100),
  1301. produce_n_error_logger_msgs(50),
  1302. timer:sleep(1000),
  1303. ?assert(count() < 20)
  1304. end
  1305. },
  1306. {"Non-notifications are not dropped",
  1307. fun () ->
  1308. ok = error_logger_lager_h:set_high_water(2),
  1309. timer:sleep(100),
  1310. spawn(fun () -> produce_n_error_logger_msgs(300) end),
  1311. timer:sleep(50),
  1312. %% if everything were dropped, this call would be dropped
  1313. %% too, so lets hope it's not
  1314. ?assert(is_integer(count())),
  1315. timer:sleep(1000),
  1316. ?assert(count() < 10)
  1317. end
  1318. }
  1319. ]
  1320. }.
  1321. -endif.