mongoo 0.1.2

data/mongoo.gemspec

@@ -0,0 +1,91 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{mongoo}
+  s.version = "0.1.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Ben Myles"]
+  s.date = %q{2011-02-15}
+  s.description = %q{Simple object mapper for MongoDB}
+  s.email = %q{ben.myles@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    ".rvmrc",
+    "Gemfile",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "lib/mongoo.rb",
+    "lib/mongoo/attribute_proxy.rb",
+    "lib/mongoo/attribute_sanitizer.rb",
+    "lib/mongoo/base.rb",
+    "lib/mongoo/changelog.rb",
+    "lib/mongoo/cursor.rb",
+    "lib/mongoo/hash_ext.rb",
+    "lib/mongoo/modifiers.rb",
+    "lib/mongoo/mongohash.rb",
+    "lib/mongoo/persistence.rb",
+    "test/helper.rb",
+    "test/test_activemodel.rb",
+    "test/test_mongohash.rb",
+    "test/test_mongoo.rb"
+  ]
+  s.homepage = %q{http://github.com/benmyles/mongoo}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Object mapper for MongoDB}
+  s.test_files = [
+    "test/helper.rb",
+    "test/test_activemodel.rb",
+    "test/test_mongohash.rb",
+    "test/test_mongoo.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<i18n>, [">= 0.4.1"])
+      s.add_runtime_dependency(%q<activesupport>, [">= 3.0.3"])
+      s.add_runtime_dependency(%q<activemodel>, [">= 3.0.3"])
+      s.add_runtime_dependency(%q<mongo-em>, [">= 1.2.1"])
+      s.add_runtime_dependency(%q<bson_ext>, [">= 1.2.2"])
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.5.1"])
+      s.add_development_dependency(%q<rcov>, [">= 0"])
+    else
+      s.add_dependency(%q<i18n>, [">= 0.4.1"])
+      s.add_dependency(%q<activesupport>, [">= 3.0.3"])
+      s.add_dependency(%q<activemodel>, [">= 3.0.3"])
+      s.add_dependency(%q<mongo-em>, [">= 1.2.1"])
+      s.add_dependency(%q<bson_ext>, [">= 1.2.2"])
+      s.add_dependency(%q<shoulda>, [">= 0"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+      s.add_dependency(%q<rcov>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<i18n>, [">= 0.4.1"])
+    s.add_dependency(%q<activesupport>, [">= 3.0.3"])
+    s.add_dependency(%q<activemodel>, [">= 3.0.3"])
+    s.add_dependency(%q<mongo-em>, [">= 1.2.1"])
+    s.add_dependency(%q<bson_ext>, [">= 1.2.2"])
+    s.add_dependency(%q<shoulda>, [">= 0"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+    s.add_dependency(%q<rcov>, [">= 0"])
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,62 @@
+require 'rubygems'
+require 'bundler'
+begin
+  groups = [:default, :development]
+  if defined?(EM) && EM.reactor_running?
+    groups << :async
+  else
+    groups << :sync
+  end
+  Bundler.setup(groups)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'mongoo'
+
+Mongoo.config = {host: "127.0.0.1", port: 27017, db: 'mongoo-test'}
+
+class SearchIndex < Mongoo::Base
+  attribute "terms", :type => :array
+  index "terms"
+end
+
+class Person < Mongoo::Base
+  attribute "name", :type => :string
+  attribute "visits", :type => :integer
+  attribute "interests", :type => :array
+  attribute "jobs.total", :type => :integer
+  attribute "jobs.professional", :type => :array
+  attribute "jobs.volunteer", :type => :array
+  attribute "jobs.internships.high_school", :type => :array
+  attribute "location.city", :type => :string
+  attribute "location.demographics.crime_rate", :type => :symbol
+  attribute "location.demographics.education_quality", :type => :symbol
+  attribute "misc", :type => :hash
+  
+  index "name"
+  index "location.city"
+end
+
+class TvShow < Mongoo::Base
+  attribute "name", :type => :string
+  attribute "cast.director", :type => :string
+  attribute "cast.lead", :type => :string
+  attribute "rating", :type => :float
+  attribute "comments", :type => :array
+  
+  index "name"
+  
+  validates_presence_of "name"
+  validates_presence_of "cast.director"
+  validates_presence_of "rating"
+end
+
+class Test::Unit::TestCase
+end

data/test/test_activemodel.rb

@@ -0,0 +1,14 @@
+require 'helper'
+
+class TestActivemodel < Test::Unit::TestCase
+  include ActiveModel::Lint::Tests
+
+  def setup
+    [TvShow].each do |obj|
+      obj.drop
+      obj.create_indexes
+    end
+    @model = TvShow.new
+  end
+end
+

data/test/test_mongohash.rb

@@ -0,0 +1,36 @@
+require 'helper'
+
+class TestMongohash < Test::Unit::TestCase
+  should "work with dot_get, dot_set, dot_delete" do
+    h = Mongoo::Mongohash.new({
+      "name" => "Ben", 
+      "travel" => {},
+      "location" => { "city" => "SF", "zip" => 94107, "demographics" => { "pop" => 120000 } }})
+    assert_equal "Ben", h.dot_get("name")
+    assert_equal "SF", h.dot_get("location.city")
+    assert_equal 94107, h.dot_get("location.zip")
+    assert_equal 120000, h.dot_get("location.demographics.pop")
+    assert_equal nil, h.dot_get("foobar")
+    assert_equal nil, h.dot_get("location.idontexist.andneitherdoi")
+    h.dot_set("location.demographics.pop", 10000)
+    assert_equal 10000, h.dot_get("location.demographics.pop")
+    h.dot_set("interests", ["skydiving"])
+    assert_equal ["skydiving"], h.dot_get("interests")
+    h.dot_set("languages.fluent", ["english"])
+    h.dot_set("languages.nonfluent", ["german"])
+    assert_equal ["english"], h.dot_get("languages.fluent")
+    assert_equal ["german"], h.dot_get("languages.nonfluent")
+    h.dot_delete("languages.fluent")
+    assert_equal nil, h.dot_get("languages.fluent")
+    h.dot_delete("languages")
+    assert_equal nil, h.dot_get("languages")
+    assert_equal "Ben", h.dot_get("name")
+    h.dot_delete("name")
+    assert_equal nil, h.dot_get("name")
+  end
+
+  # should "not be able to initialize an object with undefined attributes" do
+  #   assert_raise(Mongoo::UnknownAttributeError) { Person.new(:idontexist => "bah") }
+  # end
+end
+

data/test/test_mongoo.rb

@@ -0,0 +1,276 @@
+require 'helper'
+
+class TestMongoo < Test::Unit::TestCase
+
+  def setup
+    [Person, TvShow, SearchIndex].each do |obj|
+      obj.drop
+      obj.create_indexes
+    end
+  end
+
+  should "set and get attributes" do
+    p = Person.new("name" => "Ben")
+    assert_equal "Ben", p.g(:name)
+    assert_equal "Ben", p.get(:name)
+    assert_equal "Ben", p.get_attribute(:name)
+    p.set("location.city", "San Francisco")
+    assert_equal "San Francisco", p.get("location.city")
+    p.sets({"location.demographics.crime_rate" => :high, "location.demographics.education_quality" => :low})
+    assert_equal({"name"=>"Ben",
+     "location.demographics.crime_rate"=>:high}, p.gets(%w(name location.demographics.crime_rate)))
+    assert_raise(Mongoo::UnknownAttributeError) { p.set("idontexist", "foo") }
+    assert_raise(Mongoo::UnknownAttributeError) { p.get("idontexist") }
+    assert_raise(NoMethodError) { p.idontexist }
+    assert_raise(NoMethodError) { p.idontexist = "val" }
+  end
+
+  should "have attribute methods" do
+    p = Person.new("name" => "Ben", "jobs" => { "professional" => "iList" })
+    p.jobs.internships.high_school = ["Sun Microsystems"]
+    assert_equal ["Sun Microsystems"], p.jobs.internships.high_school
+    assert_equal ["Sun Microsystems"], p.get("jobs.internships.high_school")
+    p.name = "Ben Myles"
+    assert_equal "Ben Myles", p.name
+    assert_equal "Ben Myles", p.get("name")
+  end
+
+  should "never has the same object for persisted_mongohash and mongohash" do
+    p = Person.new("name" => "Ben")
+    p.insert
+    p.name = "Ben Myles"
+    assert_not_equal p.persisted_mongohash.raw_hash, p.mongohash.raw_hash
+    p.update
+    assert_equal p.persisted_mongohash.raw_hash, p.mongohash.raw_hash
+    p.name = "Ben 2"
+    assert_not_equal p.persisted_mongohash.raw_hash, p.mongohash.raw_hash
+  end
+
+  should "be able to do crud" do
+    p = Person.new("name" => "Ben")
+    p.jobs.internships.high_school = ["Sun Microsystems"]
+    p.insert
+  
+    p = Person.find_one(p.id)
+    p.update
+    p.location.city = "San Francisco"
+    p.update
+    p.name = "Ben Myles"
+    p.location.city = "San Diego"
+    assert_not_equal p.persisted_mongohash.raw_hash, p.mongohash.raw_hash
+    p.update
+  
+    p2 = Person.find_one(p.id)
+    assert_equal "Ben Myles", p2.name
+    assert_equal "San Diego", p2.location.city
+  
+    p.location.city = "Los Angeles"
+    p.update!
+  
+    p2.location.city = "San Jose"
+    assert_raise(Mongoo::StaleUpdateError) { p2.update! }
+    p2.location.city = "San Diego"
+    p2.name = "Benjamin"
+    p2.update!
+  
+    assert p2.reload
+  
+    assert_equal "Los Angeles", p2.location.city
+    assert_equal "Benjamin", p2.name
+  
+    assert p2.persisted_mongohash.raw_hash["location"].has_key?("city")
+    p2.unset "location.city"
+    p2.update
+    assert !p2.persisted_mongohash.raw_hash["location"].has_key?("city")
+  
+    p2.location.demographics.crime_rate = :high
+    p2.location.city = "San Bruno"
+    p2.update
+    assert_raise(NoMethodError) { p2.location.demographics = 123 }
+
+    p2.unset "location"
+    p2.update
+    p2 = Person.find_one(p2.id)
+    assert !p2.persisted_mongohash.raw_hash.has_key?("location")
+  
+    p2.location.city = "Brisbane"
+    p2.location.demographics.crime_rate = :low
+    p2.update
+    assert_equal ["Brisbane", :low], [p2.location.city, p2.location.demographics.crime_rate]
+    p2 = Person.find_one(p2.id)
+    assert_equal ["Brisbane", :low], [p2.location.city, p2.location.demographics.crime_rate]
+    p2.location.unset
+    p2.update
+    assert !p2.persisted_mongohash.raw_hash.has_key?("location")
+    p2 = Person.find_one(p2.id)
+    assert !p2.persisted_mongohash.raw_hash.has_key?("location")
+  
+    p2.location.city = "Brisbane"
+    p2.location.demographics.crime_rate = :low
+    p2.update
+    assert_equal ["Brisbane", :low], [p2.location.city, p2.location.demographics.crime_rate]
+    assert p2.persisted_mongohash.raw_hash["location"].has_key?("demographics")
+    p2.location.demographics.unset
+    p2.update
+    assert !p2.persisted_mongohash.raw_hash["location"].has_key?("demographics")
+    p2 = Person.find_one(p2.id)
+    assert !p2.persisted_mongohash.raw_hash["location"].has_key?("demographics")
+    assert_equal "Brisbane", p2.location.city
+  end
+
+  should "be able to modify fields" do
+    p = Person.new("name" => "Ben", "visits" => 0)
+    p.insert!
+    assert_equal 0, p.visits
+
+    p.mod! do |mod|
+      mod.inc "visits", 2
+      mod.set "jobs.total", 10
+    end
+    assert_equal 2, p.visits
+    assert_equal 10, p.jobs.total
+    assert_equal [], p.changelog
+    p = Person.find_one(p.id)
+    assert_equal 2, p.visits
+    assert_equal 10, p.jobs.total
+
+    p.jobs.mod! do |mod|
+      mod.inc "total", 5
+    end
+    assert_equal 15, p.jobs.total
+    p = Person.find_one(p.id)
+    assert_equal 15, p.jobs.total
+  
+    assert_equal nil, p.interests
+    p.mod! { |mod| mod.push("interests", "skydiving") }
+    assert_equal ["skydiving"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving"], p.interests
+    p.mod! { |mod| mod.push("interests", "snowboarding") }
+    assert_equal ["skydiving", "snowboarding"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving", "snowboarding"], p.interests
+  
+    p.mod! { |mod| mod.push_all("interests", ["reading","travelling"]) }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+  
+    p.mod! { |mod| mod.add_to_set("interests", "skydiving") }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+    p.mod! { |mod| mod.add_to_set("interests", "swimming") }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling", "swimming"], p.interests
+    p.mod! { |mod| mod.add_to_set("interests", "swimming") }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling", "swimming"], p.interests
+  
+    p.mod! { |mod| mod.pop("interests") }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+  
+    p.mod! { |mod| mod.pop("interests") }
+    assert_equal ["skydiving", "snowboarding", "reading"], p.interests
+  
+    p.mod! { |mod| mod.push("interests", "reading") }
+    assert_equal ["skydiving", "snowboarding", "reading", "reading"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving", "snowboarding", "reading", "reading"], p.interests
+    p.mod! { |mod| mod.pull("interests", "reading") }
+    assert_equal ["skydiving", "snowboarding"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving", "snowboarding"], p.interests
+
+    p.mod! { |mod| mod.push_all("interests", ["reading","travelling"]) }
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["skydiving", "snowboarding", "reading", "travelling"], p.interests
+  
+    p.mod! { |mod| mod.pull_all("interests", ["reading", "skydiving"]) }
+    assert_equal ["snowboarding", "travelling"], p.interests
+    p = Person.find_one(p.id)
+    assert_equal ["snowboarding", "travelling"], p.interests
+  end
+
+  should "not be able to modify fields that don't exist" do
+    p = Person.new("name" => "Ben", "visits" => 0)
+    p.insert!
+    assert_raise(Mongoo::UnknownAttributeError) do
+      p.mod! { |mod| mod.push("idontexist", "foobar") }
+    end
+  end
+
+  should "be able to access a hash type directly" do
+    p = Person.new("name" => "Ben")
+    p.insert!
+    assert_equal nil, p.misc
+    p.misc = { "height" => 5.5, "weight" => 160 }
+    p.update!
+    assert_equal({ "height" => 5.5, "weight" => 160 }, p.misc)
+    p.update!
+    p = Person.find_one(p.id)
+    assert_equal({ "height" => 5.5, "weight" => 160 }, p.misc)
+    p.misc["height"] = 5.6
+    p.misc["eyes"] = :blue
+    assert_equal({ "height" => 5.6, "weight" => 160, "eyes" => :blue }, p.misc)
+    p.update!
+    assert_equal({ "height" => 5.6, "weight" => 160, "eyes" => :blue }, p.misc)
+    p = Person.find_one(p.id)
+    assert_equal({ "height" => 5.6, "weight" => 160, "eyes" => :blue }, p.misc)
+  end
+
+  should "cast to type automatically" do
+    p = Person.new("name" => "ben")
+    p.insert!
+    p.visits = "5"
+    assert_equal 5, p.visits
+    p.update!
+    p = Person.find_one(p.id)
+    assert_equal 5, p.visits
+    p.mod! { |mod| mod.inc("visits", 3) }
+    assert_equal 8, p.visits
+    p = Person.find_one(p.id)
+    assert_equal 8, p.visits
+  end
+
+  should "validate documents" do
+    show = TvShow.new
+    assert !show.valid?
+    assert !show.insert
+    assert_equal({:name=>["can't be blank"],
+     :"cast.director"=>["can't be blank"],
+     :rating=>["can't be blank"]}, show.errors)
+    show.cast.director = "Some Guy"
+    show.name = "Some Show"
+    assert !show.valid?
+    assert !show.insert
+    assert_raise(Mongoo::NotValidError) { show.insert! }
+    assert_equal({:rating=>["can't be blank"]}, show.errors)
+    show.rating = 3.0
+    assert show.valid?
+    show.insert!
+  end
+
+  should "not care if keys are symbols or hashes" do
+    p = Person.new(:name => "Ben")
+    assert_equal "Ben", p.name
+  end
+
+  should "be able to merge properties into the mongohash" do
+    p = Person.new("name" => "Ben", "visits" => 10, "location" => { "city" => "SF" })
+    p.insert!
+    p = Person.find_one(p.id)
+    assert_equal "Ben", p.name
+    assert_equal 10, p.visits
+    assert_equal "SF", p.location.city
+    p.merge!({"visits" => 12, "location" => { "demographics" => {"crime_rate" => :high} } })
+    assert_equal "Ben", p.name
+    assert_equal 12, p.visits
+    assert_equal "SF", p.location.city
+  end
+
+  should "not fail on an empty array" do
+    i = SearchIndex.new(:terms => [])
+    i.insert!
+    i.terms << "foo"
+    i.update!
+  end
+end
+