erlbox 1.5.1

data/lib/erlbox/eunit

@@ -0,0 +1,152 @@
+#!/usr/bin/env escript
+%% Copyright (c) 2009 The Hive http://www.thehive.com/
+%%
+%% Permission is hereby granted, free of charge, to any person obtaining a copy
+%% of this software and associated documentation files (the "Software"), to deal
+%% in the Software without restriction, including without limitation the rights
+%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+%% copies of the Software, and to permit persons to whom the Software is
+%% furnished to do so, subject to the following conditions:
+%%
+%% The above copyright notice and this permission notice shall be included in
+%% all copies or substantial portions of the Software.
+%%
+%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+%% THE SOFTWARE.
+%%
+
+main(Args) ->
+    Opts = parse_args(Args, dict:new()),
+    Files = find_test_files(Opts),
+    ok = init_logging(Opts),
+    maybe_cover_compile(Opts),
+    eunit:test(test_names(Files), test_flags(Opts)),
+    maybe_analyze_cover(Opts).
+
+
+parse_args([], Opts) ->
+    Opts;
+parse_args(["-v"|Rest], Opts) ->
+    parse_args(Rest, dict:store(verbose, true, Opts));
+parse_args(["-cover"|Rest], Opts) ->
+    parse_args(Rest, dict:store(cover, true, Opts));
+parse_args(["-o", Dir|Rest], Opts) ->
+    parse_args(Rest, dict:store(cover_dir, Dir, Opts));
+parse_args(["-l", LogFile|Rest], Opts) ->
+    parse_args(Rest, dict:store(log_file, LogFile, Opts));
+parse_args(["-b", Path|Rest], Opts) ->
+    true = code:add_path(Path),
+    parse_args(Rest, dict:store(bin_dir, Path, Opts));
+parse_args(["-s", SuiteName|Rest], Opts) ->
+    Suite = list_to_atom(SuiteName),
+    F = fun(Suites) -> [Suite|Suites] end,
+    parse_args(Rest, dict:update(suites, F, [Suite], Opts));
+parse_args([Dir|Rest], Opts) ->
+    true = code:add_patha(Dir),
+    parse_args(Rest, dict:store(test_dir, Dir, Opts)).
+
+find_test_files(Opts) ->
+    case dict:find(suites, Opts) of
+        error ->
+            case dict:find(test_dir, Opts) of
+                error ->
+                    io:fwrite("Error! Must provide test directory or suite names."),
+                    [];
+
+                {ok, Dir} ->
+                    case file:list_dir(Dir) of
+                        {ok, Files} ->
+                            F = fun(File, Acc0) ->
+                                    case lists:suffix("_tests.erl", File) of
+                                        true  -> [File|Acc0];
+                                        false -> Acc0
+                                    end
+                                end,
+
+                            lists:foldl(F, [], Files);
+
+                        _ ->
+                            []
+                    end
+            end;
+
+        {ok, Suites} ->
+            Suites
+    end.
+
+init_logging(Opts) ->
+    % Turn off logging to to tty
+    ok = error_logger:tty(false),
+    % send logging to a log file
+    case dict:find(log_file, Opts) of
+        error ->
+            ok; % logging disabled
+        {ok, LogFile} ->
+            error_logger:logfile({open, LogFile})
+    end.
+
+test_names(Files) ->
+    [list_to_atom(filename:basename(F, ".erl")) || F <- Files].
+
+test_flags(Opts) ->
+    case dict:find(verbose, Opts) of
+        {ok, true}  -> [verbose];
+        {ok, false} -> [];
+        error       -> []
+    end.
+
+cover_enabled(Opts) ->
+    case dict:find(cover, Opts) of
+        {ok, true}  -> true;
+        {ok, false} -> false;
+        error       -> false
+    end.
+
+maybe_cover_compile(Opts) ->
+    case cover_enabled(Opts) of
+        true ->
+            case dict:find(bin_dir, Opts) of
+                error        -> BinDir = "./ebin";
+                {ok, BinDir} -> BinDir
+            end,
+            io:fwrite("Cover compiling modules in ~s~n", [BinDir]),
+            cover:compile_beam_directory(BinDir);
+
+        false ->
+            ok
+    end.
+
+maybe_analyze_cover(Opts) ->
+    case cover_enabled(Opts) of
+        true ->
+            io:fwrite("Generating coverage report...~n"),
+            case dict:find(cover_dir, Opts) of
+                error          -> CoverDir = "./coverage";
+                {ok, CoverDir} -> CoverDir
+            end,
+            ok = filelib:ensure_dir(filename:join(CoverDir, ".dummy")),
+    		cover:export(filename:join(CoverDir, "all.coverdata")),
+
+            Modules = lists:reverse(cover:modules()),
+            F = fun(Mod, PctList) ->
+                    {ok, {_, {Cov, NotCov}}} = cover:analyse(Mod, module),
+                    Pct = (Cov / (Cov + NotCov)) * 100,
+                    io:fwrite("~5B ~5B ~5.1f% ~p~n", [Cov, NotCov, Pct, Mod]),
+
+                    OutFile = filename:join([CoverDir,
+                        atom_to_list(Mod) ++ ".COVER.html"]),
+                    cover:analyse_to_file(Mod, OutFile, [html]),
+
+                    [Pct|PctList]
+                end,
+            PctList = lists:foldl(F, [], Modules),
+            io:fwrite("~17.1f% TOTAL", [lists:sum(PctList) / length(PctList)]);
+
+        false ->
+            ok
+    end.

