ign-mongo_mapper 0.8.6.1

data/test/functional/test_document.rb

@@ -0,0 +1,272 @@
+require 'test_helper'
+require 'models'
+
+class DocumentTest < Test::Unit::TestCase
+  def setup
+    @document = Doc do
+      key :first_name, String
+      key :last_name, String
+      key :age, Integer
+      key :date, Date
+    end
+  end
+
+  context "array key" do
+    setup do
+      @document.key :tags, Array
+    end
+
+    should "give correct default" do
+      doc = @document.new
+      doc.tags.should == []
+    end
+
+    should "work with assignment" do
+      doc = @document.new
+      doc.tags = %w(foo bar)
+      doc.tags.should == %w(foo bar)
+    end
+
+    should "work with assignment after saving" do
+      doc = @document.new
+      doc.tags = %w(foo bar)
+      doc.save
+      doc.tags.should == %w(foo bar)
+      doc.reload.tags.should == %w(foo bar)
+    end
+
+    should "work with assignment then <<" do
+      doc = @document.new
+      doc.tags = []
+      doc.tags << "foo"
+      doc.tags.should == ["foo"]
+    end
+
+    should "work with <<" do
+      doc = @document.new
+      doc.tags << "foo"
+      doc.tags.should == ["foo"]
+    end
+
+    should "work with << then save" do
+      doc = @document.new
+      doc.tags << "foo"
+      doc.tags << "bar"
+      doc.save
+      doc.tags.should == %w(foo bar)
+      doc.reload.tags.should == %w(foo bar)
+    end
+  end
+
+  context "hash key" do
+    setup do
+      @document.key :foo, Hash
+    end
+
+    should "give correct default" do
+      doc = @document.new
+      doc.foo.should == {}
+    end
+
+    should "work with []=" do
+      doc = @document.new
+      doc.foo["quux"] = "bar"
+      doc.foo["quux"].should == "bar"
+      doc.foo.should == { "quux" => "bar" }
+    end
+
+    should "work with indifferent access" do
+      doc = @document.new
+      doc.foo = {:baz => 'bar'}
+      doc.foo[:baz].should == 'bar'
+      doc.foo['baz'].should == 'bar'
+    end
+
+    should "work with indifferent access after save" do
+      doc = @document.new
+      doc.foo = {:baz => 'bar'}
+      doc.save
+
+      doc = doc.reload
+      doc.foo[:baz].should == 'bar'
+      doc.foo['baz'].should == 'bar'
+    end
+  end
+
+  context "custom type key with default" do
+    setup do
+      @document.key :window, WindowSize, :default => WindowSize.new(600, 480)
+    end
+
+    should "default to default" do
+      doc = @document.new
+      doc.window.should == WindowSize.new(600, 480)
+
+    end
+
+    should "save and load from mongo" do
+      doc = @document.new
+      doc.save
+
+      doc = doc.reload
+      doc.window.should == WindowSize.new(600, 480)
+    end
+  end
+
+  context "key with proc default value" do
+    setup do
+      @document.key :proc_default, String, :default => lambda { return 'string' }
+    end
+
+    should "detect and run proc default" do
+      doc = @document.new
+      doc.proc_default.should == 'string'
+    end
+
+    should "save and load from mongo" do
+      doc = @document.create
+      doc = doc.reload
+      doc.proc_default.should == 'string'
+    end
+  end
+
+  should "have instance method for collection" do
+    @document.new.collection.name.should == @document.collection.name
+  end
+
+  should "have instance method for database" do
+    @document.new.database.should == @document.database
+  end
+
+  context "#destroyed?" do
+    setup do
+      @doc1 = @document.create(:first_name => 'John', :last_name => 'Nunemaker', :age => '27')
+    end
+
+    should "be true if deleted" do
+      @doc1.delete
+      assert @doc1.destroyed?
+    end
+
+    should "be true if destroyed" do
+      @doc1.destroy
+      assert @doc1.destroyed?
+    end
+
+    should "be false if not deleted or destroyed" do
+      assert ! @doc1.destroyed?
+    end
+  end
+
+  context "#persisted?" do
+    setup do
+      @doc = @document.new(:first_name => 'John', :last_name => 'Nunemaker', :age => '27')
+    end
+
+    should "be false if new" do
+      @doc.should_not be_persisted
+    end
+
+    should "be false if destroyed" do
+      @doc.save
+      @doc.destroy
+      @doc.should be_destroyed
+      @doc.should_not be_persisted
+    end
+
+    should "be true if not new or destroyed" do
+      @doc.save
+      @doc.should be_persisted
+    end
+  end
+
+  context "#reload" do
+    setup do
+      @foo_class = Doc do
+        key :name
+      end
+
+      @bar_class = EDoc do
+        key :name
+      end
+
+      @document.many :foos, :class => @foo_class
+      @document.many :bars, :class => @bar_class
+      @document.belongs_to :foo, :class => @foo_class
+      @document.one :bar, :class => @bar_class
+
+      @instance = @document.create({
+        :age  => 39,
+        :foos => [@foo_class.new(:name => '1')],
+        :bars => [@bar_class.new(:name => '1')],
+        :foo  => @foo_class.new(:name => '2'),
+        :bar  => @bar_class.new(:name => '2')
+      })
+    end
+
+    should "reload keys from the database" do
+      @instance.age = 37
+      @instance.age.should == 37
+      @instance.reload
+      @instance.age.should == 39
+    end
+
+    should "reset many associations" do
+      @instance.foos.expects(:reset).at_least_once
+      @instance.bars.expects(:reset).at_least_once
+      @instance.reload
+    end
+
+    should "reset belongs_to association" do
+      @instance.foo = nil
+      @instance.reload
+      @instance.foo.should_not be_nil
+    end
+
+    should "reset one association" do
+      @instance.bar = nil
+      @instance.reload
+      @instance.bar.should_not be_nil
+    end
+
+    should "reset nil one association" do
+    end
+
+    should "reinstantiate embedded associations" do
+      @instance.reload
+      @instance.bars.first.name.should == '1'
+    end
+
+    should "return self" do
+      @instance.reload.object_id.should == @instance.object_id
+    end
+
+    should "raise DocumentNotFound if not found" do
+      @instance.destroy
+      assert_raises(MongoMapper::DocumentNotFound) { @instance.reload }
+    end
+  end
+
+  context "database has keys not defined in model" do
+    setup do
+      @id = BSON::ObjectId.new
+      @document.collection.insert({
+        :_id            => @id,
+        :first_name     => 'John',
+        :last_name      => 'Nunemaker',
+        :age            => 27,
+        :favorite_color => 'red',
+        :skills         => ['ruby', 'rails', 'javascript', 'xhtml', 'css']
+      })
+    end
+
+    should "assign all keys from database" do
+      doc = @document.find(@id)
+      doc.first_name.should == 'John'
+      doc.last_name.should == 'Nunemaker'
+      doc.age.should == 27
+      doc.favorite_color.should == 'red'
+      doc.skills.should == ['ruby', 'rails', 'javascript', 'xhtml', 'css']
+    end
+  end
+end

