mongoid 1.9.5 → 2.0.0.alpha

data/spec/models/location.rb

@@ -0,0 +1,5 @@
+class Location
+  include Mongoid::Document
+  field :name
+  embedded_in :address, :inverse_of => :locations
+end

data/spec/models/mixed_drink.rb

@@ -0,0 +1,4 @@
+class MixedDrink
+  include Mongoid::Document
+  field :name
+end

data/spec/models/name.rb

@@ -0,0 +1,13 @@
+class Name
+  include Mongoid::Document
+  field :first_name
+  field :last_name
+  field :parent_title
+  key :first_name, :last_name
+  embed_many :translations
+  embedded_in :person, :inverse_of => :name
+
+  def set_parent=(set = false)
+    self.parent_title = person.title if set
+  end
+end

data/spec/models/namespacing.rb

@@ -0,0 +1,11 @@
+module Medical
+  class Patient
+    include Mongoid::Document
+    field :name
+    embed_many :prescriptions, :class_name => "Medical::Prescription"
+  end
+  class Prescription
+    include Mongoid::Document
+    field :name
+  end
+end

data/spec/models/patient.rb

@@ -0,0 +1,4 @@
+class Patient
+  include Mongoid::Document
+  store_in :inpatient
+end

data/spec/models/person.rb

@@ -0,0 +1,99 @@
+class Person
+  include Mongoid::Document
+  include Mongoid::Timestamps
+
+  field :title
+  field :terms, :type => Boolean
+  field :pets, :type => Boolean, :default => false
+  field :age, :type => Integer, :default => 100
+  field :dob, :type => Date
+  field :mixed_drink, :type => MixedDrink
+  field :employer_id
+  field :lunch_time, :type => Time
+  field :aliases, :type => Array
+  field :map, :type => Hash
+  field :score, :type => Integer
+  field :blood_alcohol_content, :type => Float, :default => lambda{ 0.0 }
+  field :ssn
+  field :owner_id, :accessible => false, :type => Integer
+
+  index :age
+  index :addresses
+  index :dob
+  index :name
+  index :title
+  index :ssn, :unique => true
+
+  attr_reader :rescored
+
+  embed_many :addresses do
+    def extension
+      "Testing"
+    end
+    def find_by_street(street)
+      @target.select { |doc| doc.street == street }
+    end
+  end
+
+  embed_many :phone_numbers, :class_name => "Phone"
+
+  embed_one :name do
+    def extension
+      "Testing"
+    end
+    def dawkins?
+      first_name == "Richard" && last_name == "Dawkins"
+    end
+  end
+
+  embed_one :pet, :class_name => "Animal"
+
+  accepts_nested_attributes_for :addresses, :reject_if => lambda { |attrs| attrs["street"].blank? }
+  accepts_nested_attributes_for :name
+
+  has_one_related :game do
+    def extension
+      "Testing"
+    end
+  end
+
+  has_many_related :posts do
+    def extension
+      "Testing"
+    end
+  end
+
+  def score_with_rescoring=(score)
+    @rescored = score.to_i + 20
+    self.score_without_rescoring = score
+  end
+
+  alias_method_chain :score=, :rescoring
+
+  def update_addresses
+    addresses.each do |address|
+      address.street = "Updated Address"
+    end
+  end
+
+  def employer=(emp)
+    self.employer_id = emp.id
+  end
+
+  class << self
+    def accepted
+      criteria.where(:terms => true)
+    end
+    def knight
+      criteria.where(:title => "Sir")
+    end
+    def old
+      criteria.where(:age => { "$gt" => 50 })
+    end
+  end
+
+end
+
+class Doctor < Person
+  field :specialty
+end

data/spec/models/pet.rb

@@ -0,0 +1,7 @@
+class Pet
+  include Mongoid::Document
+  field :name
+  field :weight, :type => Float, :default => 0.0
+  embed_many :vet_visits
+  embedded_in :pet_owner, :inverse_of => :pet
+end

data/spec/models/pet_owner.rb

