Add riak_test behavior

src/riak_test.erl

@@ -22,6 +22,14 @@
 -module(riak_test).
 -export([main/1]).
 
+%% Define the riak_test behavior
+-export([behaviour_info/1]).
+
+behaviour_info(callbacks) ->
+    [{confirm,0}];
+behaviour_info(_Other) ->
+    undefined.
+
 add_deps(Path) ->
     {ok, Deps} = file:list_dir(Path),
     [code:add_path(lists:append([Path, "/", Dep, "/ebin"])) || Dep <- Deps],

tests/basic_command_line.erl

@@ -20,6 +20,7 @@
 -module(basic_command_line).
 -include_lib("eunit/include/eunit.hrl").
 
+-behavior(riak_test).
 -export([confirm/0]).
 
 confirm() ->

tests/client_java_verify.erl

@@ -1,4 +1,5 @@
 -module(client_java_verify).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/client_python_verify.erl

@@ -1,4 +1,5 @@
 -module(client_python_verify).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/client_ruby_verify.erl

@@ -1,4 +1,5 @@
 -module(client_ruby_verify).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/gh_riak_core_154.erl

@@ -21,6 +21,7 @@
 %% Automated test for issue riak_core#154
 %% Hinted handoff does not occur after a node has been restarted in Riak 1.1
 -module(gh_riak_core_154).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/gh_riak_core_155.erl

@@ -18,6 +18,8 @@
 %%
 %% -------------------------------------------------------------------
 -module(gh_riak_core_155).
+-behavior(riak_test).
+-export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").
 

tests/gh_riak_core_176.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(gh_riak_core_176).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/loaded_upgrade.erl

@@ -18,6 +18,8 @@
 %%
 %% -------------------------------------------------------------------
 -module(loaded_upgrade).
+-behavior(riak_test).
+-export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").
 -define(SPAM_BUCKET, <<"scotts_spam">>).

tests/mapred_verify_rt.erl

@@ -22,6 +22,7 @@
 
 -module(mapred_verify_rt).
 
+-behavior(riak_test).
 -export([confirm/0]).
 
 -define(NODE_COUNT, 3).

tests/partition_repair.erl

@@ -21,6 +21,9 @@
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").
 
+-behavior(riak_test).
+-export([confirm/0]).
+
 -define(FMT(S, L), lists:flatten(io_lib:format(S, L))).
 
 %% @doc This test verifies that partition repair successfully repairs

tests/replication.erl

@@ -1,4 +1,5 @@
 -module(replication).
+-behavior(riak_test).
 -export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").

tests/replication2.erl

@@ -1,4 +1,5 @@
 -module(replication2).
+-behavior(riak_test).
 -export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").

tests/replication_ssl.erl

@@ -1,4 +1,6 @@
 -module(replication_ssl).
+-behavior(riak_test).
+-export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").
 

tests/replication_upgrade.erl

@@ -1,5 +1,6 @@
 -module(replication_upgrade).
--compile(export_all).
+-behavior(riak_test).
+-export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 
 confirm() ->

tests/riaknostic_rt.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(riaknostic_rt).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/rolling_capabilities.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(rolling_capabilities).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/rpc_output.erl

@@ -19,7 +19,9 @@
 %% -------------------------------------------------------------------
 -module(rpc_output).
 
+-behavior(riak_test).
 -export([confirm/0]).
+
 -include_lib("eunit/include/eunit.hrl").
 -compile([{parse_transform, lager_transform}]).
 

tests/rt_basic_test.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(rt_basic_test).
+-behavior(riak_test).
 -export([confirm/0]).
 
 confirm() ->

tests/secondary_index_tests.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(secondary_index_tests).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/test_cluster.erl

@@ -18,7 +18,8 @@
 %%
 %% -------------------------------------------------------------------
 -module(test_cluster).
--compile(export_all).
+-behavior(riak_test).
+-export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 
 confirm() ->

tests/verify_backup_restore.erl

@@ -24,6 +24,7 @@
 %% depending on the configuration (last write wins, vector clocks used, etc).
 
 -module(verify_backup_restore).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("riakc/include/riakc.hrl").
 -include_lib("eunit/include/eunit.hrl").

tests/verify_basic_upgrade.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_basic_upgrade).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_build_cluster.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_build_cluster).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_busy_dist_port.erl

@@ -42,6 +42,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_busy_dist_port).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_capabilities.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_capabilities).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_claimant.erl

@@ -18,7 +18,10 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_claimant).
--compile(export_all).
+
+-behavior(riak_test).
+-export([confirm/0]).
+
 -include_lib("eunit/include/eunit.hrl").
 
 -import(rt, [build_cluster/1,

tests/verify_commit_hooks.erl

@@ -19,6 +19,7 @@
 %% -------------------------------------------------------------------
 -module(verify_commit_hooks).
 -include_lib("eunit/include/eunit.hrl").
+-behavior(riak_test).
 -export([confirm/0]).
 
 confirm() ->

tests/verify_counter_converge.erl

@@ -24,7 +24,8 @@
 %%% @end
 
 -module(verify_counter_converge).
--compile(export_all).
+-behavior(riak_test).
+-export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 
 confirm() ->

tests/verify_down.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_down).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_leave.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_leave).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_listkeys.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_listkeys).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_reset_bucket_props.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_reset_bucket_props).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_riak_lager.erl

@@ -19,7 +19,9 @@
 %% -------------------------------------------------------------------
 -module(verify_riak_lager).
 
+-behavior(riak_test).
 -export([confirm/0]).
+
 -include_lib("eunit/include/eunit.hrl").
 -include_lib("kernel/include/file.hrl").
 

tests/verify_riak_stats.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_riak_stats).
+-behavior(riak_test).
 -export([confirm/0]).
 -include_lib("eunit/include/eunit.hrl").
 

tests/verify_staged_clustering.erl

@@ -18,6 +18,7 @@
 %%
 %% -------------------------------------------------------------------
 -module(verify_staged_clustering).
+-behavior(riak_test).
 -export([confirm/0]).
 -compile(export_all).
 -include_lib("eunit/include/eunit.hrl").