Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.

1511 řádky
79 KiB

před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
před 14 roky
  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_test_() ->
  747. TestBody=fun(Name,CrashReason,Expected) -> {Name,
  748. fun() ->
  749. Pid = whereis(crash),
  750. crash(CrashReason),
  751. {Level, _, Msg,Metadata} = pop(),
  752. test_body(Expected, lists:flatten(Msg)),
  753. ?assertEqual(Pid,proplists:get_value(pid,Metadata)),
  754. ?assertEqual(lager_util:level_to_num(error),Level)
  755. end
  756. }
  757. end,
  758. {foreach,
  759. fun() ->
  760. error_logger:tty(false),
  761. application:load(lager),
  762. application:set_env(lager, error_logger_redirect, true),
  763. application:set_env(lager, handlers, [{?MODULE, error}]),
  764. lager:start(),
  765. crash:start()
  766. end,
  767. fun(_) ->
  768. application:stop(lager),
  769. application:stop(goldrush),
  770. case whereis(crash) of
  771. undefined -> ok;
  772. Pid -> exit(Pid, kill)
  773. end,
  774. error_logger:tty(true)
  775. end,
  776. [
  777. {"again, there is nothing up my sleeve",
  778. fun() ->
  779. ?assertEqual(undefined, pop()),
  780. ?assertEqual(0, count())
  781. end
  782. },
  783. TestBody("bad return value",bad_return,"gen_server crash terminated with reason: bad return value: bleh"),
  784. TestBody("bad return value with string",bad_return_string,"gen_server crash terminated with reason: bad return value: {tuple,{tuple,\"string\"}}"),
  785. TestBody("bad return uncaught throw",throw,"gen_server crash terminated with reason: bad return value: a_ball"),
  786. TestBody("case clause",case_clause,"gen_server crash terminated with reason: no case clause matching {} in crash:handle_call/3"),
  787. TestBody("case clause string",case_clause_string,"gen_server crash terminated with reason: no case clause matching \"crash\" in crash:handle_call/3"),
  788. TestBody("function clause",function_clause,"gen_server crash terminated with reason: no function clause matching crash:function({})"),
  789. TestBody("if clause",if_clause,"gen_server crash terminated with reason: no true branch found while evaluating if expression in crash:handle_call/3"),
  790. TestBody("try clause",try_clause,"gen_server crash terminated with reason: no try clause matching [] in crash:handle_call/3"),
  791. TestBody("undefined function",undef,"gen_server crash terminated with reason: call to undefined function crash:booger/0 from crash:handle_call/3"),
  792. TestBody("bad math",badarith,"gen_server crash terminated with reason: bad arithmetic expression in crash:handle_call/3"),
  793. TestBody("bad match",badmatch,"gen_server crash terminated with reason: no match of right hand value {} in crash:handle_call/3"),
  794. 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"),
  795. TestBody("bad arg1",badarg1,"gen_server crash terminated with reason: bad argument in crash:handle_call/3"),
  796. 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"),
  797. TestBody("bad record",badrecord,"gen_server crash terminated with reason: bad record state in crash:handle_call/3"),
  798. TestBody("noproc",noproc,"gen_server crash terminated with reason: no such process or port in call to gen_event:call(foo, bar, baz)"),
  799. TestBody("badfun",badfun,"gen_server crash terminated with reason: bad function booger in crash:handle_call/3")
  800. ]
  801. }.
  802. error_logger_redirect_test_() ->
  803. {foreach,
  804. fun() ->
  805. error_logger:tty(false),
  806. application:load(lager),
  807. application:set_env(lager, error_logger_redirect, true),
  808. application:set_env(lager, handlers, [{?MODULE, info}]),
  809. lager:start(),
  810. lager:log(error, self(), "flush flush"),
  811. timer:sleep(100),
  812. gen_event:call(lager_event, ?MODULE, flush)
  813. end,
  814. fun(_) ->
  815. application:stop(lager),
  816. application:stop(goldrush),
  817. error_logger:tty(true)
  818. end,
  819. [
  820. {"error reports are printed",
  821. fun() ->
  822. sync_error_logger:error_report([{this, is}, a, {silly, format}]),
  823. _ = gen_event:which_handlers(error_logger),
  824. {Level, _, Msg,Metadata} = pop(),
  825. ?assertEqual(lager_util:level_to_num(error),Level),
  826. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  827. Expected = "this: is, a, silly: format",
  828. ?assertEqual(Expected, lists:flatten(Msg))
  829. end
  830. },
  831. {"string error reports are printed",
  832. fun() ->
  833. sync_error_logger:error_report("this is less silly"),
  834. _ = gen_event:which_handlers(error_logger),
  835. {Level, _, Msg,Metadata} = pop(),
  836. ?assertEqual(lager_util:level_to_num(error),Level),
  837. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  838. Expected = "this is less silly",
  839. ?assertEqual(Expected, lists:flatten(Msg))
  840. end
  841. },
  842. {"error messages are printed",
  843. fun() ->
  844. sync_error_logger:error_msg("doom, doom has come upon you all"),
  845. _ = gen_event:which_handlers(error_logger),
  846. {Level, _, Msg,Metadata} = pop(),
  847. ?assertEqual(lager_util:level_to_num(error),Level),
  848. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  849. Expected = "doom, doom has come upon you all",
  850. ?assertEqual(Expected, lists:flatten(Msg))
  851. end
  852. },
  853. {"error messages with unicode characters in Args are printed",
  854. fun() ->
  855. sync_error_logger:error_msg("~ts", ["Привет!"]),
  856. _ = gen_event:which_handlers(error_logger),
  857. {Level, _, Msg,Metadata} = pop(),
  858. ?assertEqual(lager_util:level_to_num(error),Level),
  859. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  860. ?assertEqual("Привет!", lists:flatten(Msg))
  861. end
  862. },
  863. {"error messages are truncated at 4096 characters",
  864. fun() ->
  865. sync_error_logger:error_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  866. _ = gen_event:which_handlers(error_logger),
  867. {_, _, Msg,_Metadata} = pop(),
  868. ?assert(length(lists:flatten(Msg)) < 5100)
  869. end
  870. },
  871. {"info reports are printed",
  872. fun() ->
  873. sync_error_logger:info_report([{this, is}, a, {silly, format}]),
  874. _ = gen_event:which_handlers(error_logger),
  875. {Level, _, Msg,Metadata} = pop(),
  876. ?assertEqual(lager_util:level_to_num(info),Level),
  877. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  878. Expected = "this: is, a, silly: format",
  879. ?assertEqual(Expected, lists:flatten(Msg))
  880. end
  881. },
  882. {"info reports are truncated at 4096 characters",
  883. fun() ->
  884. sync_error_logger:info_report([[{this, is}, a, {silly, format}] || _ <- lists:seq(0, 600)]),
  885. _ = gen_event:which_handlers(error_logger),
  886. {Level, _, Msg,Metadata} = pop(),
  887. ?assertEqual(lager_util:level_to_num(info),Level),
  888. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  889. ?assert(length(lists:flatten(Msg)) < 5000)
  890. end
  891. },
  892. {"single term info reports are printed",
  893. fun() ->
  894. sync_error_logger:info_report({foolish, bees}),
  895. _ = gen_event:which_handlers(error_logger),
  896. {Level, _, Msg,Metadata} = pop(),
  897. ?assertEqual(lager_util:level_to_num(info),Level),
  898. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  899. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  900. end
  901. },
  902. {"single term error reports are printed",
  903. fun() ->
  904. sync_error_logger:error_report({foolish, bees}),
  905. _ = gen_event:which_handlers(error_logger),
  906. {Level, _, Msg,Metadata} = pop(),
  907. ?assertEqual(lager_util:level_to_num(error),Level),
  908. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  909. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  910. end
  911. },
  912. {"string info reports are printed",
  913. fun() ->
  914. sync_error_logger:info_report("this is less silly"),
  915. _ = gen_event:which_handlers(error_logger),
  916. {Level, _, Msg,Metadata} = pop(),
  917. ?assertEqual(lager_util:level_to_num(info),Level),
  918. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  919. ?assertEqual("this is less silly", lists:flatten(Msg))
  920. end
  921. },
  922. {"string info reports are truncated at 4096 characters",
  923. fun() ->
  924. sync_error_logger:info_report(string:copies("this is less silly", 1000)),
  925. _ = gen_event:which_handlers(error_logger),
  926. {Level, _, Msg,Metadata} = pop(),
  927. ?assertEqual(lager_util:level_to_num(info),Level),
  928. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  929. ?assert(length(lists:flatten(Msg)) < 5100)
  930. end
  931. },
  932. {"strings in a mixed report are printed as strings",
  933. fun() ->
  934. sync_error_logger:info_report(["this is less silly", {than, "this"}]),
  935. _ = gen_event:which_handlers(error_logger),
  936. {Level, _, Msg,Metadata} = pop(),
  937. ?assertEqual(lager_util:level_to_num(info),Level),
  938. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  939. ?assertEqual("\"this is less silly\", than: \"this\"", lists:flatten(Msg))
  940. end
  941. },
  942. {"info messages are printed",
  943. fun() ->
  944. sync_error_logger:info_msg("doom, doom has come upon you all"),
  945. _ = gen_event:which_handlers(error_logger),
  946. {Level, _, Msg,Metadata} = pop(),
  947. ?assertEqual(lager_util:level_to_num(info),Level),
  948. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  949. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  950. end
  951. },
  952. {"info messages are truncated at 4096 characters",
  953. fun() ->
  954. sync_error_logger:info_msg("doom, doom has come upon you all ~p", [string:copies("doom", 10000)]),
  955. _ = gen_event:which_handlers(error_logger),
  956. {Level, _, Msg,Metadata} = pop(),
  957. ?assertEqual(lager_util:level_to_num(info),Level),
  958. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  959. ?assert(length(lists:flatten(Msg)) < 5100)
  960. end
  961. },
  962. {"info messages with unicode characters in Args are printed",
  963. fun() ->
  964. sync_error_logger:info_msg("~ts", ["Привет!"]),
  965. _ = gen_event:which_handlers(error_logger),
  966. {Level, _, Msg,Metadata} = pop(),
  967. ?assertEqual(lager_util:level_to_num(info),Level),
  968. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  969. ?assertEqual("Привет!", lists:flatten(Msg))
  970. end
  971. },
  972. {"warning messages with unicode characters in Args are printed",
  973. fun() ->
  974. sync_error_logger:warning_msg("~ts", ["Привет!"]),
  975. Map = error_logger:warning_map(),
  976. _ = gen_event:which_handlers(error_logger),
  977. {Level, _, Msg,Metadata} = pop(),
  978. ?assertEqual(lager_util:level_to_num(Map),Level),
  979. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  980. ?assertEqual("Привет!", lists:flatten(Msg))
  981. end
  982. },
  983. {"warning messages are printed at the correct level",
  984. fun() ->
  985. sync_error_logger:warning_msg("doom, doom has come upon you all"),
  986. Map = error_logger:warning_map(),
  987. _ = gen_event:which_handlers(error_logger),
  988. {Level, _, Msg,Metadata} = pop(),
  989. ?assertEqual(lager_util:level_to_num(Map),Level),
  990. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  991. ?assertEqual("doom, doom has come upon you all", lists:flatten(Msg))
  992. end
  993. },
  994. {"warning reports are printed at the correct level",
  995. fun() ->
  996. sync_error_logger:warning_report([{i, like}, pie]),
  997. Map = error_logger:warning_map(),
  998. _ = gen_event:which_handlers(error_logger),
  999. {Level, _, Msg,Metadata} = pop(),
  1000. ?assertEqual(lager_util:level_to_num(Map),Level),
  1001. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1002. ?assertEqual("i: like, pie", lists:flatten(Msg))
  1003. end
  1004. },
  1005. {"single term warning reports are printed at the correct level",
  1006. fun() ->
  1007. sync_error_logger:warning_report({foolish, bees}),
  1008. Map = error_logger:warning_map(),
  1009. _ = gen_event:which_handlers(error_logger),
  1010. {Level, _, Msg,Metadata} = pop(),
  1011. ?assertEqual(lager_util:level_to_num(Map),Level),
  1012. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1013. ?assertEqual("{foolish,bees}", lists:flatten(Msg))
  1014. end
  1015. },
  1016. {"application stop reports",
  1017. fun() ->
  1018. sync_error_logger:info_report([{application, foo}, {exited, quittin_time}, {type, lazy}]),
  1019. _ = gen_event:which_handlers(error_logger),
  1020. {Level, _, Msg,Metadata} = pop(),
  1021. ?assertEqual(lager_util:level_to_num(info),Level),
  1022. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1023. ?assertEqual("Application foo exited with reason: quittin_time", lists:flatten(Msg))
  1024. end
  1025. },
  1026. {"supervisor reports",
  1027. fun() ->
  1028. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{name, mini_steve}, {mfargs, {a, b, [c]}}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  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. ?assertEqual("Supervisor steve had child mini_steve started with a:b(c) at bleh exit with reason fired in context france", lists:flatten(Msg))
  1034. end
  1035. },
  1036. {"supervisor reports with real error",
  1037. fun() ->
  1038. 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}}]),
  1039. _ = gen_event:which_handlers(error_logger),
  1040. {Level, _, Msg,Metadata} = pop(),
  1041. ?assertEqual(lager_util:level_to_num(error),Level),
  1042. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1043. ?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))
  1044. end
  1045. },
  1046. {"supervisor reports with real error and pid",
  1047. fun() ->
  1048. 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}]),
  1049. _ = gen_event:which_handlers(error_logger),
  1050. {Level, _, Msg,Metadata} = pop(),
  1051. ?assertEqual(lager_util:level_to_num(error),Level),
  1052. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1053. ?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))
  1054. end
  1055. },
  1056. {"supervisor_bridge reports",
  1057. fun() ->
  1058. sync_error_logger:error_report(supervisor_report, [{errorContext, france}, {offender, [{mod, mini_steve}, {pid, bleh}]}, {reason, fired}, {supervisor, {local, steve}}]),
  1059. _ = gen_event:which_handlers(error_logger),
  1060. {Level, _, Msg,Metadata} = pop(),
  1061. ?assertEqual(lager_util:level_to_num(error),Level),
  1062. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1063. ?assertEqual("Supervisor steve had child at module mini_steve at bleh exit with reason fired in context france", lists:flatten(Msg))
  1064. end
  1065. },
  1066. {"application progress report",
  1067. fun() ->
  1068. sync_error_logger:info_report(progress, [{application, foo}, {started_at, node()}]),
  1069. _ = gen_event:which_handlers(error_logger),
  1070. {Level, _, Msg,Metadata} = pop(),
  1071. ?assertEqual(lager_util:level_to_num(info),Level),
  1072. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1073. Expected = lists:flatten(io_lib:format("Application foo started on node ~w", [node()])),
  1074. ?assertEqual(Expected, lists:flatten(Msg))
  1075. end
  1076. },
  1077. {"supervisor progress report",
  1078. fun() ->
  1079. lager:set_loglevel(?MODULE, debug),
  1080. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1081. sync_error_logger:info_report(progress, [{supervisor, {local, foo}}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1082. _ = gen_event:which_handlers(error_logger),
  1083. {Level, _, Msg,Metadata} = pop(),
  1084. ?assertEqual(lager_util:level_to_num(debug),Level),
  1085. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1086. ?assertEqual("Supervisor foo started foo:bar/1 at pid baz", lists:flatten(Msg))
  1087. end
  1088. },
  1089. {"supervisor progress report with pid",
  1090. fun() ->
  1091. lager:set_loglevel(?MODULE, debug),
  1092. ?assertEqual({?DEBUG bor ?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1093. sync_error_logger:info_report(progress, [{supervisor, somepid}, {started, [{mfargs, {foo, bar, 1}}, {pid, baz}]}]),
  1094. _ = gen_event:which_handlers(error_logger),
  1095. {Level, _, Msg,Metadata} = pop(),
  1096. ?assertEqual(lager_util:level_to_num(debug),Level),
  1097. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1098. ?assertEqual("Supervisor somepid started foo:bar/1 at pid baz", lists:flatten(Msg))
  1099. end
  1100. },
  1101. {"crash report for emfile",
  1102. fun() ->
  1103. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, emfile, [{stack, trace, 1}]}}], []]),
  1104. _ = gen_event:which_handlers(error_logger),
  1105. {Level, _, Msg,Metadata} = pop(),
  1106. ?assertEqual(lager_util:level_to_num(error),Level),
  1107. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1108. 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()])),
  1109. ?assertEqual(Expected, lists:flatten(Msg))
  1110. end
  1111. },
  1112. {"crash report for system process limit",
  1113. fun() ->
  1114. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn, 1}]}}], []]),
  1115. _ = gen_event:which_handlers(error_logger),
  1116. {Level, _, Msg,Metadata} = pop(),
  1117. ?assertEqual(lager_util:level_to_num(error),Level),
  1118. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1119. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1120. ?assertEqual(Expected, lists:flatten(Msg))
  1121. end
  1122. },
  1123. {"crash report for system process limit2",
  1124. fun() ->
  1125. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, spawn_opt, 1}]}}], []]),
  1126. _ = gen_event:which_handlers(error_logger),
  1127. {Level, _, Msg,Metadata} = pop(),
  1128. ?assertEqual(lager_util:level_to_num(error),Level),
  1129. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1130. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of processes exceeded", [self()])),
  1131. ?assertEqual(Expected, lists:flatten(Msg))
  1132. end
  1133. },
  1134. {"crash report for system port limit",
  1135. fun() ->
  1136. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, open_port, 1}]}}], []]),
  1137. _ = gen_event:which_handlers(error_logger),
  1138. {Level, _, Msg,Metadata} = pop(),
  1139. ?assertEqual(lager_util:level_to_num(error),Level),
  1140. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1141. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~w with 0 neighbours crashed with reason: system limit: maximum number of ports exceeded", [self()])),
  1142. ?assertEqual(Expected, lists:flatten(Msg))
  1143. end
  1144. },
  1145. {"crash report for system port limit",
  1146. fun() ->
  1147. sync_error_logger:error_report(crash_report, [[{pid, self()}, {registered_name, []}, {error_info, {error, system_limit, [{erlang, list_to_atom, 1}]}}], []]),
  1148. _ = gen_event:which_handlers(error_logger),
  1149. {Level, _, Msg,Metadata} = pop(),
  1150. ?assertEqual(lager_util:level_to_num(error),Level),
  1151. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1152. 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()])),
  1153. ?assertEqual(Expected, lists:flatten(Msg))
  1154. end
  1155. },
  1156. {"crash report for system ets table limit",
  1157. fun() ->
  1158. 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}]}}], []]),
  1159. _ = gen_event:which_handlers(error_logger),
  1160. {Level, _, Msg,Metadata} = pop(),
  1161. ?assertEqual(lager_util:level_to_num(error),Level),
  1162. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1163. 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])),
  1164. ?assertEqual(Expected, lists:flatten(Msg))
  1165. end
  1166. },
  1167. {"crash report for unknown system limit should be truncated at 500 characters",
  1168. fun() ->
  1169. sync_error_logger:error_report(crash_report, [[{pid, self()}, {error_info, {error, system_limit, [{wtf,boom,[string:copies("aaaa", 4096)]}]}}], []]),
  1170. _ = gen_event:which_handlers(error_logger),
  1171. {_, _, Msg,_Metadata} = pop(),
  1172. ?assert(length(lists:flatten(Msg)) > 550),
  1173. ?assert(length(lists:flatten(Msg)) < 600)
  1174. end
  1175. },
  1176. {"crash reports for 'special processes' should be handled right - function_clause",
  1177. fun() ->
  1178. {ok, Pid} = special_process:start(),
  1179. unlink(Pid),
  1180. Pid ! function_clause,
  1181. timer:sleep(500),
  1182. _ = gen_event:which_handlers(error_logger),
  1183. {_, _, Msg, _Metadata} = pop(),
  1184. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: no function clause matching special_process:foo(bar)",
  1185. [Pid])),
  1186. test_body(Expected, lists:flatten(Msg))
  1187. end
  1188. },
  1189. {"crash reports for 'special processes' should be handled right - case_clause",
  1190. fun() ->
  1191. {ok, Pid} = special_process:start(),
  1192. unlink(Pid),
  1193. Pid ! {case_clause, wtf},
  1194. timer:sleep(500),
  1195. _ = gen_event:which_handlers(error_logger),
  1196. {_, _, Msg, _Metadata} = pop(),
  1197. 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",
  1198. [Pid])),
  1199. test_body(Expected, lists:flatten(Msg))
  1200. end
  1201. },
  1202. {"crash reports for 'special processes' should be handled right - exit",
  1203. fun() ->
  1204. {ok, Pid} = special_process:start(),
  1205. unlink(Pid),
  1206. Pid ! exit,
  1207. timer:sleep(500),
  1208. _ = gen_event:which_handlers(error_logger),
  1209. {_, _, Msg, _Metadata} = pop(),
  1210. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours exited with reason: byebye in special_process:loop/0",
  1211. [Pid])),
  1212. test_body(Expected, lists:flatten(Msg))
  1213. end
  1214. },
  1215. {"crash reports for 'special processes' should be handled right - error",
  1216. fun() ->
  1217. {ok, Pid} = special_process:start(),
  1218. unlink(Pid),
  1219. Pid ! error,
  1220. timer:sleep(500),
  1221. _ = gen_event:which_handlers(error_logger),
  1222. {_, _, Msg, _Metadata} = pop(),
  1223. Expected = lists:flatten(io_lib:format("CRASH REPORT Process ~p with 0 neighbours crashed with reason: mybad in special_process:loop/0",
  1224. [Pid])),
  1225. test_body(Expected, lists:flatten(Msg))
  1226. end
  1227. },
  1228. {"webmachine error reports",
  1229. fun() ->
  1230. Path = "/cgi-bin/phpmyadmin",
  1231. Reason = {error,{error,{badmatch,{error,timeout}},
  1232. [{myapp,dostuff,2,[{file,"src/myapp.erl"},{line,123}]},
  1233. {webmachine_resource,resource_call,3,[{file,"src/webmachine_resource.erl"},{line,169}]}]}},
  1234. sync_error_logger:error_msg("webmachine error: path=~p~n~p~n", [Path, Reason]),
  1235. _ = gen_event:which_handlers(error_logger),
  1236. {Level, _, Msg,Metadata} = pop(),
  1237. ?assertEqual(lager_util:level_to_num(error),Level),
  1238. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1239. ?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))
  1240. end
  1241. },
  1242. {"Cowboy error reports, 8 arg version",
  1243. fun() ->
  1244. Stack = [{my_handler,init, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1245. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1246. sync_error_logger:error_msg(
  1247. "** Cowboy handler ~p terminating in ~p/~p~n"
  1248. " for the reason ~p:~p~n"
  1249. "** Options were ~p~n"
  1250. "** Request was ~p~n"
  1251. "** Stacktrace: ~p~n~n",
  1252. [my_handler, init, 3, error, {badmatch, {error, timeout}}, [],
  1253. "Request", Stack]),
  1254. _ = gen_event:which_handlers(error_logger),
  1255. {Level, _, Msg,Metadata} = pop(),
  1256. ?assertEqual(lager_util:level_to_num(error),Level),
  1257. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1258. ?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))
  1259. end
  1260. },
  1261. {"Cowboy error reports, 10 arg version",
  1262. fun() ->
  1263. Stack = [{my_handler,somecallback, 3,[{file,"src/my_handler.erl"},{line,123}]},
  1264. {cowboy_handler,handler_init,4,[{file,"src/cowboy_handler.erl"},{line,169}]}],
  1265. sync_error_logger:error_msg(
  1266. "** Cowboy handler ~p terminating in ~p/~p~n"
  1267. " for the reason ~p:~p~n** Message was ~p~n"
  1268. "** Options were ~p~n** Handler state was ~p~n"
  1269. "** Request was ~p~n** Stacktrace: ~p~n~n",
  1270. [my_handler, somecallback, 3, error, {badmatch, {error, timeout}}, hello, [],
  1271. {}, "Request", Stack]),
  1272. _ = gen_event:which_handlers(error_logger),
  1273. {Level, _, Msg,Metadata} = pop(),
  1274. ?assertEqual(lager_util:level_to_num(error),Level),
  1275. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1276. ?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))
  1277. end
  1278. },
  1279. {"Cowboy error reports, 5 arg version",
  1280. fun() ->
  1281. sync_error_logger:error_msg(
  1282. "** Cowboy handler ~p terminating; "
  1283. "function ~p/~p was not exported~n"
  1284. "** Request was ~p~n** State was ~p~n~n",
  1285. [my_handler, to_json, 2, "Request", {}]),
  1286. _ = gen_event:which_handlers(error_logger),
  1287. {Level, _, Msg,Metadata} = pop(),
  1288. ?assertEqual(lager_util:level_to_num(error),Level),
  1289. ?assertEqual(self(),proplists:get_value(pid,Metadata)),
  1290. ?assertEqual("Cowboy handler my_handler terminated with reason: call to undefined function my_handler:to_json/2", lists:flatten(Msg))
  1291. end
  1292. },
  1293. {"messages should not be generated if they don't satisfy the threshold",
  1294. fun() ->
  1295. lager:set_loglevel(?MODULE, error),
  1296. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1297. sync_error_logger:info_report([hello, world]),
  1298. _ = gen_event:which_handlers(error_logger),
  1299. ?assertEqual(0, count()),
  1300. ?assertEqual(0, count_ignored()),
  1301. lager:set_loglevel(?MODULE, info),
  1302. ?assertEqual({?INFO bor ?NOTICE bor ?WARNING bor ?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1303. sync_error_logger:info_report([hello, world]),
  1304. _ = gen_event:which_handlers(error_logger),
  1305. ?assertEqual(1, count()),
  1306. ?assertEqual(0, count_ignored()),
  1307. lager:set_loglevel(?MODULE, error),
  1308. ?assertEqual({?ERROR bor ?CRITICAL bor ?ALERT bor ?EMERGENCY, []}, lager_config:get(loglevel)),
  1309. lager_config:set(loglevel, {element(2, lager_util:config_to_mask(debug)), []}),
  1310. sync_error_logger:info_report([hello, world]),
  1311. _ = gen_event:which_handlers(error_logger),
  1312. ?assertEqual(1, count()),
  1313. ?assertEqual(1, count_ignored())
  1314. end
  1315. }
  1316. ]
  1317. }.
  1318. safe_format_test() ->
  1319. ?assertEqual("foo bar", lists:flatten(lager:safe_format("~p ~p", [foo, bar], 1024))),
  1320. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:safe_format("~p ~p ~p", [foo, bar], 1024))),
  1321. ok.
  1322. unsafe_format_test() ->
  1323. ?assertEqual("foo bar", lists:flatten(lager:unsafe_format("~p ~p", [foo, bar]))),
  1324. ?assertEqual("FORMAT ERROR: \"~p ~p ~p\" [foo,bar]", lists:flatten(lager:unsafe_format("~p ~p ~p", [foo, bar]))),
  1325. ok.
  1326. async_threshold_test_() ->
  1327. {foreach,
  1328. fun() ->
  1329. error_logger:tty(false),
  1330. ets:new(async_threshold_test, [set, named_table, public]),
  1331. ets:insert_new(async_threshold_test, {sync_toggled, 0}),
  1332. ets:insert_new(async_threshold_test, {async_toggled, 0}),
  1333. application:load(lager),
  1334. application:set_env(lager, error_logger_redirect, false),
  1335. application:set_env(lager, async_threshold, 2),
  1336. application:set_env(lager, async_threshold_window, 1),
  1337. application:set_env(lager, handlers, [{?MODULE, info}]),
  1338. lager:start()
  1339. end,
  1340. fun(_) ->
  1341. application:unset_env(lager, async_threshold),
  1342. application:stop(lager),
  1343. application:stop(goldrush),
  1344. ets:delete(async_threshold_test),
  1345. error_logger:tty(true)
  1346. end,
  1347. [
  1348. {"async threshold works",
  1349. fun() ->
  1350. %% we start out async
  1351. ?assertEqual(true, lager_config:get(async)),
  1352. %% put a ton of things in the queue
  1353. Workers = [spawn_monitor(fun() -> [lager:info("hello world") || _ <- lists:seq(1, 1000)] end) || _ <- lists:seq(1, 15)],
  1354. %% serialize on mailbox
  1355. _ = gen_event:which_handlers(lager_event),
  1356. timer:sleep(500),
  1357. %% By now the flood of messages will have
  1358. %% forced the backend throttle to turn off
  1359. %% async mode, but it's possible all
  1360. %% outstanding requests have been processed,
  1361. %% so checking the current status (sync or
  1362. %% async) is an exercise in race control.
  1363. %% Instead, we'll see whether the backend
  1364. %% throttle has toggled into sync mode at any
  1365. %% point in the past
  1366. ?assertMatch([{sync_toggled, N}] when N > 0,
  1367. ets:lookup(async_threshold_test, sync_toggled)),
  1368. %% 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)
  1369. collect_workers(Workers),
  1370. %% serialize on the mailbox again
  1371. _ = gen_event:which_handlers(lager_event),
  1372. %% just in case...
  1373. timer:sleep(1000),
  1374. lager:info("hello world"),
  1375. _ = gen_event:which_handlers(lager_event),
  1376. %% async is true again now that the mailbox has drained
  1377. ?assertEqual(true, lager_config:get(async)),
  1378. ok
  1379. end
  1380. }
  1381. ]
  1382. }.
  1383. collect_workers([]) ->
  1384. ok;
  1385. collect_workers(Workers) ->
  1386. receive
  1387. {'DOWN', Ref, _, _, _} ->
  1388. collect_workers(lists:keydelete(Ref, 2, Workers))
  1389. end.
  1390. produce_n_error_logger_msgs(N) ->
  1391. lists:foreach(fun (K) ->
  1392. error_logger:error_msg("Foo ~p!", [K])
  1393. end,
  1394. lists:seq(0, N-1)
  1395. ).
  1396. high_watermark_test_() ->
  1397. {foreach,
  1398. fun() ->
  1399. error_logger:tty(false),
  1400. application:load(lager),
  1401. application:set_env(lager, error_logger_redirect, true),
  1402. application:set_env(lager, handlers, [{lager_test_backend, info}]),
  1403. application:set_env(lager, async_threshold, undefined),
  1404. lager:start()
  1405. end,
  1406. fun(_) ->
  1407. application:stop(lager),
  1408. error_logger:tty(true)
  1409. end,
  1410. [
  1411. {"Nothing dropped when error_logger high watermark is undefined",
  1412. fun () ->
  1413. ok = error_logger_lager_h:set_high_water(undefined),
  1414. timer:sleep(100),
  1415. produce_n_error_logger_msgs(10),
  1416. timer:sleep(500),
  1417. ?assert(count() >= 10)
  1418. end
  1419. },
  1420. {"Mostly dropped according to error_logger high watermark",
  1421. fun () ->
  1422. ok = error_logger_lager_h:set_high_water(5),
  1423. timer:sleep(100),
  1424. produce_n_error_logger_msgs(50),
  1425. timer:sleep(1000),
  1426. ?assert(count() < 20)
  1427. end
  1428. },
  1429. {"Non-notifications are not dropped",
  1430. fun () ->
  1431. ok = error_logger_lager_h:set_high_water(2),
  1432. timer:sleep(100),
  1433. spawn(fun () -> produce_n_error_logger_msgs(300) end),
  1434. timer:sleep(50),
  1435. %% if everything were dropped, this call would be dropped
  1436. %% too, so lets hope it's not
  1437. ?assert(is_integer(count())),
  1438. timer:sleep(1000),
  1439. ?assert(count() < 10)
  1440. end
  1441. }
  1442. ]
  1443. }.
  1444. -endif.