%% Copyright (c) 2013-2019 EMQ Technologies Co., Ltd. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%%     http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.

-module(emqx_pool_SUITE).

-compile(export_all).
-compile(nowarn_export_all).

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

all() ->
    [
     {group, submit_case},
     {group, async_submit_case},
     t_unexpected
    ].

groups() ->
    [
     {submit_case, [sequence], [submit_mfa, submit_fa]},
     {async_submit_case, [sequence], [async_submit_mfa, async_submit_crash]}
    ].

init_per_suite(Config) ->
    application:ensure_all_started(gproc),
    Config.

end_per_suite(_Config) ->
    ok.

init_per_testcase(_, Config) ->
    {ok, Sup} = emqx_pool_sup:start_link(),
    [{pool_sup, Sup}|Config].

end_per_testcase(_, Config) ->
    Sup = proplists:get_value(pool_sup, Config),
    exit(Sup, normal).

submit_mfa(_Config) ->
    Result = emqx_pool:submit({?MODULE, test_mfa, []}),
    ?assertEqual(15, Result).

submit_fa(_Config) ->
    Fun = fun(X) -> case X rem 2 of 0 -> {true, X div 2}; _ -> false end end,
    Result = emqx_pool:submit(Fun, [2]),
    ?assertEqual({true, 1}, Result).

async_submit_mfa(_Config) ->
    emqx_pool:async_submit({?MODULE, test_mfa, []}),
    emqx_pool:async_submit(fun ?MODULE:test_mfa/0, []).

async_submit_crash(_) ->
    emqx_pool:async_submit(fun() -> error(unexpected_error) end).

t_unexpected(_) ->
    Pid = emqx_pool:worker(),
    ?assertEqual(ignored, gen_server:call(Pid, bad_request)),
    ?assertEqual(ok, gen_server:cast(Pid, bad_msg)),
    Pid ! bad_info,
    ok = gen_server:stop(Pid).

test_mfa() ->
    lists:foldl(fun(X, Sum) -> X + Sum end, 0, [1,2,3,4,5]).