@@ -0,0 +1,6 @@
+class PetOwner
+  include Mongoid::Document
+  field :title
+  embed_one :pet
+  embed_one :address
+end

data/spec/models/phone.rb

@@ -0,0 +1,7 @@
+class Phone
+  include Mongoid::Document
+  field :number
+  key :number
+  embed_one :country_code
+  embedded_in :person, :inverse_of => :phone_numbers
+end

data/spec/models/post.rb

@@ -0,0 +1,15 @@
+class Post
+  include Mongoid::Document
+  include Mongoid::Versioning
+  include Mongoid::Timestamps
+  field :title
+  belongs_to_related :person
+
+  named_scope :recent, where(:created_at => { "$lt" => Time.now, "$gt" => 30.days.ago })
+
+  class << self
+    def old
+      where(:created_at => { "$lt" => 30.days.ago })
+    end
+  end
+end

data/spec/models/translation.rb

@@ -0,0 +1,5 @@
+class Translation
+  include Mongoid::Document
+  field :language
+  embedded_in :name, :inverse_of => :translations
+end

data/spec/models/vet_visit.rb

@@ -0,0 +1,5 @@
+class VetVisit
+  include Mongoid::Document
+  field :date, :type => Date
+  embedded_in :pet, :inverse_of => :vet_visits
+end

data/spec/spec.opts

@@ -0,0 +1,3 @@
+--colour
+--format nested
+--drb

data/spec/spec_helper.rb

@@ -0,0 +1,31 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
+
+MODELS = File.join(File.dirname(__FILE__), "models")
+$LOAD_PATH.unshift(MODELS)
+
+require 'rubygems'
+
+gem "mocha", ">= 0.9.8"
+
+require "mongoid"
+require "mocha"
+require "spec"
+
+Mongoid.configure do |config|
+  name = "mongoid_test"
+  host = "localhost"
+  config.master = Mongo::Connection.new.db(name)
+  # config.slaves = [
+    # Mongo::Connection.new(host, 27018, :slave_ok => true).db(name)
+  # ]
+end
+
+Dir[ File.join(MODELS, "*.rb") ].sort.each { |file| require File.basename(file) }
+
+Spec::Runner.configure do |config|
+  config.mock_with :mocha
+  config.after :suite do
+    Mongoid.master.collections.each(&:drop)
+  end
+end

data/spec/unit/mongoid/associations/belongs_to_related_spec.rb

