$LOAD_PATH.unshift(File.dirname(__FILE__))
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
require 'rubygems'

gem "mocha", "0.9.8"

require "mocha"
require "mongoid"
require "spec"

Mongoid.connect_to("mongoid_test")

Spec::Runner.configure do |config|
  config.mock_with :mocha
  Mocha::Configuration.prevent(:stubbing_non_existent_method)
end

class MixedDrink < Mongoid::Document
  field :name
end

class Person < Mongoid::Document
  include Mongoid::Timestamps
  field :title
  field :terms, :type => Boolean
  field :age, :type => Integer, :default => 100
  field :dob, :type => Date
  field :mixed_drink, :type => MixedDrink
  has_many :addresses
  has_many :phone_numbers, :class_name => "Phone"
  has_one :name
  has_one :pet, :class_name => "Animal"

  def update_addresses
    addresses.each_with_index do |address, i|
      address.street = "Updated #{i}"
    end
  end
end

class CountryCode < Mongoid::Document
  field :code, :type => Integer
  key :code
  belongs_to :phone_number
end

class Phone < Mongoid::Document
  field :number
  key :number
  belongs_to :person
  has_one :country_code
end

class Animal < Mongoid::Document
  field :name
  key :name
  belongs_to :person
end

class PetOwner < Mongoid::Document
  field :title
  has_one :pet
end

class Pet < Mongoid::Document
  field :name
  field :weight, :type => Float, :default => 0.0
  has_many :vet_visits
  belongs_to :pet_owner
end

class VetVisit < Mongoid::Document
  field :date, :type => Date
  belongs_to :pet
end

class Address < Mongoid::Document
  field :street
  field :city
  field :state
  field :comment_code
  key :street
  belongs_to :addressable
end

class Name < Mongoid::Document
  field :first_name
  field :last_name
  key :first_name, :last_name
  belongs_to :person
end

class Comment < Mongoid::Document
  include Mongoid::Versioning
  field :text
end

class Decorated
  include Mongoid::Associations::Decorator
  def initialize(doc)
    @document = doc
  end
end

class Post < Mongoid::Document
  include Mongoid::Versioning
  field :title
end