Rakefile in embulk-input-elasticsearch-0.2.0 vs Rakefile in embulk-input-elasticsearch-0.2.1

@@ -1,10 +1,11 @@
 require "bundler/gem_tasks"
+require 'rake/testtask'
 
-task default: :build
-
-desc "Run tests"
-task :test do
-  require "test-unit"
-
-  Test::Unit::AutoRunner.run(true, './')
+desc 'Run test_unit based test'
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.test_files = Dir["test/**/test_*.rb"].sort
+  t.verbose = true
+  t.warning = false
 end
+task :default => :test