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

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