mongoid-pre 2.0.0.beta1

data/spec/unit/mongoid/associations_spec.rb

@@ -0,0 +1,595 @@
+require "spec_helper"
+
+describe Mongoid::Associations do
+
+  before do
+    @collection = stub(:name => "people")
+    @database = stub(:collection => @collection)
+    Mongoid.stubs(:database).returns(@database)
+  end
+
+  after do
+    Person.instance_variable_set(:@collection, nil)
+    @database = nil
+    @collection = nil
+  end
+
+  describe "#association=" do
+
+    context "when child is a has one" do
+
+      before do
+        @person = Person.new(:title => "Sir", :age => 30)
+        @name = Name.new(:first_name => "Test", :last_name => "User")
+        @person.name = @name
+      end
+
+      it "parentizes the association" do
+        @name._parent.should == @person
+      end
+
+      it "sets the child attributes on the parent" do
+        @person.attributes[:name].should ==
+          { "_id" => "test-user", "first_name" => "Test", "last_name" => "User", "_type" => "Name" }
+      end
+
+    end
+
+    context "when child and parent are namespaced" do
+
+      before do
+        @patient = Medical::Patient.new(:name => "Ridley")
+        @prescription = Medical::Prescription.new(:name => "Zoloft")
+        @patient.prescriptions << @prescription
+        @second = @patient.prescriptions.build(:name => "Codeine")
+      end
+
+      it "sets the correct association classes" do
+        @patient.prescriptions.first.should == @prescription
+        @patient.prescriptions.last.should == @second
+      end
+
+    end
+
+    context "when setting a parent" do
+
+      context "when the child is one level deep" do
+
+        before do
+          @person = Person.new(:title => "Mr")
+          @address = Address.new(:street => "Picadilly Circus")
+          @address.addressable = @person
+        end
+
+        it "re-parentizes the association" do
+          @address._parent.should == @person
+        end
+
+        it "adds the child attributes to the parent" do
+          @person.attributes[:addresses].should ==
+            [{ "_id" => "picadilly-circus", "street" => "Picadilly Circus", "_type" => "Address" }]
+        end
+
+      end
+
+      context "when the child is multiple levels deep" do
+
+        before do
+          @person = Person.new(:title => "Mr")
+          @phone = Phone.new(:number => "415-555-1212")
+          @person.phone_numbers = [@phone]
+          @country_code = CountryCode.new(:code => 1)
+          @phone.country_code = @country_code
+        end
+
+        it "properly decorates all parent references" do
+          @country_code.phone_number.should == @phone
+          @phone.person.should == @person
+          @country_code.phone_number.person.should == @person
+        end
+
+      end
+
+    end
+
+  end
+
+  describe "#associations" do
+
+    before do
+      @person = Person.new
+    end
+
+    it "is a hash with name keys and meta data values" do
+      @person.associations["addresses"].should be_a_kind_of(Mongoid::Associations::MetaData)
+    end
+  end
+
+  describe ".embedded_in" do
+
+    it "creates a reader for the association" do
+      address = Address.new
+      address.should respond_to(:addressable)
+    end
+
+    it "creates a writer for the association" do
+      address = Address.new
+      address.should respond_to(:addressable=)
+    end
+
+    it "allows the parent to be any type of class" do
+      phone_number = Phone.new(:number => "415-555-1212")
+      code = CountryCode.new(:code => 1)
+      phone_number.country_code = code
+      code.phone_number.should == phone_number
+    end
+
+    context "when adding an anonymous extension" do
+
+      before do
+        @person = Person.new(:title => "Dr")
+        @address = Address.new(:street => "Clarkenwell Road")
+        @person.addresses << @address
+      end
+
+      it "defines the method on the association" do
+        @address.addressable.extension.should == "Testing"
+      end
+
+    end
+
+    context "when inverse_of not supplied" do
+
+      it "raises an error" do
+        lambda { Person.class_eval { embedded_in :nothing } }.should raise_error
+      end
+
+    end
+
+    context "when navigating the graph" do
+
+      before do
+        @person = Person.new(:title => "Mr")
+        @name = Name.new(:first_name => "Mason")
+        @address = Address.new(:street => "King St.")
+        @person.name = @name
+        @person.addresses << @address
+      end
+
+      it "allows referencing another child through the parent" do
+        @name.person.addresses.first.should == @address
+      end
+
+    end
+
+  end
+
+  describe "#build_*" do
+
+    before do
+      @canvas = Canvas.new
+    end
+
+    context "when type is passed in" do
+
+      before do
+        @writer = @canvas.build_writer(:speed => 250, :_type => "HtmlWriter")
+      end
+
+      it "returns a new document" do
+        @writer.should_not be_nil
+      end
+
+      it "returns the properly typed document" do
+        @writer.should be_a_kind_of(HtmlWriter)
+      end
+
+      it "sets the appropriate attributes" do
+        @writer.speed.should == 250
+      end
+
+    end
+
+    context "when type is not passed in" do
+
+      before do
+        @writer = @canvas.build_writer(:speed => 250)
+      end
+
+      it "returns a new document" do
+        @writer.should_not be_nil
+      end
+
+      it "returns the properly typed document" do
+        @writer.should be_a_kind_of(Writer)
+      end
+
+      it "sets the appropriate attributes" do
+        @writer.speed.should == 250
+      end
+
+    end
+
+    context "when attributes are nil" do
+
+      before do
+        @writer = @canvas.build_writer(nil)
+      end
+
+      it "defaults them to empty" do
+        @writer.should be_a_kind_of(Writer)
+      end
+
+    end
+
+  end
+
+  describe "#create_*" do
+
+    before do
+      @canvas = Canvas.new
+    end
+
+    context "when type is passed in" do
+
+      before do
+        Mongoid::Commands::Save.expects(:execute)
+        @writer = @canvas.create_writer(:speed => 250, :_type => "HtmlWriter")
+      end
+
+      it "returns a new document" do
+        @writer.should_not be_nil
+      end
+
+      it "returns the properly typed document" do
+        @writer.should be_a_kind_of(HtmlWriter)
+      end
+
+      it "sets the appropriate attributes" do
+        @writer.speed.should == 250
+      end
+
+    end
+
+    context "when type is not passed in" do
+
+      before do
+        Mongoid::Commands::Save.expects(:execute)
+        @writer = @canvas.create_writer(:speed => 250, :_type => "HtmlWriter")
+      end
+
+      it "returns a new document" do
+        @writer.should_not be_nil
+      end
+
+      it "returns the properly typed document" do
+        @writer.should be_a_kind_of(Writer)
+      end
+
+      it "sets the appropriate attributes" do
+        @writer.speed.should == 250
+      end
+
+    end
+
+  end
+
+  describe ".embeds_many" do
+
+    it "adds a new Association to the collection" do
+      person = Person.new
+      person.addresses.should_not be_nil
+    end
+
+    it "creates a reader for the association" do
+      person = Person.new
+      person.should respond_to(:addresses)
+    end
+
+    it "creates a writer for the association" do
+      person = Person.new
+      person.should respond_to(:addresses=)
+    end
+
+    context "when adding an anonymous extension" do
+
+      it "defines the method on the association" do
+        person = Person.new
+        person.addresses.extension.should == "Testing"
+      end
+
+    end
+
+    context "when setting the association directly" do
+
+      before do
+        @attributes = { :title => "Sir",
+          :addresses => [
+            { :street => "Street 1" },
+            { :street => "Street 2" } ] }
+        @person = Person.new(@attributes)
+      end
+
+      it "sets the attributes for the association" do
+        address = Address.new(:street => "New Street")
+        @person.addresses = [address]
+        @person.addresses.first.street.should == "New Street"
+      end
+
+    end
+
+    context "when a class_name is supplied" do
+
+      before do
+        @attributes = { :title => "Sir",
+          :phone_numbers => [ { :number => "404-555-1212" } ]
+        }
+        @person = Person.new(@attributes)
+      end
+
+      it "sets the association name" do
+        @person.phone_numbers.first.should == Phone.new(:number => "404-555-1212")
+      end
+
+    end
+
+    context "when updating objects internally" do
+
+      before do
+        @address = Address.new(:street => "Bourke Street")
+        @person = Person.new(:title => "Sir")
+        @person.addresses << @address
+        @person.update_addresses
+      end
+
+      it "retains its references to the original objects" do
+        @address.street.should == "Updated Address"
+      end
+
+    end
+
+  end
+
+  describe ".embeds_one" do
+
+    before do
+      @person = Person.new
+    end
+
+    it "adds a new Association to the document" do
+      @person.name.should be_nil
+    end
+
+    it "creates a reader for the association" do
+      @person.should respond_to(:name)
+    end
+
+    it "creates a writer for the association" do
+      @person.should respond_to(:name=)
+    end
+
+    it "creates a builder for the association" do
+      @person.should respond_to(:build_name)
+    end
+
+    it "creates a creator for the association" do
+      @person.should respond_to(:create_name)
+    end
+
+    context "when adding an anonymous extension" do
+
+      it "defines the method on the association" do
+        @person.name = Name.new(:first_name => "Richard")
+        @person.name.extension.should == "Testing"
+      end
+
+    end
+
+    context "when setting the association directly" do
+
+      before do
+        @attributes = { :title => "Sir",
+          :name => { :first_name => "Test" } }
+        @person = Person.new(@attributes)
+      end
+
+      it "sets the attributes for the association" do
+        name = Name.new(:first_name => "New Name")
+        @person.name = name
+        @person.name.first_name.should == "New Name"
+      end
+
+    end
+
+    context "when a class_name is supplied" do
+
+      before do
+        @attributes = { :title => "Sir",
+          :pet => { :name => "Fido" }
+        }
+        @person = Person.new(@attributes)
+      end
+
+      it "sets the association name" do
+        @person.pet.should == Animal.new(:name => "Fido")
+      end
+
+    end
+
+  end
+
+  describe ".reflect_on_association" do
+
+    it "returns the association class for the name" do
+      Person.reflect_on_association(:addresses).should == :embeds_many
+    end
+
+  end
+
+  describe ".belongs_to_related" do
+
+    before do
+      @game = Game.new
+    end
+
+    it "creates an id field for the relationship" do
+      @game.should respond_to(:person_id)
+    end
+
+    it "creates a getter for the parent" do
+      @game.should respond_to(:person)
+    end
+
+    it "defaults the foreign_key option to the name_id" do
+      @game.associations["person"].foreign_key.should == "person_id"
+    end
+
+    context "when document is root level" do
+
+      it "puts an index on the foreign key" do
+        Game.expects(:index).with("person_id")
+        Game.belongs_to_related :person
+      end
+
+    end
+
+    context "when using object ids" do
+
+      before do
+        Mongoid.use_object_ids = true
+      end
+
+      after do
+        Mongoid.use_object_ids = false
+      end
+
+      it "sets the foreign key as an object id" do
+        Game.expects(:field).with("person_id", :type => Mongo::ObjectID)
+        Game.belongs_to_related :person
+      end
+    end
+
+  end
+
+  describe ".has_one_related" do
+
+    before do
+      @person = Person.new
+      @game = Game.new
+      @person.game = @game
+    end
+
+    it "creates a getter for the relationship" do
+      @person.should respond_to(:game)
+    end
+
+    it "creates a setter for the relationship" do
+      @person.should respond_to(:game=)
+    end
+
+    context "when adding an anonymous extension" do
+
+      it "defines the method on the association" do
+        @person.game.extension.should == "Testing"
+      end
+
+    end
+
+  end
+
+  describe ".has_many_related" do
+
+    it "creates a getter for the association" do
+      Person.new.should respond_to(:posts)
+    end
+
+    it "creates a setter for the association" do
+      Person.new.should respond_to(:posts=)
+    end
+
+    context "when adding an anonymous extension" do
+
+      before do
+        @person = Person.new
+      end
+
+      it "defines the method on the association" do
+        @person.posts.extension.should == "Testing"
+      end
+
+    end
+
+  end
+
+  describe "#update_associations" do
+
+    context "when associations exist" do
+
+      context "when the document is a new record" do
+
+        before do
+          @related = stub(:id => "100", :person= => true)
+          @person = Person.new
+          @person.posts = [@related]
+        end
+
+        it "saves each association" do
+          @related.expects(:save).returns(@related)
+          @person.update_associations(:posts)
+        end
+
+      end
+
+      context "when the document is not new" do
+
+        before do
+          @related = stub(:id => "100", :person= => true)
+          @person = Person.new
+          @person.instance_variable_set(:@new_record, false)
+          @person.posts = [@related]
+        end
+
+        it "does not save each association" do
+          @person.update_associations(:posts)
+        end
+
+      end
+
+    end
+
+    context "when no associations exist" do
+
+      before do
+        @person = Person.new
+      end
+
+      it "does nothing" do
+        Post.expects(:find).returns([])
+        @person.update_associations(:posts)
+        @person.posts.first.should be_nil
+      end
+
+    end
+
+  end
+
+  describe "#update_association" do
+
+    context "when the association exists" do
+
+      before do
+        @related = stub(:id => "100", :person= => true)
+        @person = Person.new
+        @person.game = @related
+      end
+
+      it "saves each association" do
+        @related.expects(:save).returns(@related)
+        @person.update_association(:game)
+      end
+
+    end
+
+  end
+
+end