Add SASL test suite

1 files changed, 175 insertions, 0 deletions

diff --git a/lib/sasl/test/overload_SUITE.erl b/lib/sasl/test/overload_SUITE.erl
new file mode 100644
index 0000000000..92b1aaed6e
--- /dev/null
+++ b/lib/sasl/test/overload_SUITE.erl
@@ -0,0 +1,175 @@
+%%
+%% %CopyrightBegin%
+%%
+%% Copyright Ericsson AB 2011. All Rights Reserved.
+%%
+%% The contents of this file are subject to the Erlang Public License,
+%% Version 1.1, (the "License"); you may not use this file except in
+%% compliance with the License. You should have received a copy of the
+%% Erlang Public License along with this software. If not, it can be
+%% retrieved online at http://www.erlang.org/.
+%%
+%% Software distributed under the License is distributed on an "AS IS"
+%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
+%% the License for the specific language governing rights and limitations
+%% under the License.
+%%
+%% %CopyrightEnd%
+%%
+
+-module(overload_SUITE).
+-include("test_server.hrl").
+
+-compile(export_all).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+all() -> [info, set_config_data, set_env_vars, request, timeout].
+all(suite) -> all().
+
+init_per_testcase(_Case,Config) ->
+    restart_sasl(),
+    Config.
+
+end_per_testcase(Case,Config) ->
+    try apply(?MODULE,Case,[cleanup,Config])
+    catch error:undef -> ok
+    end,
+    ok.
+
+%%%-----------------------------------------------------------------
+info(suite) -> [];
+info(_Config) ->
+    ?line Info = overload:get_overload_info(),
+    ?line [{total_intensity,0.0},
+	   {accept_intensity,0.0},
+	   {max_intensity,0.8},
+	   {weight,0.1},
+	   {total_requests,0},
+	   {accepted_requests,0}] = Info.
+
+%%%-----------------------------------------------------------------
+set_config_data(suite) -> [];
+set_config_data(_Config) ->
+    ?line InfoDefault = overload:get_overload_info(),
+    ?line ok = check_info(0.8,0.1,InfoDefault),
+    ?line ok = overload:set_config_data(0.5,0.4),
+    ?line Info1 = overload:get_overload_info(),
+    ?line ok = check_info(0.5,0.4,Info1),
+    ok.
+
+%%%-----------------------------------------------------------------
+set_env_vars(suite) -> [];
+set_env_vars(_Config) ->
+    ?line InfoDefault = overload:get_overload_info(),
+    ?line ok = check_info(0.8,0.1,InfoDefault),
+    ?line ok = application:set_env(sasl,overload_max_intensity,0.5),
+    ?line ok = application:set_env(sasl,overload_weight,0.4),
+    ?line ok = application:stop(sasl),
+    ?line ok = application:start(sasl),
+    ?line Info1 = overload:get_overload_info(),
+    ?line ok = check_info(0.5,0.4,Info1),
+    ok.
+set_env_vars(cleanup,_Config) ->
+    application:unset_env(sasl,overload_max_intensity),
+    application:unset_env(sasl,overload_weight),
+    ok.
+
+%%%-----------------------------------------------------------------
+request(suite) -> [];
+request(_Config) ->
+    %% Find number of request that can be done with default settings
+    %% and no delay
+    ?line overload:set_config_data(0.8, 0.1),
+    ?line NDefault = do_many_requests(0),
+    ?line restart_sasl(),
+    ?line ?t:format("NDefault: ~p",[NDefault]),
+ 
+    %% Check that the number of requests increases when max_intensity
+    %% increases
+    ?line overload:set_config_data(2, 0.1),
+    ?line NLargeMI = do_many_requests(0),
+    ?line restart_sasl(),
+    ?line ?t:format("NLargeMI: ~p",[NLargeMI]),
+    ?line true = NLargeMI > NDefault,
+
+    %% Check that the number of requests decreases when weight
+    %% increases
+    ?line overload:set_config_data(0.8, 1),
+    ?line NLargeWeight = do_many_requests(0),
+    ?line restart_sasl(),
+    ?line ?t:format("NLargeWeight: ~p",[NLargeWeight]),
+    ?line true = NLargeWeight < NDefault,
+
+    %% Check that number of requests increases when delay between
+    %% requests increases.
+    %% (Keeping same config and comparing to large weight in order to
+    %% minimize the time needed for this case.)
+    ?line overload:set_config_data(0.8, 1),
+    ?line NLargeTime = do_many_requests(500),
+    ?line restart_sasl(),
+    ?line ?t:format("NLargeTime: ~p",[NLargeTime]),
+    ?line true = NLargeTime > NLargeWeight,
+    ok.
+
+%%%-----------------------------------------------------------------
+timeout(suite) -> [];
+timeout(_Config) ->
+    ?line overload:set_config_data(0.8, 1),
+    ?line _N = do_many_requests(0),
+    
+    %% Check that the overload alarm is raised
+    ?line [{overload,_}] = alarm_handler:get_alarms(),
+
+    %% Fake a clear timeout in overload.erl and check that, since it
+    %% came very soon after the overload situation, the alarm is not
+    %% cleared
+    ?line overload ! timeout,
+    ?line timer:sleep(1000),
+    ?line [{overload,_}] = alarm_handler:get_alarms(),
+
+    %% A bit later, try again and check that this time the alarm is
+    %% cleared
+    ?line overload ! timeout,
+    ?line timer:sleep(1000),
+    ?line [] = alarm_handler:get_alarms(),
+
+    ok.
+
+
+%%%-----------------------------------------------------------------
+%%% INTERNAL FUNCTIONS
+
+%%%-----------------------------------------------------------------
+%%% Call overload:request/0 up to 30 times with the given time delay
+%%% between. Stop when 'reject' is returned.
+do_many_requests(T) ->
+    30 - do_requests(30,T).
+
+do_requests(0,_) ->
+    ?t:fail(never_rejected);
+do_requests(N,T) ->
+    case overload:request() of
+	accept ->
+	    timer:sleep(T),
+	    do_requests(N-1,T);
+	reject ->
+	    N
+    end.
+
+%%%-----------------------------------------------------------------
+%%% Restart the sasl application
+restart_sasl() ->
+    application:stop(sasl),
+    application:start(sasl),
+    ok.
+
+%%%-----------------------------------------------------------------
+%%% Check that max_intensity and weight is set as expected
+check_info(MI,W,Info) -> 
+    case {lists:keyfind(max_intensity,1,Info), lists:keyfind(weight,1,Info)} of
+	{{_,MI},{_,W}} -> ok;
+	_ -> ?t:fail({unexpected_info,MI,W,Info})
+    end.
+    
+