data/lib/erlbox/eunit.rb

@@ -0,0 +1,96 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Included tasks for running eunit tests
+## Copyright (c) 2009 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+## -------------------------------------------------------------------
+## Constants
+
+EUNIT_SRC = FileList['test/*.erl']
+EUNIT_SRC.exclude('test/*_SUITE.erl') # exclude CT suites
+
+EUNIT_BEAM = EUNIT_SRC.pathmap('%X.beam')
+EUNIT_LOG_DIR = TEST_LOG_DIR
+EUNIT_WORK_DIR = "#{EUNIT_LOG_DIR}/working"
+
+CLOBBER.include 'coverage'
+
+## -------------------------------------------------------------------
+## Tasks
+
+rule '.beam' => "%X.erl" do |t|
+  puts "compiling #{t.source}..."
+  dir = t.name.pathmap("%d")
+  sh "erlc #{print_flags(ERLC_FLAGS)} #{expand_path(ERL_PATH)} -o #{dir} #{t.source}"
+end
+
+directory EUNIT_LOG_DIR
+directory EUNIT_WORK_DIR
+
+namespace :eunit do
+
+  desc 'Eunit test preparation'
+  task :prepare => [EUNIT_LOG_DIR] do
+    # Remove the working directory to ensure there isn't stale data laying around
+    FileUtils.rm_rf EUNIT_WORK_DIR
+    # Always compile tests with debug info
+    puts 'Debugging is enabled for test builds.'
+    ERLC_FLAGS << '+debug_info'
+  end
+
+  desc 'Compile eunit test sources'
+  task :compile => ['prepare', 'build:compile'] + EUNIT_BEAM
+
+  desc 'Run eunit tests'
+  task :test => [:compile, EUNIT_WORK_DIR] do
+    run_eunit('test', ENV['cover'])
+  end
+
+end
+
+task :eunit => 'eunit:test'
+
+## -------------------------------------------------------------------
+## Helpers
+
+def run_eunit(dir, cover = false, rest = '')
+  puts "running tests in #{dir}#{' with coverage' if cover}..."
+
+  log_dir = abspath(EUNIT_LOG_DIR)
+
+  cover_flags = cover ? "-cover -o #{log_dir}/coverage" : ''
+
+  suites = ENV['suites']
+  all_suites = ''
+  suites.each(' ') {|s| all_suites << "-s #{s.strip} "} if suites
+
+  script = __FILE__.sub('.rb', '')
+
+  cmd = "cd #{EUNIT_WORK_DIR} &&\
+         #{script} -b #{abspath('./ebin')} -l #{log_dir}/eunit.log\
+                   #{cover_flags} #{all_suites} #{abspath(dir)}"
+
+  puts cmd.squeeze(' ') if verbose?
+
+  sh cmd
+end

