lib/multiverse.rb in multiverse-0.0.1 vs lib/multiverse.rb in multiverse-0.0.2

@@ -28,14 +28,18 @@
     end
 
     def migrate_path
       "#{db_dir}/migrate"
     end
+
+    def env(environment)
+      db ? "#{db}_#{environment}" : environment
+    end
   end
 end
 
 ActiveSupport.on_load(:active_record) do
   ActiveRecord::Tasks::DatabaseTasks.singleton_class.prepend Multiverse::DatabaseTasks
   ActiveRecord::Migration.prepend Multiverse::Migration
-  ActiveRecord::SchemaMigration.singleton_class.prepend Multiverse::SchemaMigration
+  ActiveRecord::Migrator.prepend Multiverse::Migrator
   ActiveRecord::SchemaDumper.singleton_class.prepend Multiverse::SchemaDumper
 end