require 'test_helper'
require 'models'

class EmbeddedDocumentTest < Test::Unit::TestCase
  def setup
    @document = Class.new do
      include MongoMapper::Document
      set_collection_name 'users'

      key :first_name, String
      key :last_name, String
    end
    @document.collection.remove
  end
    
  context "Saving a document with an embedded document" do
    setup do
      @document.class_eval do
        key :foo, Address
      end
      
      @address = Address.new(:city => 'South Bend', :state => 'IN')
      @doc = @document.new(:foo => @address)
    end
    
    should "embed embedded document" do
      @doc.save
      @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
  end
  
  context "Instantiating single collection inherited embedded documents" do
    setup do
      @document = Class.new do
        include MongoMapper::Document
        key :message, Message
      end
    end

    should "work" do
      doc1 = @document.create(:message => Enter.new)
      doc2 = @document.create(:message => Exit.new)
      doc3 = @document.create(:message => Chat.new)

      doc1.reload.message.class.should be(Enter)
      doc2.reload.message.class.should be(Exit)
      doc3.reload.message.class.should be(Chat)
    end
  end
  
  context "new?" do
    setup do
      @document.class_eval do
        key :foo, Address
      end
    end

    should "be new until document is saved" do
      address = Address.new(:city => 'South Bend', :state => 'IN')
      doc = @document.new(:foo => address)
      address.new?.should == true
    end
    
    should "not be new after document is saved" do
      address = Address.new(:city => 'South Bend', :state => 'IN')
      doc = @document.new(:foo => address)
      doc.save
      doc.foo.new?.should == false
    end
    
    should "not be new when document is read back" do
      address = Address.new(:city => 'South Bend', :state => 'IN')
      doc = @document.new(:foo => address)
      doc.save
      
      doc = doc.reload
      doc.foo.new?.should == false
    end
  end
  
  context "save" do
    should "save the root document" do
      person = RealPerson.create

      pet = Pet.new :name => 'sparky'
      person.pets << pet
      pet.save

      person = person.reload
      person.pets.first.should == pet
    end
    
    should "save new keys" do
      person = RealPerson.new
      person[:new_attribute] = 'foobar'
      person.save
      
      person = person.reload
      person.new_attribute.should == 'foobar'
    end
  end
  
  context "update_attributes" do
    should "save the root document" do
      person = RealPerson.create

      pet = Pet.new(:name => 'sparky')
      person.pets << pet
      pet.save

      person = person.reload
      pet = person.pets.first
      pet.update_attributes :name => 'koda'

      person = person.reload
      person.pets.first._id.should == pet._id
      person.pets.first.name.should == 'koda'
    end
  end  

  context "update_attributes!" do
    should "pass the attributes to self.attributes" do
      person = RealPerson.create
      attributes = { :foo => 'bar' }
      person.expects(:attributes=).with(attributes)
      person.expects(:save!)
      person.update_attributes!(attributes)
    end
  end  
end