eLog/test/pr_stacktrace_test.erl

-module(pr_stacktrace_test).

-compile([{parse_transform, lager_transform}]).

-ifdef(OTP_RELEASE). %% this implies 21 or higher
-define(EXCEPTION(Class, Reason, Stacktrace), Class:Reason:Stacktrace).
-define(GET_STACK(Stacktrace), Stacktrace).
-else.
-define(EXCEPTION(Class, Reason, _), Class:Reason).
-define(GET_STACK(_), erlang:get_stacktrace()).
-endif.

-include_lib("eunit/include/eunit.hrl").

make_throw() ->
   throw({test, exception}).

bad_arity() ->
   lists:concat([], []).

bad_arg() ->
   integer_to_list(1.0).

pr_stacktrace_throw_test() ->
   Got = try
            make_throw()
         catch
            ?EXCEPTION(Class, Reason, Stacktrace) ->
               eRum:pr_stacktrace(?GET_STACK(Stacktrace), {Class, Reason})
         end,
   Want = "pr_stacktrace_test:pr_stacktrace_throw_test/0 line 26\n    pr_stacktrace_test:make_throw/0 line 16\nthrow:{test,exception}",
   ?assertNotEqual(nomatch, string:find(Got, Want)).

pr_stacktrace_bad_arg_test() ->
   Got = try
            bad_arg()
         catch
            ?EXCEPTION(Class, Reason, Stacktrace) ->
               eRum:pr_stacktrace(?GET_STACK(Stacktrace), {Class, Reason})
         end,
   Want = "pr_stacktrace_test:pr_stacktrace_bad_arg_test/0 line 36\n    pr_stacktrace_test:bad_arg/0 line 22\nerror:badarg",
   ?assertNotEqual(nomatch, string:find(Got, Want)).

pr_stacktrace_bad_arity_test() ->
   Got = try
            bad_arity()
         catch
            ?EXCEPTION(Class, Reason, Stacktrace) ->
               eRum:pr_stacktrace(?GET_STACK(Stacktrace), {Class, Reason})
         end,
   Want = "pr_stacktrace_test:pr_stacktrace_bad_arity_test/0 line 46\n    lists:concat([], [])\nerror:undef",
   ?assertNotEqual(nomatch, string:find(Got, Want)).

pr_stacktrace_no_reverse_test() ->
   application:set_env(lager, reverse_pretty_stacktrace, false),
   Got = try
            bad_arity()
         catch
            ?EXCEPTION(Class, Reason, Stacktrace) ->
               eRum:pr_stacktrace(?GET_STACK(Stacktrace), {Class, Reason})
         end,
   Want = "error:undef\n    lists:concat([], [])\n    pr_stacktrace_test:pr_stacktrace_bad_arity_test/0 line 57",
   ?assertEqual(nomatch, string:find(Got, Want)).