data/test/functional/test_dynamic_querying.rb

@@ -0,0 +1,75 @@
+require 'test_helper'
+
+class DynamicQueryingTest < Test::Unit::TestCase
+  def setup
+    @document = Doc do
+      scope :nunes, where(:last_name => 'Nunemaker')
+
+      key :first_name, String
+      key :last_name, String
+      key :age, Integer
+      key :date, Date
+    end
+
+    @doc1 = @document.create(:first_name => 'John',   :last_name => 'Nunemaker', :age => 27)
+    @doc2 = @document.create(:first_name => 'Steve',  :last_name => 'Smith',     :age => 28)
+    @doc3 = @document.create(:first_name => 'Steph',  :last_name => 'Nunemaker', :age => 26)
+  end
+
+  should "find document based on argument" do
+    @document.find_by_first_name('John').should == @doc1
+    @document.find_by_last_name('Nunemaker', :order => 'age desc').should == @doc1
+    @document.find_by_age(27).should == @doc1
+  end
+
+  should "not raise error" do
+    @document.find_by_first_name('Mongo').should be_nil
+  end
+
+  should "define a method for each key" do
+    @document.methods(false).select { |e| e =~ /^find_by_/ }.size == @document.keys.size
+  end
+
+  should "find document based on all arguments" do
+    @document.find_by_first_name_and_last_name_and_age('John', 'Nunemaker', 27).should == @doc1
+  end
+
+  should "not find the document if an argument is wrong" do
+    @document.find_by_first_name_and_last_name_and_age('John', 'Nunemaker', 28).should be_nil
+  end
+
+  should "find all documents based on arguments" do
+    docs = @document.find_all_by_last_name('Nunemaker')
+    docs.should be_kind_of(Array)
+    docs.should include(@doc1)
+    docs.should include(@doc3)
+  end
+
+  should "initialize document with given arguments" do
+    doc = @document.find_or_initialize_by_first_name_and_last_name('David', 'Cuadrado')
+    doc.should be_new
+    doc.first_name.should == 'David'
+  end
+
+  should "not initialize document if document is found" do
+    doc = @document.find_or_initialize_by_first_name('John')
+    doc.should_not be_new
+  end
+
+  should "create document with given arguments" do
+    doc = @document.find_or_create_by_first_name_and_last_name('David', 'Cuadrado')
+    doc.should_not be_new
+    doc.first_name.should == 'David'
+  end
+
+  should "raise error if document is not found when using !" do
+    lambda {
+      @document.find_by_first_name_and_last_name!(1,2)
+    }.should raise_error(MongoMapper::DocumentNotFound)
+  end
+
+  should "work on scopes" do
+    @document.nunes.find_by_first_name('Steph').should == @doc3
+    @document.nunes.find_all_by_first_name('Steph').should == [@doc3]
+  end
+end

