-module(pr_stacktrace_test).

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

-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
            Class:Reason:Stacktrace ->
               lager:pr_stacktrace(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
            Class:Reason:Stacktrace ->
               lager:pr_stacktrace(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
            Class:Reason:Stacktrace ->
               lager:pr_stacktrace(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
            Class:Reason:Stacktrace ->
               lager:pr_stacktrace(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)).