@@ -0,0 +1,145 @@
+require "spec_helper"
+
+describe Mongoid::Associations::BelongsToRelated do
+
+  describe ".initialize" do
+
+    context "when related id has been set" do
+
+      before do
+        @document = stub(:person_id => "5")
+        @options = Mongoid::Associations::Options.new(:name => :person)
+        @related = stub
+      end
+
+      it "finds the object by id" do
+        Person.expects(:find).with(@document.person_id).returns(@related)
+        association = Mongoid::Associations::BelongsToRelated.new(@document, "5", @options)
+        association.should == @related
+      end
+
+    end
+
+    context "when options have an extension" do
+
+      before do
+        @document = stub(:person_id => "5")
+        @block = Proc.new {
+          def extension
+            "Testing"
+          end
+        }
+        @options = Mongoid::Associations::Options.new(:name => :person, :extend => @block)
+        @related = stub
+        Person.expects(:find).with(@document.person_id).returns(@related)
+        @association = Mongoid::Associations::BelongsToRelated.new(@document, "5", @options)
+      end
+
+      it "adds the extension module" do
+        @association.extension.should == "Testing"
+      end
+
+    end
+
+  end
+
+  describe ".instantiate" do
+
+    context "when foreign key is not nil" do
+
+      before do
+        @document = stub(:person_id => "5")
+        @options = Mongoid::Associations::Options.new(:name => :person)
+      end
+
+      it "delegates to new" do
+        Mongoid::Associations::BelongsToRelated.expects(:new).with(@document, "5", @options, nil)
+        Mongoid::Associations::BelongsToRelated.instantiate(@document, @options)
+      end
+
+    end
+
+    context "when foreign key is nil" do
+
+      before do
+        @document = stub(:person_id => nil)
+        @options = Mongoid::Associations::Options.new(:name => :person)
+      end
+
+      it "returns nil" do
+        Mongoid::Associations::BelongsToRelated.instantiate(@document, @options).should be_nil
+      end
+
+    end
+
+  end
+
+  describe "#method_missing" do
+
+    before do
+      @person = Person.new(:title => "Mr")
+      @document = stub(:person_id => "5")
+      @options = Mongoid::Associations::Options.new(:name => :person)
+      Person.expects(:find).with(@document.person_id).returns(@person)
+      @association = Mongoid::Associations::BelongsToRelated.new(@document, "5", @options)
+    end
+
+    context "when getting values" do
+
+      it "delegates to the document" do
+        @association.title.should == "Mr"
+      end
+
+    end
+
+    context "when setting values" do
+
+      it "delegates to the document" do
+        @association.title = "Sir"
+        @association.title.should == "Sir"
+      end
+
+    end
+
+  end
+
+  describe ".macro" do
+
+    it "returns :belongs_to_related" do
+      Mongoid::Associations::BelongsToRelated.macro.should == :belongs_to_related
+    end
+
+  end
+
+  describe ".update" do
+
+    before do
+      @related = stub(:id => "5")
+      @child = Game.new
+      @options = Mongoid::Associations::Options.new(:name => :person)
+      @association = Mongoid::Associations::BelongsToRelated.update(@related, @child, @options)
+    end
+
+    it "sets the related object id on the parent" do
+      @child.person_id.should == "5"
+    end
+
+    it "returns the proxy" do
+      @association.target.should == @related
+    end
+
+    context "when target is nil" do
+
+      it "returns nil" do
+        Mongoid::Associations::BelongsToRelated.update(nil, @child, @options).should be_nil
+      end
+
+      it "removes the association" do
+        Mongoid::Associations::BelongsToRelated.update(nil, @child, @options)
+        @child.person.should be_nil
+      end
+    end
+
+  end
+
+end

data/spec/unit/mongoid/associations/embed_many_spec.rb