data/test/functional/test_embedded_document.rb

@@ -0,0 +1,210 @@
+require 'test_helper'
+require 'models'
+
+class EmbeddedDocumentTest < Test::Unit::TestCase
+  def setup
+    @klass = Doc('Person') do
+      key :name, String
+    end
+
+    @pet_klass = EDoc('Pet') do
+      key :name, String
+    end
+
+    @klass.many :pets, :class => @pet_klass
+
+    @address_class = EDoc('Address') do
+      key :city, String
+      key :state, String
+    end
+  end
+
+  context "Saving a document with a key that is an embedded document" do
+    setup do
+      @klass.key :foo, @address_class
+    end
+
+    should "embed embedded document" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.create(:foo => address)
+      doc.foo.city.should == 'South Bend'
+      doc.foo.state.should == 'IN'
+
+      doc = doc.reload
+      doc.foo.city.should == 'South Bend'
+      doc.foo.state.should == 'IN'
+    end
+
+    should "assign _parent_document and _root_document" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      address._parent_document.should be_nil
+      doc = @klass.create(:foo => address)
+      address._parent_document.should be(doc)
+      address._root_document.should be(doc)
+    end
+
+    should "assign _parent_document and _root_document when loading" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.create(:foo => address)
+      doc.reload
+      doc.foo._parent_document.should be(doc)
+      doc.foo._root_document.should be(doc)
+    end
+  end
+
+  should "correctly instantiate single collection inherited embedded documents" do
+    document = Doc('Foo') do
+      key :message, Message
+    end
+
+    doc1 = document.create(:message => Enter.new)
+    doc1.reload.message.class.should be(Enter)
+  end
+
+  context "new? (embedded key)" do
+    setup do
+      @klass.key :foo, @address_class
+    end
+
+    should "be true until document is saved" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.new(:foo => address)
+      address.new?.should be_true
+    end
+
+    should "be false after document is saved" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.new(:foo => address)
+      doc.save
+      doc.foo.new?.should be_false
+    end
+
+    should "be false when loaded from database" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.new(:foo => address)
+      doc.save
+
+      doc.reload
+      doc.foo.new?.should be_false
+    end
+  end
+
+  context "new? (embedded association)" do
+    setup do
+      @doc = @klass.new(:pets => [{:name => 'poo bear'}])
+    end
+
+    should "be true until document is saved" do
+      @doc.should be_new
+      @doc.pets.first.should be_new
+    end
+
+    should "be false after document is saved" do
+      @doc.save
+      @doc.pets.first.should_not be_new
+    end
+
+    should "be false when loaded from database" do
+      @doc.save
+      @doc.pets.first.should_not be_new
+      @doc.reload
+      @doc.pets.first.should_not be_new
+    end
+  end
+
+  context "#destroyed?" do
+    setup do
+      @doc = @klass.create(:pets => [@pet_klass.new(:name => 'sparky')])
+    end
+
+    should "be false if root document is not destroyed" do
+      @doc.should_not be_destroyed
+      @doc.pets.first.should_not be_destroyed
+    end
+
+    should "be true if root document is destroyed" do
+      @doc.destroy
+      @doc.should be_destroyed
+      @doc.pets.first.should be_destroyed
+    end
+  end
+
+  context "#persisted?" do
+    setup do
+      @doc = @klass.new(:name => 'persisted doc', :pets => [@pet_klass.new(:name => 'persisted pet')])
+    end
+
+    should "be false if new" do
+      @doc.pets.first.should_not be_persisted
+    end
+
+    should "be false if destroyed" do
+      @doc.save
+      @doc.destroy
+      @doc.pets.first.should be_destroyed
+      @doc.pets.first.should_not be_persisted
+    end
+
+    should "be true if not new or destroyed" do
+      @doc.save
+      @doc.pets.first.should be_persisted
+    end
+  end
+
+  should "be able to save" do
+    person = @klass.create
+
+    pet = @pet_klass.new(:name => 'sparky')
+    person.pets << pet
+    pet.should be_new
+    pet.save
+    pet.should_not be_new
+
+    person.reload
+    person.pets.first.should == pet
+  end
+
+  should "be able to dynamically add new keys and save" do
+    person = @klass.create
+
+    pet = @pet_klass.new(:name => 'sparky', :crazy_key => 'crazy')
+    person.pets << pet
+    pet.save
+
+    person.reload
+    person.pets.first.crazy_key.should == 'crazy'
+  end
+
+  should "be able to update_attributes" do
+    pet = @pet_klass.new(:name => 'sparky')
+    person = @klass.create(:pets => [pet])
+    person.reload
+    pet = person.pets.first
+
+    pet.update_attributes(:name => 'koda').should be_true
+    person.reload
+    person.pets.first._id.should == pet._id
+    person.pets.first.name.should == 'koda'
+  end
+
+  should "be able to update_attributes!" do
+    person = @klass.create(:pets => [@pet_klass.new(:name => 'sparky')])
+    person.reload
+    pet = person.pets.first
+
+    attributes = {:name => 'koda'}
+    pet.expects(:attributes=).with(attributes)
+    pet.expects(:save!)
+    pet.update_attributes!(attributes)
+  end
+
+  should "have database instance method that is equal to root document" do
+    person = @klass.create(:pets => [@pet_klass.new(:name => 'sparky')])
+    person.pets.first.database.should == person.database
+  end
+
+  should "have collection instance method that is equal to root document" do
+    person = @klass.create(:pets => [@pet_klass.new(:name => 'sparky')])
+    person.pets.first.collection.name.should == person.collection.name
+  end
+end