seed-fu-ndo 0.0.1 → 0.0.2

data/lib/seed-fu-ndo/recorder.rb

@@ -13,15 +13,6 @@ module SeedFuNdo
       seeders << seeder
     end
     
-    # Unseed the data of all the recorded seeders in reverse order.
-    def unseed
-      ActiveRecord::Base.transaction do 
-        seeders.reverse.each do |seeder|
-          seeder.unseed
-        end
-      end
-    end
-    
   end
   
 end

data/lib/seed-fu-ndo/seeder.rb

@@ -3,6 +3,8 @@ module SeedFu
   # Enhanced class from Seed Fu to allow for recording and destroying seed data.
   class Seeder
     
+    attr_reader :model_class
+    
     # Record instead of inserting the data if in recording mode. 
     def seed_with_undo
       if r = SeedFuNdo.recorder
@@ -19,17 +21,19 @@ module SeedFu
     # Destroy the seed data.
     def unseed
       @model_class.transaction do
-        @data.reverse.map { |record_data| unseed_record(record_data.symbolize_keys) }
+        existing_records.reverse.each { |record| unseed_record(record) }
       end
     end
     
+    # List of the seeded records that exist in the database.
+    def existing_records
+      @data.map { |record_data| find_record(record_data.symbolize_keys) }.compact
+    end
+    
     private
     
-    def unseed_record(data)
-        record = find_record(data)
-        return if record.nil?
-
-        puts " - Remove #{@model_class} #{data.inspect}" unless @options[:quiet]
+    def unseed_record(record)
+        puts " - Remove #{record.inspect}" unless @options[:quiet]
 
         record.destroy || raise(ActiveRecord::ActiveRecordError, record.errors.full_messages.join(", ").presence)
     end

data/lib/seed-fu-ndo/version.rb

@@ -1,4 +1,4 @@
 module SeedFuNdo
   # The current version of Seed Fu-ndo
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

data/lib/seed-fu-ndo.rb

@@ -11,11 +11,30 @@ module SeedFuNdo
   # [fixture_paths] The paths to look for seed files in.
   # [filter] A regexp. If given, only filenames matching this expression will be loaded.
   def self.unseed(fixture_paths = SeedFu.fixture_paths, filter = nil)
+    capture(fixture_paths, filter).reverse.each do |seeder|
+      seeder.unseed
+    end
+  end
+  
+  # Returns all seed records from the given fixture_paths that currently
+  # exist in the database. No changes to the database are performed.
+  def self.existing_seeds(fixture_paths = SeedFu.fixture_paths, filter = nil)
+    existing = Hash.new {|h, k| h[k] = [] }
+    capture(fixture_paths, filter).each do |seeder|
+      existing[seeder.model_class] += seeder.existing_records
+    end
+    existing
+  end
+  
+  # Captures the seed fixtures without actually inserting anything.
+  # Returns a list of seeder objects.
+  def self.capture(fixture_paths = SeedFu.fixture_paths, filter = nil)
     @@recorder = Recorder.new
     SeedFu::Runner.new(fixture_paths, filter).run
-    @@recorder.unseed
+    @@recorder.seeders
   ensure
     @@recorder = nil
   end
   
+  
 end

data/spec/seeder_spec.rb

@@ -44,15 +44,20 @@ describe SeedFu::Seeder do
       end
     
       it "should ignore non-existing records" do
-        SeedFuNdo.recorder.unseed
+        unseed
         SeededModel.count.should == 0
       end
       
       it "should unseed existing records" do
         SeededModel.create!(:title => 'Foo')
-        SeedFuNdo.recorder.unseed
+        unseed
         SeededModel.count.should == 0
       end
+      
+      it "should find existing records" do
+        rec = SeededModel.create!(:title => 'Foo')
+        SeedFuNdo.recorder.seeders.first.existing_records.should == [rec]
+      end
     end
     
     context "for multiple seeds" do
@@ -71,10 +76,14 @@ describe SeedFu::Seeder do
       it "should rollback if one record cannot be destroyed" do
         SeededModel.create!(:title => 'Foo', :fail_to_destroy => true)
         SeededModel.create!(:title => 'Bar')
-        expect { SeedFuNdo.recorder.unseed }.to raise_error(ActiveRecord::ActiveRecordError)
+        expect { unseed }.to raise_error(ActiveRecord::ActiveRecordError)
         SeededModel.count.should == 2
       end
     end
 
   end
 end
+
+def unseed
+  SeedFuNdo.recorder.seeders.reverse.each {|s| s.unseed }
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: seed-fu-ndo
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: 
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Pascal Zumkehr
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-06-06 00:00:00 Z
+date: 2012-06-14 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   version_requirements: &id001 !ruby/object:Gem::Requirement