lib/named_seeds/railtie.rb in named_seeds-1.1.0 vs lib/named_seeds/railtie.rb in named_seeds-2.0.0

@@ -1,44 +1,26 @@
+require 'active_record/tasks/database_tasks'
+
 module NamedSeeds
   class Railtie < Rails::Railtie
 
     config.named_seeds = ActiveSupport::OrderedOptions.new
-    config.named_seeds.app_load_seed          = true
     config.named_seeds.engines_with_load_seed = []
 
     config.before_initialize do |app|
-      Rails.application.paths.add 'db/test/seeds', with: 'db/test/seeds.rb'
+      ActiveRecord::Tasks::DatabaseTasks.seed_loader = NamedSeeds::Railtie
     end
 
-    rake_tasks do
-      load "named_seeds/railties/databases.rake"
+    def load_seed
+      Rails.application.load_seed
+      engines_load_seed
     end
 
-    def setup
-      load test_seed_file if test_seed_file
-    end
 
-    def prepare
-      return unless test_seed_file
-      load_all_seeds
-      setup
-    end
-
-
     protected
 
-    def load_all_seeds
-      ActiveRecord::Tasks::DatabaseTasks.load_seed if config.named_seeds.app_load_seed
+    def engines_load_seed
       config.named_seeds.engines_with_load_seed.each { |engine| engine.load_seed }
     end
 
-    def test_seed_file
-      Rails.application.paths["db/test/seeds"].existent.first
-    end
-
   end
-
-  def self.prepare
-    NamedSeeds::Railtie.prepare
-  end
-
 end