mongoid_geo 0.5.4.1 → 0.6.0

data/spec/mongoid/geo/geo_fields_spec.rb

@@ -0,0 +1,140 @@
+require "mongoid/spec_helper"
+
+describe Mongoid::Fields do
+
+  let(:address) do
+    Address.new        
+  end
+  
+  let(:point) do
+    b = (Struct.new :lat, :lng).new
+    b.lat = 72
+    b.lng = -44
+    b
+  end  
+
+  let(:location) do
+    a = (Struct.new :location).new
+    a.location = point
+    a
+  end  
+
+  let(:position) do
+    a = (Struct.new :position).new
+    a.position = point
+    a
+  end  
+
+
+  let(:other_point) do
+    b = (Struct.new :latitude, :longitude).new
+    b.latitude = 72
+    b.longitude = -44
+    b
+  end  
+
+
+  describe "Special geo Array setter" do
+    it "should split a String into parts and convert to floats" do
+      address.location = "23.5, -47"
+      address.location.should == [23.5, -47]
+    end
+    
+    it "should convert Strings into floats" do
+      address.location = "23.5", "-48"
+      address.location.should == [23.5, -48]
+    end
+    
+    it "should work with point Hash, keys :lat, :lng" do
+      address.location = {:lat => 23.5, :lng => -49}
+      address.location.should == [23.5, -49]
+    end
+    
+    it "should work with point Hash, keys :latitude, :longitude" do
+      address.location = {:latitude => 23.5, :longitude => -49}
+      address.location.should == [23.5, -49]
+    end
+    
+    it "should work with point Hash keys 0 and 1" do
+      address.location = {"0" => 41, "1" => -71}
+      address.location.should == [41, -71]
+    end
+    
+    
+    it "should work with point hashes using the first point only" do
+      address.location = [{:lat => 23.5, :lng => -49}, {:lat => 72, :lng => -49}]
+      address.location.should == [23.5, -49]
+    end
+    
+    it "should work with point object has #lat and #lng methods" do
+      address.location = point
+      address.location.should == [72, -44]
+    end
+
+    it 'should work with point object that has location attribute' do
+      address.location = location
+      address.location.should == [72, -44]
+    end
+
+    it 'should work with point object that has position attribute add should add #lat and #lng methods' do
+      address.location = position
+      address.location.should == [72, -44]
+      address.lat.should == 72
+      address.lng.should == -44
+    end
+
+    it 'should work with point object that has position attribute and should add #latitude and #longitude methods' do
+      address.pos = position
+      address.pos.should == [72, -44]
+      address.latitude.should == 72
+      address.latitude = 45
+      address.latitude.should == 45
+      address.longitude.should == -44
+    end
+
+    it 'should work with point object that has #latitude and #longitude methods' do
+      address.location = other_point
+      address.location.should == [72, -44]
+    end
+    
+    it "should work with point objects using the first point only" do
+      address.location = [point, {:lat => 72, :lng => -49}]
+      address.location.should == [72, -44]
+    end
+    
+    it "should drop nils" do
+      address.location = [nil, point, {:lat => 72, :lng => -49}]
+      address.location.should == [72, -44]
+    end
+    
+    it "should default to normal behavior" do
+      address.location = 23.5, -49
+      address.location.should == [23.5, -49]
+    
+      address.location = [23.5, -50]
+      address.location.should == [23.5, -50]
+    end
+    
+    it "should handle nil values" do
+      address.location = nil
+      address.location.should be_nil
+      address.lat.should      be_nil
+      address.lng.should      be_nil
+    end
+    
+    it "should allow lat then lon assignment when nil" do
+      address.location = nil
+      address.lat = 41
+      address.lng = -71
+      address.location.should == [41, -71]
+    end
+    
+    it "should allow lon then lat assignment when nil" do
+      address.location = nil
+      address.lng = -71
+      address.lat = 41
+      address.location.should == [41, -71]
+    end
+    
+  end
+end

data/spec/mongoid/geo/geo_inclusions_spec.rb