data/lib/erlbox/faxien.rb

@@ -0,0 +1,65 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Included tasks for packaging and publishing to Faxien
+## Copyright (c) 2008 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+## -------------------------------------------------------------------
+## Constants
+
+PACKAGE_DIR="#{APP_NAME}-#{erl_app_version(APP_NAME)}"
+
+CLOBBER.include PACKAGE_DIR
+
+## -------------------------------------------------------------------
+## Rules
+
+directory PACKAGE_DIR
+
+## -------------------------------------------------------------------
+## Tasks
+
+namespace :faxien do
+
+  desc "Prepare the application for packaging"
+  task :prepare do
+    FileUtils.rm_rf PACKAGE_DIR
+  end
+
+  desc "Package app for publication to a faxien repo"
+  task :package => [:prepare, PACKAGE_DIR] do
+    FileUtils.cp_r 'ebin', PACKAGE_DIR
+    FileUtils.cp_r 'src', PACKAGE_DIR
+    FileUtils.cp_r 'include', PACKAGE_DIR if File.exist?('include')
+    FileUtils.cp_r 'priv', PACKAGE_DIR if File.exist?('priv')
+    FileUtils.cp_r 'mibs', PACKAGE_DIR if File.exist?('mibs')
+    puts "Packaged to #{PACKAGE_DIR}"
+  end
+
+  desc "Publish a packaged application to a faxien repo"
+  task :publish do
+    sh "faxien publish #{PACKAGE_DIR}"
+  end
+
+end
+
+task :faxien => ['faxien:package', 'faxien:publish']

data/lib/erlbox/install.rb

@@ -0,0 +1,38 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Helper tasks for installing an app into erlang
+## Copyright (c) 2009 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+namespace :install do
+
+  task :appid, [:root_dir] do |t, args|
+    puts "#{APP_NAME}-#{erl_app_version(APP_NAME, :erl_root => args.root_dir)}"
+  end
+
+  desc "Hook to allow bootstrapping a new repo during installation"
+  task :prepare
+
+  desc "Build the application for installation"
+  task :build => [:compile]
+
+end

data/lib/erlbox/recurse.rb

@@ -0,0 +1,44 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Recursive task definitions
+## Copyright (c) 2008 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+verbose false
+
+DIRS = FileList.new
+
+def recurse(task_name)
+  DIRS.each do |dir|
+    puts "===> Building #{dir}"
+    sh "(cd #{dir} && rake --silent #{task_name})"
+  end
+end
+
+def recursive_task(target)
+  task_name = target.kind_of?(Hash) ? target.keys.first : target
+
+  desc "Run #{task_name} in subdirs"
+  task target do
+    recurse task_name
+  end
+end

data/lib/erlbox/release.rb

