lib/ardb/runner/create_command.rb in ardb-0.24.0 vs lib/ardb/runner/create_command.rb in ardb-0.25.0

@@ -1,22 +1,24 @@
 require 'active_record'
 require 'ardb/runner'
 
 class Ardb::Runner::CreateCommand
 
-  def initialize
+  def initialize(out_io = nil, err_io = nil)
+    @out_io = out_io || $stdout
+    @err_io = err_io || $stderr
     @adapter = Ardb::Adapter.send(Ardb.config.db.adapter)
   end
 
   def run
     begin
       @adapter.create_db
-      $stdout.puts "created #{Ardb.config.db.adapter} db `#{Ardb.config.db.database}`"
+      @out_io.puts "created #{Ardb.config.db.adapter} db `#{Ardb.config.db.database}`"
     rescue Ardb::Runner::CmdError => e
       raise e
-    rescue Exception => e
-      $stderr.puts e
-      $stderr.puts "error creating #{Ardb.config.db.database.inspect} database"
+    rescue StandardError => e
+      @err_io.puts e
+      @err_io.puts "error creating #{Ardb.config.db.database.inspect} database"
       raise Ardb::Runner::CmdFail
     end
   end
 
 end