mongoid_geospatial 1.0.0rc0

data/spec/functional/criterion/inclusion_spec.rb

@@ -0,0 +1,356 @@
+require "spec_helper"
+
+describe Mongoid::Criterion::Inclusion do
+
+  before do
+    Person.delete_all
+  end
+
+  describe "#where" do
+
+    let(:dob) do
+      33.years.ago.to_date
+    end
+
+    let(:lunch_time) do
+      30.minutes.ago
+    end
+
+    let!(:person) do
+      Person.create(
+        :title => "Sir",
+        :dob => dob,
+        :lunch_time => lunch_time,
+        :age => 33,
+        :aliases => [ "D", "Durran" ],
+        :things => [ { :phone => 'HTC Incredible' } ]
+      )
+    end
+
+    context "when providing 24 character strings" do
+
+      context "when the field is not an id field" do
+
+        let(:string) do
+          BSON::ObjectId.new.to_s
+        end
+
+        let!(:person) do
+          Person.create(:title => string)
+        end
+
+        let(:from_db) do
+          Person.where(:title => string)
+        end
+
+        it "does not convert the field to a bson id" do
+          from_db.should == [ person ]
+        end
+      end
+    end
+
+    context "when providing string object ids" do
+
+      context "when providing a single id" do
+
+        let(:from_db) do
+          Person.where(:_id => person.id.to_s).first
+        end
+
+        it "returns the matching documents" do
+          from_db.should == person
+        end
+      end
+    end
+
+    context "chaining multiple wheres" do
+
+      context "when chaining on the same key" do
+
+        let(:from_db) do
+          Person.where(:title => "Maam").where(:title => "Sir")
+        end
+
+        it "overrides the previous key" do
+          from_db.should == [ person ]
+        end
+      end
+
+      context "with different criteria on the same key" do
+
+        it "merges criteria" do
+          Person.where(:age.gt => 30).where(:age.lt => 40).should == [person]
+        end
+
+        it "typecasts criteria" do
+          before_dob = (dob - 1.month).to_s
+          after_dob = (dob + 1.month).to_s
+          Person.where(:dob.gt => before_dob).and(:dob.lt => after_dob).should == [person]
+        end
+
+      end
+    end
+
+    context "with untyped criteria" do
+
+      it "typecasts integers" do
+        Person.where(:age => "33").should == [ person ]
+      end
+
+      it "typecasts datetimes" do
+        Person.where(:lunch_time => lunch_time.to_s).should == [ person ]
+      end
+
+      it "typecasts dates" do
+        Person.where({:dob => dob.to_s}).should == [ person ]
+      end
+
+      it "typecasts times with zones" do
+        time = lunch_time.in_time_zone("Alaska")
+        Person.where(:lunch_time => time).should == [ person ]
+      end
+
+      it "typecasts array elements" do
+        Person.where(:age.in => [17, "33"]).should == [ person ]
+      end
+
+      it "typecasts size criterion to integer" do
+        Person.where(:aliases.size => "2").should == [ person ]
+      end
+
+      it "typecasts exists criterion to boolean" do
+        Person.where(:score.exists => "f").should == [ person ]
+      end
+    end
+
+    context "with multiple complex criteria" do
+
+      before do
+        Person.create(:title => "Mrs", :age => 29)
+        Person.create(:title => "Ms", :age => 41)
+      end
+
+      it "returns those matching both criteria" do
+        Person.where(:age.gt => 30, :age.lt => 40).should == [person]
+      end
+
+      it "returns nothing if in and nin clauses cancel each other out" do
+        Person.any_in(:title => ["Sir"]).not_in(:title => ["Sir"]).should == []
+      end
+
+      it "returns nothing if in and nin clauses cancel each other out ordered the other way" do
+        Person.not_in(:title => ["Sir"]).any_in(:title => ["Sir"]).should == []
+      end
+
+      it "returns the intersection of in and nin clauses" do
+        Person.any_in(:title => ["Sir", "Mrs"]).not_in(:title => ["Mrs"]).should == [person]
+      end
+
+      it "returns the intersection of two in clauses" do
+        Person.where(:title.in => ["Sir", "Mrs"]).where(:title.in => ["Sir", "Ms"]).should == [person]
+      end
+    end
+
+    context "with complex criterion" do
+
+      context "#all" do
+
+        it "returns those matching an all clause" do
+          Person.where(:aliases.all => ["D", "Durran"]).should == [person]
+        end
+      end
+
+      context "#exists" do
+
+        it "returns those matching an exists clause" do
+          Person.where(:title.exists => true).should == [person]
+        end
+      end
+
+      context "#gt" do
+
+        it "returns those matching a gt clause" do
+          Person.where(:age.gt => 30).should == [person]
+        end
+      end
+
+      context "#gte" do
+
+        it "returns those matching a gte clause" do
+          Person.where(:age.gte => 33).should == [person]
+        end
+      end
+
+      context "#in" do
+
+        it "returns those matching an in clause" do
+          Person.where(:title.in => ["Sir", "Madam"]).should == [person]
+        end
+
+        it "allows nil" do
+          Person.where(:ssn.in => [nil]).should == [person]
+        end
+      end
+
+      context "#lt" do
+
+        it "returns those matching a lt clause" do
+          Person.where(:age.lt => 34).should == [person]
+        end
+      end
+
+      context "#lte" do
+
+        it "returns those matching a lte clause" do
+          Person.where(:age.lte => 33).should == [person]
+        end
+      end
+
+      context "#ne" do
+
+        it "returns those matching a ne clause" do
+          Person.where(:age.ne => 50).should == [person]
+        end
+      end
+
+      context "#nin" do
+
+        it "returns those matching a nin clause" do
+          Person.where(:title.nin => ["Esquire", "Congressman"]).should == [person]
+        end
+      end
+
+      context "#size" do
+
+        it "returns those matching a size clause" do
+          Person.where(:aliases.size => 2).should == [person]
+        end
+      end
+
+      context "#match" do
+
+        it "returns those matching a partial element in a list" do
+          Person.where(:things.matches => { :phone => "HTC Incredible" }).should == [person]
+        end
+      end
+
+    end
+
+    context "Geo Spatial Complex Where" do
+
+      let!(:home) do
+        [-73.98,40.77]
+      end
+
+      describe "#near" do
+        before do
+          Bar.delete_all
+          Bar.create_indexes
+        end
+
+        let!(:berlin) do
+          Bar.create(:location => [ 52.30, 13.25 ])
+        end
+
+        let!(:prague) do
+          Bar.create(:location => [ 50.5, 14.26 ])
+        end
+
+        let!(:paris) do
+          Bar.create(:location => [ 48.48, 2.20 ])
+        end
+
+        it "returns the documents sorted closest to furthest" do
+          Bar.where(:location.near => [ 41.23, 2.9 ]).should == [ paris, prague, berlin ]
+        end
+
+        it "returns the documents sorted closest to furthest" do
+          Bar.where(:location.near => {:point=>[ 41.23, 2.9 ],:max => 20}).should == [ paris, prague, berlin ]
+        end
+
+        it "returns the documents sorted closest to furthest" do
+          Bar.where(:location.near_sphere => [ 41.23, 2.9 ]).should == [ paris, prague, berlin ]
+        end
+
+      end
+
+      context "#within" do
+
+        context ":box, :polygon" do
+          before do
+            Bar.delete_all
+            Bar.create_indexes
+          end
+
+          let!(:berlin) do
+            Bar.create(:name => 'berlin', :location => [ 52.30, 13.25 ])
+          end
+
+          let!(:prague) do
+            Bar.create(:name => 'prague',:location => [ 50.5, 14.26 ])
+          end
+
+          let!(:paris) do
+            Bar.create(:name => 'prague',:location => [ 48.48, 2.20 ])
+          end
+
+          it "returns the documents within a box" do
+            Bar.where(:location.within(:box) => [[ 47, 1 ],[ 49, 3 ]]).should == [ paris ]
+          end
+
+          it "returns the documents within a polygon", :if => (Mongoid.master.connection.server_version >= '1.9') do
+            Bar.where(:location.within(:polygon) => [[ 47, 1 ],[49,1.5],[ 49, 3 ],[46,5]]).should == [ paris ]
+          end
+
+          it "returns the documents within a center" do
+            Bar.where(:location.within(:center) => [[ 47, 1 ],4]).should == [ paris ]
+          end
+
+          it "returns the documents within a center_sphere" do
+            Bar.where(:location.within(:center_sphere) => [[ 48, 2 ],0.1]).should == [ paris ]
+          end
+
+        end
+        context ":circle :center_sphere" do
+          before do
+            Bar.delete_all
+            Bar.create_indexes
+          end
+          let!(:mile1) do
+            Bar.create(:name => 'mile1', :location => [-73.997345, 40.759382])
+          end
+
+          let!(:mile3) do
+            Bar.create(:name => 'mile2', :location => [-73.927088, 40.752151])
+          end
+
+          let!(:mile7) do
+            Bar.create(:name => 'mile3', :location => [-74.0954913, 40.7161472])
+          end
+
+          let!(:mile11) do
+            Bar.create(:name => 'mile4', :location => [-74.0604951, 40.9178011])
+          end
+
+          it "returns the documents within a center_sphere" do
+            Bar.where(:location.within(:center_sphere) => {:point => home,:max => 2, :unit => :mi}).should == [ mile1 ]
+          end
+
+          it "returns the documents within a center_sphere" do
+            Bar.where(:location.within(:center_sphere) => {:point => home,:max => 4, :unit => :mi}).should include(mile3)
+          end
+
+          it "returns the documents within a center_sphere" do
+            Bar.where(:location.within(:center_sphere) => {:point => home,:max => 8, :unit => :mi}).should include(mile7)
+          end
+
+          it "returns the documents within a center_sphere" do
+            Bar.where(:location.within(:center_sphere) => {:point => home,:max => 12, :unit => :mi}).should include(mile11)
+          end
+        end
+      end
+    end
+
+  end
+
+end

