mongo-fixture 0.0.1 → 0.0.2

data/lib/mongo-fixture.rb

@@ -25,6 +25,7 @@ module Mongo
       load fixture if fixture
       
       @connection = connection if connection
+      @inserted = []
       push if fixture && connection && option_push
     end    
     
@@ -92,6 +93,8 @@ module Mongo
         matrix.each do |element, values|
           begin
             @connection[collection].insert simplify values.to_hash
+            @inserted ||= Array.new
+            @inserted << collection
           rescue MissingProcessedValueError => m
             rollback
             raise MissingProcessedValueError, "In record '#{element}' to be inserted into '#{collection}', the processed value of field '#{m.field}' is missing, aborting"
@@ -101,17 +104,32 @@ module Mongo
     end
     
     # Simplifies the hash in order to insert it into the database
-    # (Note: I'm well aware that this functionality belongs in a dependency,
-    # specially now that is repeated here and in Sequel::Fixture)
+    # Resolves external references and flattens the values that provide alternatives
     # @param [Hash] the hash to be processed 
     def simplify the_hash
       the_returned_hash = {}
       the_hash.each do |key, value|
         if value.is_a? Hash
-          unless value.has_key? :processed
-            raise MissingProcessedValueError.new "The processed value to insert into the db is missing from the field '#{key}', aborting", key 
+          
+          # If no alternative matches the name of a collection, look for a :processed value
+          if (value.keys & @data.keys).empty?
+            unless value.has_key? :processed
+              raise MissingProcessedValueError.new "The processed value to insert into the db is missing from the field '#{key}', aborting", key 
+            end
+            the_returned_hash[key] = value[:processed]
+          else
+          
+            # Does any of the options hold a record named after the value of the option?
+            options = value.keys & @data.keys
+            actual_option = options.each do |option|
+              break option if @data[option].has_key? value[option].to_sym
+            end
+            
+            unless data_was_inserted_in? actual_option
+              insert_data_for actual_option
+            end
+            the_returned_hash[key] = @connection[actual_option].find( @data[actual_option][value[actual_option].to_sym] ).first[:_id]
           end
-          the_returned_hash[key] = value[:processed]
         else
           the_returned_hash[key] = value
         end
@@ -119,6 +137,14 @@ module Mongo
       return the_returned_hash
     end
     
+    # Inserts the collection data into the database
+    def insert_data_for collection
+      @data[collection].each do |key, record|
+        @connection[collection].insert record
+      end
+      @inserted << collection
+    end
+    
     # Empties the collections, only if they were empty to begin with
     def rollback
       begin
@@ -144,6 +170,11 @@ module Mongo
         @field = field
         super message
       end
-    end
+    end    
+    
+    private
+      def data_was_inserted_in? collection
+        @inserted.include? collection
+      end
   end
 end

data/lib/mongo-fixture/version.rb

@@ -1,5 +1,5 @@
 module Mongo
   class Fixture
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/spec/mongo/fixture_spec.rb

@@ -1,4 +1,5 @@
 require "mongo-fixture"
+require "pry"
 
 describe Mongo::Fixture do
   describe ".path" do
@@ -398,7 +399,7 @@ describe Mongo::Fixture do
       end
     end
     
-    context "a fixture with a field with alternatives yet missing the <processed> one" do
+    context "a fixture with a field with alternatives missing the <processed> and the option doesn't match a collection" do
       before do
         Fast.file.write "test/fixtures/test/users.yaml", "hey: { pass: { raw: There } }"
       end
@@ -424,6 +425,110 @@ describe Mongo::Fixture do
         Fast.dir.remove! :test
       end
     end
