mongoid-locomotive 2.0.0.beta9

data/spec/models/account.rb

@@ -0,0 +1,5 @@
+class Account
+  include Mongoid::Document
+  referenced_in :creator, :class_name => "User", :foreign_key => :creator_id
+  field :number
+end

data/spec/models/address.rb

@@ -0,0 +1,40 @@
+class Address
+  include Mongoid::Document
+  field :address_type
+  field :number, :type => Integer
+  field :street
+  field :city
+  field :state
+  field :post_code
+  field :parent_title
+  field :services, :type => Array
+  key :street
+  embeds_many :locations
+
+  embedded_in :addressable, :inverse_of => :addresses do
+    def extension
+      "Testing"
+    end
+    def doctor?
+      title == "Dr"
+    end
+  end
+
+  named_scope :rodeo, where(:street => "Rodeo Dr")
+
+  validates_presence_of :street, :on => :update
+
+  def set_parent=(set = false)
+    self.parent_title = addressable.title if set
+  end
+
+  class << self
+    def california
+      where(:state => "CA")
+    end
+
+    def homes
+      where(:address_type => "Home")
+    end
+  end
+end

data/spec/models/agent.rb

@@ -0,0 +1,7 @@
+class Agent
+  include Mongoid::Document
+  field :title
+  field :number
+  embeds_many :names
+  references_many :posts, :foreign_key => :poster_id
+end

data/spec/models/animal.rb

@@ -0,0 +1,15 @@
+class Animal
+  include Mongoid::Document
+  field :name
+  field :tags, :type => Array  
+  key :name
+  embedded_in :person, :inverse_of => :pet
+  
+  def tag_list
+    tags.join(", ")
+  end
+  
+  def tag_list=(_tag_list)
+    self.tags = _tag_list.split(",").map(&:strip)
+  end  
+end

data/spec/models/answer.rb

@@ -0,0 +1,4 @@
+class Answer
+  include Mongoid::Document
+  embedded_in :question, :inverse_of => :answers
+end

data/spec/models/callbacks.rb

@@ -0,0 +1,47 @@
+class Artist
+  include Mongoid::Document
+  field :name
+  embeds_many :songs
+  embeds_many :labels
+
+  before_create :before_create_stub
+  after_create :create_songs
+
+  protected
+  def before_create_stub
+    true
+  end
+
+  def create_songs
+    2.times { |n| songs.create!(:title => "#{n}") }
+  end
+end
+
+class Song
+  include Mongoid::Document
+  field :title
+  embedded_in :artist, :inverse_of => :songs
+end
+
+class Label
+  include Mongoid::Document
+  field :name
+  embedded_in :artist, :inverse_of => :labels
+  before_validation :cleanup
+
+  private
+  def cleanup
+    self.name = self.name.downcase.capitalize
+  end
+end
+
+class ValidationCallback
+  include Mongoid::Document
+  field :history, :type => Array, :default => []
+  validate do
+    self.history << :validate
+  end
+
+  before_validation { self.history << :before_validation }
+  after_validation { self.history << :after_validation }
+end

data/spec/models/category.rb

@@ -0,0 +1,13 @@
+class RootCategory
+  include Mongoid::Document
+  embeds_many :categories
+end
+
+class Category
+  include Mongoid::Document
+  embedded_in :root_category, :inverse_of => :categories
+  embedded_in :category, :inverse_of => :categories
+  embeds_many :categories
+
+  field :name
+end

data/spec/models/comment.rb

@@ -0,0 +1,10 @@
+class Comment
+  include Mongoid::Document
+  include Mongoid::Versioning
+  include Mongoid::Timestamps
+  field :title
+
+  field :text
+  key :text
+  validates_presence_of :text
+end

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/favorite.rb

@@ -0,0 +1,8 @@
+class Favorite
+  include Mongoid::Document
+
+  field :title
+  validates_uniqueness_of :title
+
+  embedded_in :person, :inverse_of => :favorites
+end