@@ -0,0 +1,20 @@
+require "mongoid/spec_helper"
+
+describe Mongoid::Criterion::Inclusion do
+
+  let(:base) do
+    Mongoid::Criteria.new(Address)
+  end
+
+  describe "#near" do
+
+    let(:criteria) do
+      base.nearSphere(:location => [ 72, -44 ])
+    end
+
+    it "adds the $near modifier to the selector" do
+      criteria.selector.should ==
+        { :location => { "$nearSphere" => [ 72, -44 ] } }
+    end
+  end
+end

data/spec/mongoid/geo/geo_inflections_spec.rb

@@ -0,0 +1,237 @@
+require "mongoid/spec_helper"
+
+describe Mongoid::Extensions::Symbol::Inflections do
+
+  let(:base) do
+    Mongoid::Criteria.new(Address)
+  end
+  
+  describe "#nearSphere" do
+    let(:criteria) do
+      base.where(:location.nearSphere => [ 72, -44 ])
+    end
+
+    it "returns a selector matching a ne clause" do
+      criteria.selector.should ==
+        { :location => { "$nearSphere" => [ 72, -44 ] } }
+    end
+  end
+
+  describe "#nearMax" do
+  
+    let(:criteria) do
+      base.where(:location.nearMax => [{:latitude => 72, :longitude => -44 }, 5])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$near" => [ 72, -44 ], "$maxDistance" => 5 } }
+    end
+  end  
+
+  # this could be used to optimize calculation speed
+  describe "#nearMax sphere and flat" do
+  
+    let(:criteria) do
+      base.where(:location.nearMax(:flat, :sphere) => [{:lat => 72, :lng => -44 }, 5])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$near" => [ 72, -44 ], "$maxDistanceSphere" => 5 } }
+    end
+  end  
+
+  # this could be used to optimize calculation speed
+  describe "#nearMax hash values" do
+
+    let(:point) do
+      b = (Struct.new :lat, :lng).new
+      b.lat = 72
+      b.lng = -44
+      b
+    end
+  
+    let(:criteria) do
+      base.where(:location.nearMax => {:point => point, :distance =>  5})
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$near" => [ 72, -44 ], "$maxDistance" => 5 } }
+    end
+  end  
+
+  # this could be used to optimize calculation speed
+  describe "#nearMax hash values" do
+    let(:point_distance) do
+      b = (Struct.new :point, :distance).new
+      b.point = [72, -44]
+      b.distance = 5
+      b
+    end
+  
+    let(:criteria) do
+      base.where(:location.nearMax => point_distance)
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$near" => [ 72, -44 ], "$maxDistance" => 5 } }
+    end
+  end  
+
+  describe "#nearMax sphere" do
+  
+    let(:criteria) do
+      base.where(:location.nearMax(:sphere) => [[ 72, -44 ], 5])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$nearSphere" => [ 72, -44 ], "$maxDistanceSphere" => 5 } }
+    end
+  end  
+
+  
+  describe "#withinBox" do  
+    let(:point_a) { [ 72, -44 ] }
+    let(:point_b) { [ 71, -45 ] }      
+    
+    let(:criteria) do
+      base.where(:location.withinBox => [point_a, point_b])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$box" => [point_a, point_b] } } }
+    end
+  end  
+
+  describe "#withinBox, one hash point" do  
+    let(:point_a) do 
+      {:lat => 72, :lng => -44 }
+    end
+    let(:point_b) { [ 71, -45 ] }      
+    
+    let(:criteria) do
+      base.where(:location.withinBox => [point_a, point_b])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$box" => [[72, -44], [ 71, -45 ]] } } }
+    end
+  end  
+
+
+  describe "#withinBox hash box" do  
+    let(:point_a) { [ 72, -44 ] }
+    let(:point_b) { [ 71, -45 ] }      
+    let(:box) do
+      {:lower_left => point_a, :upper_right => point_b}
+    end
+    
+    let(:criteria) do
+      base.where(:location.withinBox => box)
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$box" => [point_a, point_b] } } }
+    end
+  end  
+
+  describe "#withinBox Struct box" do  
+    let(:point_a) { [ 72, -44 ] }
+    let(:point_b) { [ 71, -45 ] }      
+    let(:box) do
+      b = (Struct.new :lower_left, :upper_right).new
+      b.lower_left = point_a
+      b.upper_right = point_b
+      b
+    end
+    
+    let(:criteria) do
+      base.where(:location.withinBox => box)
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$box" => [point_a, point_b] } } }
+    end
+  end  
+
+  describe "#withinBox sphere" do  
+    let(:point_a) { [ 72, -44 ] }
+    let(:point_b) { [ 71, -45 ] }      
+    
+    let(:criteria) do
+      base.where(:location.withinBox(:sphere) => [point_a, point_b])
+    end
+  
+    it "adds the $near and $maxDistance modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$boxSphere" => [point_a, point_b] } } }
+    end
+  end  
+  
+  describe "#withinCenter" do  
+    let(:criteria) do
+      base.where(:location.withinCenter => [[ 72, -44 ], 5])
+    end
+  
+    it "adds the $within and $center modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$center" => [[ 72, -44 ], 5]} }}
+    end
+  end  
+
+  describe "#withinCenter hash circle" do  
+    let(:center) { [ 72, -44 ] }
+    let(:circle) do
+      {:center => center, :radius => 5}
+    end
+    
+    let(:criteria) do
+      base.where(:location.withinCenter => circle)
+    end
+  
+    it "adds the $within and $center modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$center" => [[ 72, -44 ], 5]} }}
+    end
+  end  
+
+  describe "#withinBox Struct circle" do  
+    let(:center) { [ 71, -45 ] }      
+    let(:box) do
+      b = (Struct.new :center, :radius).new
+      b.center = center
+      b.radius = radius
+      b
+    end
+    
+    let(:criteria) do
+      base.where(:location.withinCenter => [[ 72, -44 ], 5])
+    end
+  
+    it "adds the $within and $center modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$center" => [[ 72, -44 ], 5]} }}
+    end
+  end  
+
+
+  describe "#withinCenter sphere" do  
+    let(:criteria) do
+      base.where(:location.withinCenter(:sphere) => [[ 72, -44 ], 5])
+    end
+  
+    it "adds the $within and $center modifiers to the selector" do
+      criteria.selector.should ==
+        { :location => { "$within" => { "$centerSphere" => [[ 72, -44 ], 5]} }}
+    end
+  end  
+
+end