@@ -0,0 +1,72 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: OTP release tasks
+## Copyright (c) 2008 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+require 'rake/clean'
+include FileUtils
+
+if !defined?(REL_APPNAME)
+  fail "The variable REL_APPNAME must be defined"
+end
+
+verbose false
+
+CLEAN.include FileList["#{REL_APPNAME}*"]
+
+REL_APPS = %w( runtime_tools )
+REL_VERSION  = `cat vers/#{REL_APPNAME}.version`.strip
+REL_FULLNAME = "#{REL_APPNAME}-#{REL_VERSION}"
+
+ERTS_VSN = `scripts/get-erts-vsn`.strip
+
+directory REL_APPNAME
+
+
+task :build_app do
+  cd "../apps" do
+    sh "rake"
+  end
+end
+
+desc "Run the make-rel script"
+task :make_rel do
+  sh "scripts/make-rel #{REL_APPNAME} #{REL_VERSION} #{REL_APPS.join(' ')}"
+end
+
+task :prepare => [REL_APPNAME, :make_rel] do
+  sh "tar -xzf #{REL_FULLNAME}.tar.gz -C #{REL_APPNAME}"
+  rm "#{REL_FULLNAME}.tar.gz"
+
+  sh %Q(echo "#{ERTS_VSN} #{REL_VERSION}" > #{REL_APPNAME}/releases/start_erl.data)
+  cp_r Dir.glob("overlays/#{REL_APPNAME}/*"), REL_APPNAME
+end
+
+desc "Stage the application into a directory"
+task :stage => [:clean, :build_app, :prepare]
+
+desc "Create a tarball from the staged application"
+task :release => :stage do
+  mv REL_APPNAME, REL_FULLNAME
+  sh "tar -cjf #{REL_FULLNAME}-#{RUBY_PLATFORM}.tar.bz2 #{REL_FULLNAME}"
+  rm_rf REL_FULLNAME
+end

data/lib/erlbox/snmp.rb

@@ -0,0 +1,55 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Optional SNMP Tasks
+## Copyright (c) 2008 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+if !defined?(APP_NAME)
+  fail "You must require 'erlbox' before requiring this file"
+end
+
+## -------------------------------------------------------------------
+## Constants
+
+MIBS_DIR = 'priv/mibs'
+MIB_SRCS = FileList["mibs/*.mib"]
+MIB_BINS = MIB_SRCS.pathmap("%{mibs,#{MIBS_DIR}}X.bin")
+
+CLEAN.include MIBS_DIR
+
+## -------------------------------------------------------------------
+## Rules
+
+directory MIBS_DIR
+
+rule ".bin" => ["%{#{MIBS_DIR},mibs}X.mib"] do |t|
+  puts "compiling #{t.source}..."
+  sh "erlc -I #{MIBS_DIR} -o #{MIBS_DIR} #{t.source}"
+end
+
+## -------------------------------------------------------------------
+## Tasks
+
+desc "Compile SNMP MIBs"
+task :snmp => [MIBS_DIR] + MIB_BINS
+
+task :compile => :snmp

data/lib/erlbox/test.rb

@@ -0,0 +1,165 @@
+## -------------------------------------------------------------------
+##
+## Erlang Toolbox: Included tasks for running tests
+## Copyright (c) 2008 The Hive http://www.thehive.com/
+##
+## Permission is hereby granted, free of charge, to any person obtaining a copy
+## of this software and associated documentation files (the "Software"), to deal
+## in the Software without restriction, including without limitation the rights
+## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+## copies of the Software, and to permit persons to whom the Software is
+## furnished to do so, subject to the following conditions:
+##
+## The above copyright notice and this permission notice shall be included in
+## all copies or substantial portions of the Software.
+##
+## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+## THE SOFTWARE.
+##
+## -------------------------------------------------------------------
+
+## -------------------------------------------------------------------
+## Constants
+
+UNIT_TEST_FLAGS = []
+INT_TEST_FLAGS  = []
+PERF_TEST_FLAGS = []
+
+TEST_ROOT = "."
+TEST_LOG_DIR = "#{TEST_ROOT}/logs"
+
+UNIT_TEST_DIR = "#{PWD}/#{TEST_ROOT}/test"
+INT_TEST_DIR  = "#{PWD}/#{TEST_ROOT}/int_test"
+PERF_TEST_DIR = "#{PWD}/#{TEST_ROOT}/perf_test"
+
+CLOBBER.include TEST_LOG_DIR
+
+## -------------------------------------------------------------------
+## Tasks
+
+namespace :test do
+
+  desc "Test preparation run before all tests"
+  task :prepare do
+    fail "No tests defined" unless File.directory?(TEST_ROOT)
+    Dir.mkdir(TEST_LOG_DIR) unless File.directory?(TEST_LOG_DIR)
+
+    # Always compile tests with debug info
+    puts "Debugging is enabled for test builds."
+    ERLC_FLAGS << '+debug_info'
+  end
+
+  desc "Show test results in a browser"
+  task :results do
+    `open #{TEST_LOG_DIR}/index.html`
+  end
+
+  ['unit', 'int', 'perf'].each do |type|
+    desc "Run #{type} tests"
+    task type => "test:#{type}:prepare" do
+      check_and_run_tests(type, false)
+    end
+
+    namespace type do
+      desc "Prepare #{type} tests"
+      task :prepare => ['^prepare', 'compile']
+
+      desc "Compile #{type} tests"
+      task :compile => 'rake:compile'
+
+      desc "Run #{type} tests with coverage"
+      task :cover => :prepare do
+        check_and_run_tests(type, true)
+      end
+    end
+  end
+
+  desc "Run all tests"
+  task :all => [:unit, :int, :perf]
+
+end
+
+task :test => 'test:unit'
+task :int_test => 'test:int'
+task :perf_test => 'test:perf'
+
+## -------------------------------------------------------------------
+## Helpers
+
+def test_dir(type)
+  eval("#{type.to_s.upcase}_TEST_DIR")
+end
+
+def check_and_run_tests(type, use_cover = false)
+  dir = test_dir(type)
+  if File.directory?(dir)
+    run_tests(dir, use_cover, print_flags(eval("#{type.upcase}_TEST_FLAGS")))
+  else
+    puts "No #{type} tests defined. Skipping."
+  end
+end
+
+def run_tests(dir, cover = false, rest = "")
+  puts "running tests in #{dir}#{' with coverage' if cover}..."
+
+  config_flags = ""
+  if File.exists?("#{dir}/test.config")
+    config_flags << "-ct_config #{dir}/test.config"
+  end
+
+  if File.exists?("#{dir}/app.config")
+    config_flags << " -config #{dir}/app.config"
+  end
+
+  cmd = "erl #{expand_path(ERL_PATH)} -pa #{PWD}/ebin -I#{PWD}/include\
+             -noshell\
+             -s ct_run script_start\
+             -s erlang halt\
+             -name test@#{`hostname`.strip}\
+             #{cover_flags(dir, cover)}\
+             #{get_suites(dir)}\
+             -logdir #{TEST_LOG_DIR}\
+             -env TEST_DIR #{dir}\
+             #{config_flags} #{rest}"
+
+  if !defined?(NOISY_TESTS) && !verbose?
+    output = `#{cmd}`
+
+    fail if $?.exitstatus != 0 && !ENV["stop_on_fail"].nil?
+
+    File.open("#{PWD}/#{TEST_LOG_DIR}/raw.log", "w") do |file|
+      file.write "--- Test run on #{Time.now.to_s} ---\n"
+      file.write output
+      file.write "\n\n"
+    end
+
+    if output[/, 0 failed/]
+      puts "==> " + output[/TEST COMPLETE,.*test cases$/]
+    else
+      puts output
+    end
+  else
+    puts cmd.squeeze(' ')
+    sh cmd
+  end
+end
+
+def cover_flags(dir, use_cover)
+  use_cover ? "-cover #{dir}/cover.spec" : ""
+end
+
+def get_suites(dir)
+  suites = ENV['suites']
+  if suites
+    all_suites = ""
+    suites.each(' ') {|s| all_suites << "#{dir}/#{s.strip}_SUITE "}
+    "-suite #{all_suites}"
+  else
+    "-dir #{dir}"
+  end
+end