data/spec/functional/mongoid_geospatial_spec.rb

@@ -0,0 +1,54 @@
+require "spec_helper"
+
+describe Mongoid::Geospatial do
+
+  it "should instantiate with no problems" do
+    Bar.create!(name: "Moe's")
+    Bar.count.should eql(1)
+  end
+
+  it "should have a field mapped as point" do
+    bar = Bar.create!(location: [5,5])
+    bar.location.should be_a RGeo::Geographic::SphericalPointImpl
+  end
+
+  it "should accept an RGeo object" do
+    point = RGeo::Geographic.spherical_factory.point 1, 2
+    bar = Bar.create!(location: point)
+    bar.location.x.should be_within(0.1).of(1)
+    bar.location.y.should be_within(0.1).of(2)
+  end
+
+  it "should calculate distance between points" do
+    bar = Bar.create!(location: [5,5])
+    bar2 = Bar.create!(location: [15,15])
+    bar.location.distance(bar2.location).should be_within(1).of(1561283.8)
+  end
+
+  it "should calculate distance between points miles" do
+    pending
+    bar = Bar.create!(location: [5,5])
+    bar2 = Bar.create!(location: [15,15])
+    bar.location.distance(bar2.location).should be_within(1).of(1561283.8)
+  end
+
+  it "should calculate 3d distances by default" do
+    bar = Bar.create! location: [-73.77694444, 40.63861111 ]
+    bar2 = Bar.create! location: [-118.40, 33.94] #,:unit=>:mi, :spherical => true)
+    bar.location.distance(bar2.location).to_i.should be_within(1).of(2469)
+  end
+
+  it "should have a nice simple way to ovewrite geo factory" do
+    pending
+    bar = Bar.create!(location: [5,5])
+    bar2 = Bar.create!(location: [15,15])
+    bar.location.distance(bar2.location).should be_within(1).of(1561283.8)
+  end
+
+  it "should have a field mapped as polygon" do
+    farm = Farm.create!(area: [[5,5],[6,5],[6,6],[5,6]])
+    farm.area.should be_a RGeo::Geographic::SphericalPolygonImpl
+  end
+
+
+end

