mongoid-pre 2.0.0.beta1

data/spec/models/country_code.rb

@@ -0,0 +1,6 @@
+class CountryCode
+  include Mongoid::Document
+  field :code, :type => Integer
+  key :code
+  embedded_in :phone_number, :inverse_of => :country_codes
+end

data/spec/models/employer.rb

@@ -0,0 +1,5 @@
+class Employer
+  def id
+    "1"
+  end
+end

data/spec/models/game.rb

@@ -0,0 +1,7 @@
+class Game
+  include Mongoid::Document
+  field :high_score, :default => 500
+  field :score, :type => Integer, :default => 0
+  belongs_to_related :person
+  enslave and cache
+end

data/spec/models/inheritance.rb

@@ -0,0 +1,56 @@
+class Canvas
+  include Mongoid::Document
+  field :name
+  embeds_many :shapes
+  embeds_one :writer
+
+  def render
+    shapes.each { |shape| render }
+  end
+end
+
+class Browser < Canvas
+  field :version, :type => Integer
+  def render; end
+end
+
+class Firefox < Browser
+  field :user_agent
+  def render; end
+end
+
+class Shape
+  include Mongoid::Document
+  field :x, :type => Integer, :default => 0
+  field :y, :type => Integer, :default => 0
+
+  embedded_in :canvas, :inverse_of => :shapes
+
+  def render; end
+end
+
+class Square < Shape
+  field :width, :type => Integer, :default => 0
+  field :height, :type => Integer, :default => 0
+end
+
+class Circle < Shape
+  field :radius, :type => Integer, :default => 0
+end
+
+class Writer
+  include Mongoid::Document
+  field :speed, :type => Integer, :default => 0
+
+  embedded_in :canvas, :inverse_of => :writer
+
+  def write; end
+end
+
+class HtmlWriter < Writer
+  def write; end
+end
+
+class PdfWriter < Writer
+  def write; end
+end

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
+  embeds_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
+    embeds_many :prescriptions, :class_name => "Medical::Prescription"
+  end
+  class Prescription
+    include Mongoid::Document
+    field :name
+  end
+end

data/spec/models/patient.rb

@@ -0,0 +1,6 @@
+class Patient
+  include Mongoid::Document
+  field :title
+  store_in :inpatient
+  embeds_many :addresses
+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
+
+  embeds_many :addresses do
+    def extension
+      "Testing"
+    end
+    def find_by_street(street)
+      @target.select { |doc| doc.street == street }
+    end
+  end
+
+  embeds_many :phone_numbers, :class_name => "Phone"
+
+  embeds_one :name do
+    def extension
+      "Testing"
+    end
+    def dawkins?
+      first_name == "Richard" && last_name == "Dawkins"
+    end
+  end
+
+  embeds_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
+  embeds_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
+  embeds_one :pet
+  embeds_one :address
+end

data/spec/models/phone.rb

@@ -0,0 +1,7 @@
+class Phone
+  include Mongoid::Document
+  field :number
+  key :number
+  embeds_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