@@ -0,0 +1,516 @@
+require "spec_helper"
+
+describe Mongoid::Associations::EmbedMany do
+
+  before do
+    @attributes = { "addresses" => [
+      { "_id" => "street-1", "street" => "Street 1", "state" => "CA" },
+      { "_id" => "street-2", "street" => "Street 2" } ] }
+    @document = stub(:raw_attributes => @attributes, :add_observer => true, :update => true)
+  end
+
+  describe "#[]" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+    end
+
+    context "when the index is present in the association" do
+
+      it "returns the document at the index" do
+        @association[0].should be_a_kind_of(Address)
+        @association[0].street.should == "Street 1"
+      end
+
+    end
+
+    context "when the index is not present in the association" do
+
+      it "returns nil" do
+        @association[3].should be_nil
+      end
+
+    end
+
+  end
+
+  describe "#<<" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+      @address = Address.new
+    end
+
+    it "adds the parent document before appending to the array" do
+      @association << @address
+      @association.length.should == 3
+      @address._parent.should == @document
+    end
+
+    it "allows multiple additions" do
+      @association << @address
+      @association << @address
+      @association.length.should == 4
+    end
+
+  end
+
+  describe "#build" do
+
+    context "setting the parent relationship" do
+
+      before do
+        @person = Person.new
+      end
+
+      it "happens before any other operation" do
+        address = @person.addresses.build(:set_parent => true, :street => "Madison Ave")
+        address._parent.should == @person
+        @person.addresses.first.should == address
+      end
+
+    end
+
+    context "when a type is not provided" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+      end
+
+      it "adds a new document to the array with the suppied parameters" do
+        @association.build({ :street => "Street 1" })
+        @association.length.should == 3
+        @association[2].should be_a_kind_of(Address)
+        @association[2].street.should == "Street 1"
+      end
+
+      it "returns the newly built object in the association" do
+        address = @association.build({ :street => "Yet Another" })
+        address.should be_a_kind_of(Address)
+        address.street.should == "Yet Another"
+      end
+
+    end
+
+    context "when a type is provided" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :shapes)
+        )
+      end
+
+      it "instantiates a class of the type" do
+        circle = @association.build({ :radius => 100 }, Circle)
+        circle.should be_a_kind_of(Circle)
+        circle.radius.should == 100
+      end
+
+    end
+
+  end
+
+  describe "#create" do
+
+    context "when a type is not provided" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+        @address = mock(:parentize => true, :write_attributes => true)
+        Address.expects(:instantiate).returns(@address)
+        @address.expects(:run_callbacks).with(:create).yields
+      end
+
+      it "builds and saves a new object" do
+        @address.expects(:save).returns(true)
+        address = @association.create({ :street => "Yet Another" })
+        address.should == @address
+      end
+
+    end
+
+    context "when a type is provided" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :shapes)
+        )
+        @circle = mock(:parentize => true, :write_attributes => true)
+        Circle.expects(:instantiate).returns(@circle)
+        @circle.expects(:run_callbacks).with(:create).yields
+      end
+
+      it "instantiates a class of that type" do
+        @circle.expects(:save).returns(true)
+        circle = @association.create({ :radius => 100 }, Circle)
+        circle.should == @circle
+      end
+
+    end
+
+  end
+
+  describe "#create!" do
+
+    context "when validations pass" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+        @address = mock(:parentize => true, :write_attributes => true, :errors => [])
+        Address.expects(:instantiate).returns(@address)
+        @address.expects(:run_callbacks).with(:create).yields
+      end
+
+      it "builds and saves a new object" do
+        @address.expects(:save).returns(true)
+        address = @association.create!({ :street => "Yet Another" })
+        address.should == @address
+      end
+    end
+
+    context "when validations fail" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+        @address = mock(:parentize => true, :write_attributes => true, :errors => [ "test" ])
+        Address.expects(:instantiate).returns(@address)
+        @address.expects(:run_callbacks).with(:create).yields
+      end
+
+      it "builds and saves a new object" do
+        @address.expects(:save).returns(false)
+        lambda {
+          @association.create!({ :street => "Yet Another" })
+        }.should raise_error(Mongoid::Errors::Validations)
+      end
+    end
+  end
+
+  describe "#concat" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+      @address = Address.new
+    end
+
+    it "adds the parent document before appending to the array" do
+      @association.concat [@address]
+      @association.length.should == 3
+      @address._parent.should == @document
+    end
+
+  end
+
+  describe "#clear" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+      @address = Address.new
+      @association << @address
+    end
+
+    it "clears out the association" do
+      @association.clear
+      @association.size.should == 0
+    end
+
+  end
+
+  describe "#find" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+    end
+
+    context "when finding all" do
+
+      it "returns all the documents" do
+        @association.find(:all).should == @association
+      end
+
+    end
+
+    context "when finding by id" do
+
+      it "returns the document in the array with that id" do
+        address = @association.find("street-2")
+        address.should_not be_nil
+      end
+
+    end
+
+  end
+
+  describe "#first" do
+
+    context "when there are elements in the array" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+      end
+
+      it "returns the first element" do
+        @association.first.should be_a_kind_of(Address)
+        @association.first.street.should == "Street 1"
+      end
+
+    end
+
+    context "when the array is empty" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          Person.new,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+      end
+
+      it "returns nil" do
+        @association.first.should be_nil
+      end
+
+    end
+
+  end
+
+  describe "#initialize" do
+
+    context "when no extension exists" do
+
+      before do
+        @canvas = stub(:raw_attributes => { "shapes" => [{ "_type" => "Circle", "radius" => 5 }] }, :update => true)
+        @association = Mongoid::Associations::EmbedMany.new(
+          @canvas,
+          Mongoid::Associations::Options.new(:name => :shapes)
+        )
+      end
+
+      it "creates the classes based on their types" do
+        circle = @association.first
+        circle.should be_a_kind_of(Circle)
+        circle.radius.should == 5
+      end
+
+    end
+
+    context "when an extension is in the options" do
+
+      before do
+        @person = Person.new
+        @block = Proc.new do
+          def extension
+            "Testing"
+          end
+        end
+        @association = Mongoid::Associations::EmbedMany.new(
+          @person,
+          Mongoid::Associations::Options.new(:name => :addresses, :extend => @block)
+        )
+      end
+
+      it "adds the extension module" do
+        @association.extension.should == "Testing"
+      end
+
+    end
+
+  end
+
+  describe ".instantiate" do
+
+    it "delegates to new" do
+      Mongoid::Associations::EmbedMany.expects(:new).with(@document, @options)
+      Mongoid::Associations::EmbedMany.instantiate(@document, @options)
+    end
+
+  end
+
+  describe "#length" do
+
+    context "#length" do
+
+      it "returns the length of the delegated array" do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+        @association.length.should == 2
+      end
+
+    end
+
+  end
+
+  describe ".macro" do
+
+    it "returns :embed_many" do
+      Mongoid::Associations::EmbedMany.macro.should == :embed_many
+    end
+
+  end
+
+  describe "#nested_build" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+    end
+
+    it "returns the newly built object in the association" do
+      @association.nested_build({ "0" => { :street => "Yet Another" } })
+      @association.size.should == 3
+      @association.last.street.should == "Yet Another"
+    end
+
+  end
+
+  describe "#method_missing" do
+
+    context "when the association class has a criteria class method" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+      end
+
+      it "returns the criteria" do
+        @association.california.should be_a_kind_of(Mongoid::Criteria)
+      end
+
+      it "sets the documents on the criteria" do
+        criteria = @association.california
+        criteria.documents.should == @association.entries
+      end
+
+      it "returns the scoped documents" do
+        addresses = @association.california
+        addresses.size.should == 1
+        addresses.first.should be_a_kind_of(Address)
+        addresses.first.state.should == "CA"
+      end
+
+    end
+
+    context "when no class method exists" do
+
+      before do
+        @association = Mongoid::Associations::EmbedMany.new(
+          @document,
+          Mongoid::Associations::Options.new(:name => :addresses)
+        )
+      end
+
+      it "delegates to the array" do
+        @association.entries.size.should == 2
+      end
+
+    end
+
+  end
+
+  describe "#paginate" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+      @options = { :page => 1, :per_page => 10 }
+      @criteria = mock
+    end
+
+    it "creates a criteria and paginates it" do
+      Mongoid::Criteria.expects(:translate).with(Address, @options).returns(@criteria)
+      @criteria.expects(:documents=).with(@association.target)
+      @criteria.expects(:paginate).returns([])
+      @association.paginate(@options).should == []
+    end
+  end
+
+  describe "#push" do
+
+    before do
+      @association = Mongoid::Associations::EmbedMany.new(
+        @document,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+      @address = Address.new
+    end
+
+    it "adds the parent document before appending to the array" do
+      @association.push @address
+      @association.length.should == 3
+      @address._parent.should == @document
+    end
+
+    it "appends the document to the end of the array" do
+      @association.push(Address.new)
+      @association.length.should == 3
+    end
+
+  end
+
+  describe ".update" do
+
+    before do
+      @address = Address.new(:street => "Madison Ave")
+      @person = Person.new(:title => "Sir")
+      @association = Mongoid::Associations::EmbedMany.update(
+        [@address],
+        @person,
+        Mongoid::Associations::Options.new(:name => :addresses)
+      )
+    end
+
+    it "parentizes the child document" do
+      @address._parent.should == @person
+    end
+
+    it "sets the attributes of the child on the parent" do
+      @person.attributes[:addresses].should ==
+        [{ "_id" => "madison-ave", "street" => "Madison Ave", "_type" => "Address" }]
+    end
+
+    it "returns the association proxy" do
+      @association.target.size.should == 1
+    end
+
+  end
+
+end