data/spec/models/game.rb

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

data/spec/models/inheritance.rb

@@ -0,0 +1,72 @@
+class Canvas
+  include Mongoid::Document
+  field :name
+  embeds_many :shapes
+  embeds_one :writer
+  embeds_one :palette
+
+  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
+
+class Palette
+  include Mongoid::Document
+  embedded_in :canvas, :inverse_of => :palette
+  embeds_many :tools
+end
+
+class Tool
+  include Mongoid::Document
+  embedded_in :palette, :inverse_of => :tools
+end
+
+class Pencil < Tool; end
+
+class Eraser < Tool; 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/login.rb

@@ -0,0 +1,6 @@
+class Login
+  include Mongoid::Document
+  field :username
+  key :username
+  validates_uniqueness_of :username
+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 :namable, :inverse_of => [:name, :names]
+
+  def set_parent=(set = false)
+    self.parent_title = namable.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/paranoid_post.rb

@@ -0,0 +1,18 @@
+class ParanoidPost
+  include Mongoid::Document
+  include Mongoid::Versioning
+  include Mongoid::Timestamps
+  include Mongoid::Paranoia
+  field :title
+  referenced_in :person
+
+  references_many :tags, :stored_as => :array
+
+  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/parents.rb

@@ -0,0 +1,32 @@
+class ParentDoc
+  include Mongoid::Document
+
+  embeds_many :child_docs
+
+  field :statistic
+  field :children_order, :type => Array, :default => [] # hold all the children's id
+end
+
+
+class ChildDoc
+  include Mongoid::Document
+
+  embedded_in :parent_doc, :inverse_of => :child_docs
+
+  attr_writer :position
+
+  after_save :update_position
+
+  def position
+    exsited_position = parent_doc.children_order.index(id)
+    exsited_position ? exsited_position + 1 : parent_doc.aspects.size
+  end
+
+  def update_position
+    if @position && (@position.to_i > 0)
+      parent_doc.children_order.delete(id)
+      parent_doc.children_order.insert(@position.to_i - 1, id)
+      parent_doc.save
+    end
+  end
+end

data/spec/models/patient.rb

@@ -0,0 +1,15 @@
+class Email
+  include Mongoid::Document
+  field :address
+  validates_uniqueness_of :address
+  embedded_in :patient, :inverse_of => :email
+end
+
+class Patient
+  include Mongoid::Document
+  field :title
+  store_in :inpatient
+  embeds_many :addresses
+  embeds_one :email
+  validates_presence_of :title, :on => :create
+end

data/spec/models/person.rb

@@ -0,0 +1,106 @@
+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
+  field :security_code
+
+  index :age
+  index :addresses
+  index :dob
+  index :name
+  index :title
+  index :ssn, :unique => true
+
+  attr_reader :rescored
+
+  attr_protected :security_code
+
+  embeds_many :favorites
+  embeds_many :videos
+  embeds_many :phone_numbers, :class_name => "Phone"
+  embeds_many :addresses do
+    def extension
+      "Testing"
+    end
+    def find_by_street(street)
+      @target.select { |doc| doc.street == street }
+    end
+  end
+
+  embeds_one :pet, :class_name => "Animal"
+  embeds_one :name do
+    def extension
+      "Testing"
+    end
+    def dawkins?
+      first_name == "Richard" && last_name == "Dawkins"
+    end
+  end
+
+  accepts_nested_attributes_for :addresses, :reject_if => lambda { |attrs| attrs["street"].blank? }
+  accepts_nested_attributes_for :name, :update_only => true
+  accepts_nested_attributes_for :pet
+  accepts_nested_attributes_for :favorites, :allow_destroy => true, :limit => 5
+
+  references_one :game do
+    def extension
+      "Testing"
+    end
+  end
+
+  references_many :posts do
+    def extension
+      "Testing"
+    end
+  end
+  references_many :paranoid_posts
+  references_many :preferences, :stored_as => :array, :inverse_of => :people
+
+  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