data/spec/functional/spatial/geo_near_results_spec.rb

@@ -0,0 +1,78 @@
+require "spec_helper"
+
+describe Mongoid::Geospatial::GeoNearResults do
+  before do
+    Bar.delete_all
+    Bar.create_indexes
+
+    50.times do |i|
+      Bar.create!(:name => i.to_s, :location => [rand(358)-179,rand(358)-179])
+    end
+  end
+
+
+
+  context ":paginator :array" do
+    let(:bars) { Bar.geo_near([1,1]) }
+    let(:sorted_bars) {
+      bars = Bar.geo_near([1,1])
+      bars.sort_by! {|b| b.name.to_i}
+      bars
+    }
+    [nil,1,2].each do |page|
+      it "page=#{page} should have 25" do
+        bars.page(page).size.should == 25
+      end
+    end
+
+    [1,2].each do |page|
+      it "modified result should keep order after pagination" do
+        sorted_bars.page(page).should == sorted_bars.slice((page-1)*25,25)
+      end
+    end
+
+    { nil => 25, 20 => 20 , 30 => 20, 50 => 0}.each do |per, total|
+      it "page=2 per=#{per} should have #{total}" do
+        bars.per(per).page(2).size.should == total
+        bars.page(2).per(per).size.should == total
+      end
+    end
+
+    it "page=3 should have 0" do
+      bars.page(3).size.should == 0
+    end
+
+    it "per=5" do
+      bars.per(5).size.should == 5
+    end
+
+    it "page=10 per=5" do
+      bars.per(5).page(10).should == bars[45..50]
+    end
+
+  end
+
+  context ":paginator :kaminari" do
+    let!(:near) {Bar.geo_near([1,1]).page(1)}
+    it "should have current_page" do
+      near.current_page.should == 1
+    end
+
+    it "should have num_pages" do
+      near.total_entries.should == 50
+      near.num_pages.should == 2
+    end
+
+    it "should have limit_value" do
+      near.limit_value.should == 25
+    end
+  end
+
+
+  context ":paginator :num_pages" do
+    it "when total=55 per=10 ,num_pages should be 6" do
+      5.times { |i| Bar.create(:name => i.to_s, :location => [rand(358)-179,rand(358)-179]) }
+      Bar.geo_near([1,1]).per(10).num_pages.should    == 6
+    end
+  end
+end

