lib/active_wrapper/tasks.rb in winton-active_wrapper-0.1.0 vs lib/active_wrapper/tasks.rb in winton-active_wrapper-0.1.1

@@ -4,11 +4,12 @@
   class Tasks
     
     def initialize(options={}, &block)
       
       task :environment do
-        $db, $log = ActiveWrapper.new(options)
-        yield
+        $db, $log = ActiveWrapper.setup(options)
+        $db.establish_connection
+        yield if block
       end
       
       namespace :db do
         desc "Migrate the database with optional VERSION"
         task :migrate => :environment do
\ No newline at end of file