+    
+    context "a fixture with a field with one alternative name matches a collection name" do
+      context "the alternative value matches a record and in the collection" do
+        before do 
+          Fast.file.write "test/fixtures/test/users.yaml", "pepe: { name: Jonah }"
+          Fast.file.write "test/fixtures/test/comments.yaml", "flamewar: { user: { users: pepe }, text: 'FLAME' }"
+        end
+        
+        it "should insert the comment so that the comment user value matches the '_id' of the user" do
+          database = double 'database'
+          comm = double 'comments', :count => 0, :drop => nil
+          comm.should_receive( :insert ).with( :user => "un id", :text => "FLAME" )
+          record = stub 'record'
+          record.should_receive( :[] ).with( :_id ).and_return "un id"
+          usrs = double 'users', :count => 0, :find => stub( :first => record ), :drop => nil, :insert => nil
+          database.stub :[] do |coll|
+            case coll
+              when :users
+                usrs
+              when :comments
+                comm
+            end
+          end
+          fix = Mongo::Fixture.new :test, database
+        end
+        
+        context "the collection is ordered so that the comment collection comes before the users one" do
+          it "should stop and process the users first" do
+            database = double 'database'
+            usrs = double 'users', :count => 0, :insert => nil, :find => stub( :first => stub( :[] => "un id" ) )
+            database.stub :[] do |argument|
+              case argument
+                when :comments
+                  double 'comments', :count => 0, :insert => nil
+                when :users
+                  usrs
+              end
+            end
+            fix = Mongo::Fixture.new :test, database, false
+            def fix.stub_data
+              @data = {
+                :comments => SymbolMatrix.new("test/fixtures/test/comments.yaml"),
+                :users => SymbolMatrix.new("test/fixtures/test/users.yaml") }
+            end
+            fix.stub_data
+            
+            fix.push
+          end
+        end
+        
+        after do
+          Fast.dir.remove! :test
+        end
+      end      
+    end
+  end
+  
+  describe "#data_was_inserted_in?" do
+    it "should be private" do
+      fix = Mongo::Fixture.new
+      fix.private_methods(false).should include :data_was_inserted_in?
+    end
+    
+    context "there is a simple fixture and a collection has been inserted by this fixture" do
+      before do
+        Fast.file.write "test/fixtures/test/users.yaml", "pepe: { user: pepe }"
+      end
+      
+      it "should return true" do
+        database = double 'database'
+        coll = double 'collection', :count => 0, :insert => nil
+        database.stub :[] => coll
+        fix = Mongo::Fixture.new :test, database
+        def fix.loot
+          data_was_inserted_in?(:users).should === true
+        end
+        fix.loot
+      end
+      
+      after do
+        Fast.dir.remove! :test
+      end
+    end
+    
+    context "there is a simple fixture and a collection was inserted but not this" do
+      before do
+        Fast.file.write "test/fixtures/test/users.yaml", "pepe: { user: pepe }"
+      end
+      
+      it "should return false" do
+        database = double 'database'
+        coll = double 'collection', :count => 0, :insert => nil
+        database.stub :[] => coll
+        fix = Mongo::Fixture.new :test, database
+        def fix.loot
+          data_was_inserted_in?(:comment).should === false
+        end
+        fix.loot
+      end
+      
+      after do
+        Fast.dir.remove! :test
+      end
+    end
   end
   
   # This should go in a dependency, pending refactoring TODO
@@ -444,6 +549,10 @@ describe Mongo::Fixture do
         }
         
         fix = Mongo::Fixture.new
+        def fix.stub_data
+          @data = {}
+        end
+        fix.stub_data
         simplified = fix.simplify base_hash
         simplified.should == {
           :name => "Jane",
@@ -469,6 +578,10 @@ describe Mongo::Fixture do
         }
         
         fix = Mongo::Fixture.new
+        def fix.stub_data
+          @data = {}
+        end
+        fix.stub_data
         expect { fix.simplify base_hash
         }.to raise_error Mongo::Fixture::MissingProcessedValueError, 
           "The processed value to insert into the db is missing from the field 'pass', aborting"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo-fixture
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-09 00:00:00.000000000 Z
+date: 2012-08-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mongo
@@ -114,7 +114,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 94604481
+      hash: 803460883
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -123,7 +123,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 94604481
+      hash: 803460883
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24