data/spec/models/account.rb

@@ -0,0 +1,19 @@
+class Account
+  include Mongoid::Document
+  field :number, :type => String
+  field :balance, :type => String
+  field :nickname, :type => String
+  field :name, :type => String
+  key :name
+
+  embeds_many :memberships
+  referenced_in :creator, :class_name => "User", :foreign_key => :creator_id
+  referenced_in :person
+  references_many :alerts
+  references_and_referenced_in_many :agents
+
+  attr_accessible :nickname, :name
+
+  validates_presence_of :name
+  validates_length_of :name, :maximum => 10, :on => :create
+end

data/spec/models/acolyte.rb

@@ -0,0 +1,9 @@
+class Acolyte
+  include Mongoid::Document
+  field :status
+  field :name
+
+  scope :active, where(:status => "active")
+
+  default_scope asc(:name)
+end

data/spec/models/address.rb

@@ -0,0 +1,62 @@
+class Address
+  include Mongoid::Document
+
+  attr_accessor :mode
+
+  field :address_type
+  field :number, :type => Integer
+  field :street
+  field :city
+  field :state
+  field :post_code
+  field :parent_title
+  field :services, :type => Array
+  field :latlng, :type => Array
+  key :street
+  embeds_many :locations
+
+  embedded_in :addressable, :polymorphic => true do
+    def extension
+      "Testing"
+    end
+    def doctor?
+      title == "Dr"
+    end
+  end
+
+  accepts_nested_attributes_for :locations
+
+  referenced_in :account
+
+  scope :without_postcode, where(:postcode => nil)
+  named_scope :rodeo, where(:street => "Rodeo Dr") do
+    def mansion?
+      all? { |address| address.street == "Rodeo Dr" }
+    end
+  end
+
+  validates_presence_of :street, :on => :update
+  validates_format_of :street, :with => /\D/, :allow_nil => true
+
+  def set_parent=(set = false)
+    self.parent_title = addressable.title if set
+  end
+
+  def <=>(other)
+    street <=> other.street
+  end
+
+  class << self
+    def california
+      where(:state => "CA")
+    end
+
+    def homes
+      where(:address_type => "Home")
+    end
+
+    def streets
+      all.map(&:street)
+    end
+  end
+end

data/spec/models/address_component.rb

@@ -0,0 +1,5 @@
+class AddressComponent
+  include Mongoid::Document
+  field :street, :type => String
+  embedded_in :person
+end

data/spec/models/agent.rb

@@ -0,0 +1,10 @@
+class Agent
+  include Mongoid::Document
+  include Mongoid::Timestamps::Updated
+  field :title
+  field :number
+  embeds_many :names, :as => :namable
+  referenced_in :game
+
+  references_and_referenced_in_many :accounts
+end

data/spec/models/alert.rb

@@ -0,0 +1,5 @@
+class Alert
+  include Mongoid::Document
+  field :message, :type => String
+  referenced_in :account
+end

data/spec/models/animal.rb

@@ -0,0 +1,21 @@
+class Animal
+  include Mongoid::Document
+  field :name
+  field :tags, :type => Array
+  key :name
+
+  embedded_in :person
+  embedded_in :circus
+
+  validates_format_of :name, :without => /\$\$\$/
+
+  accepts_nested_attributes_for :person
+
+  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
+end

data/spec/models/bar.rb

@@ -0,0 +1,9 @@
+class Bar
+  include Mongoid::Document
+  include Mongoid::Geospatial
+
+  field :name, :type => String
+  field :location, :type => Point, :spatial => true
+  references_one :rating, :as => :ratable
+  spatial_index :location
+end

data/spec/models/birthday.rb

@@ -0,0 +1,13 @@
+class Birthday
+  include Mongoid::Document
+  field :title
+  field :date, :type => Date
+  embedded_in :owner, :inverse_of => :birthdays
+
+  def self.each_day(start_date, end_date)
+    groups = only(:date).asc(:date).where(:date.gte => start_date, :date.lte => end_date).group
+    groups.each do |date, group|
+      yield(date, group)
+    end
+  end
+end

data/spec/models/book.rb

@@ -0,0 +1,5 @@
+class Book
+  include Mongoid::Document
+  field :title, :type => String
+  references_one :rating, :as => :ratable, :dependent => :nullify
+end

data/spec/models/business.rb

@@ -0,0 +1,7 @@
+class Business
+  include Mongoid::Document
+
+  set_database :secondary
+
+  field :name
+end