Merge pull request #1 from bopm/bopm-patch-1

Fix seeding for a first time and duplication on second run

features/seed_once.feature

@@ -1,5 +0,0 @@
-Feature: Seed Once
-  Scenario: Seed a record once
-    Given I have already seeded a record
-    When I seed the record using seed_once
-    Then I should not seed the record again

features/step_definitions/given/i_have_already_seeded_a_record.rb

@@ -1,6 +0,0 @@
-Given(/^I have already seeded a record$/) do
-  record = Record.new
-  record[:_id] = 1
-  record[:data] = 'existing'
-  record.upsert
-end

features/step_definitions/then/i_should_not_seed_the_record_again.rb

@@ -1,4 +0,0 @@
-Then(/^I should not seed the record again$/) do
-  Record.count.should == 1
-  Record.find(1).data.should == 'existing'
-end

features/step_definitions/when/i_seed_the_record_using_seed_once.rb

@@ -1,6 +0,0 @@
-When(/^I seed the record using seed_once$/) do
-  Record.seed_once :id do |s|
-    s.id = 1
-    s.data = 'data'
-  end
-end

lib/seed-fu-mongoid.rb

@@ -27,8 +27,12 @@ end
 
 module Mongoid::Document
   module ClassMethods
-    def seed_once(*constraints, &block)
-      SeedFuMongoid::DocumentSeeder.new(self, constraints, block).seed_once!
+    def seed_once(*constraints)
+      seeder = SeedFuMongoid::DocumentSeeder.new(self, constraints, block)
+
+      if seeder.new?
+        seeder.seed!
+      end
     end
 
     def seed(*constraints_and_objects, &block)

lib/seed-fu-mongoid/document_seeder.rb

@@ -48,7 +48,7 @@ module SeedFuMongoid
 
       def document
         @doc ||= begin
-          @klass.find_by(constraint_search)
+          @klass.find_by(constraint_search) || @klass.new 
         rescue => e
           @klass.new
         end
@@ -74,14 +74,10 @@ module SeedFuMongoid
           document.send :[]=, key, value
         end
 
-        puts "#{@klass.name} #{document.attributes}" unless SeedFuMongoid.quiet
+        puts "#{@klass.name} #{document.attributes}"
 
         document.upsert
       end
-
-      def new?
-        !document.persisted?
-      end
     end
 
     def create_document(block_or_object)
@@ -98,20 +94,9 @@ module SeedFuMongoid
       end
     end
 
-    def seed_once!
-      if @objects.empty?
-        document = create_document(@block)
-        if document.new?
-          document.seed!
-        end
-      else
-        @objects.each do |object|
-          document = create_document(object)
-          if document.new?
-            document.seed!
-          end
-        end
-      end
+    def new?
+      !document.persisted?
     end
   end
 end
+