data/spec/mongoid/geo/geo_near_spec.rb

@@ -0,0 +1,58 @@
+require "mongoid/spec_helper"
+
+Address.collection.create_index([['location', Mongo::GEO2D]], :min => -180, :max => 180)
+
+describe Mongoid::Geo::Near do
+
+  let(:address) do
+    Address.new        
+  end  
+  
+  before(:each) do
+    Address.create(:location => [45, 11], :city => 'Munich')
+    Address.create(:location => [46, 12], :city => 'Berlin')
+  end
+
+  describe "geoNear" do
+    it "should work with specifying specific center and different location attribute on collction" do
+      address.location = "23.5, -47"
+      Address.geoNear(address.location, :location).first.location[0].should == 45
+    end
+    
+    it "should assume same attribute searched on both center instance and collection" do
+      address.location = "23.5, -47"
+      Address.geoNear(address, :location).first.location[0].should == 45
+    end
+
+    describe 'option :num' do
+      it "should limit number of results to 1" do
+        address.location = "23.5, -47"
+        Address.geoNear(address, :location, :num => 1).size.should == 1
+      end
+    end
+    
+    describe 'option :maxDistance' do
+      it "should limit on maximum distance" do
+        address.location = "45.1, 11.1"
+        # db.runCommand({ geoNear : "points", near :[45.1, 11.1]}).results;
+        # dis: is 0.14141869255648362  and  1.2727947855285668 
+        Address.geoNear(address, :location, :maxDistance => 0.2).size.should == 1
+      end
+    end
+    
+    describe 'option :distanceMultiplier' do
+      it "should multiply returned distance with multiplier" do
+        address.location = "45.1, 11.1"
+        Address.geoNear(address, :location, :distanceMultiplier => 4).map(&:distance).first.should > 0
+      end
+    end
+    
+    describe 'option :query' do
+      it "should filter using extra query option" do
+        address.location = "45.1, 11.1"
+        # two record in the collection, only one's city is Munich
+        Address.geoNear(address, :location, :query => {:city => 'Munich'}).size.should == 1
+      end
+    end
+  end
+end