spatial_adapter 0.2.0

data/spec/postgis/migration_spec.rb

@@ -0,0 +1,218 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+class MigratedGeometryModel < ActiveRecord::Base
+end
+
+describe "Spatially-enabled Migrations" do
+  before :each do
+    postgis_connection
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  describe "creating tables" do
+    after :each do
+      @connection.drop_table "migrated_geometry_models"
+    end
+    
+    SpatialAdapter.geometry_data_types.keys.each do |type|
+      it "should create #{type.to_s} columns" do
+        ActiveRecord::Schema.define do
+          create_table :migrated_geometry_models, :force => true do |t|
+            t.integer :extra
+            t.send(type, :geom)
+          end
+        end
+
+        geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+        geom_column.should be_a(SpatialAdapter::SpatialColumn)
+        geom_column.geometry_type.should == type
+        geom_column.type.should == :geometry
+        geom_column.with_z.should == false
+        geom_column.with_m.should == false
+        geom_column.srid.should == -1
+      end
+    end
+    
+    it "should create 3d (xyz) geometry columns" do
+      ActiveRecord::Schema.define do
+        create_table :migrated_geometry_models, :force => true do |t|
+          t.integer :extra
+          t.point   :geom, :with_z => true
+        end
+      end
+      
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.with_z.should == true
+      geom_column.with_m.should == false
+      geom_column.srid.should == -1
+    end
+    
+    
+    it "should create 3d (xym) geometry columns" do
+      ActiveRecord::Schema.define do
+        create_table :migrated_geometry_models, :force => true do |t|
+          t.integer :extra
+          t.point   :geom, :with_m => true
+        end
+      end
+      
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :point
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == false
+      geom_column.with_m.should == true
+      geom_column.srid.should == -1
+    end
+    
+    
+    it "should create 4d (xyzm) geometry columns" do
+      ActiveRecord::Schema.define do
+        create_table :migrated_geometry_models, :force => true do |t|
+          t.integer :extra
+          t.point   :geom, :with_z => true, :with_m => true
+        end
+      end
+      
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :point
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == true
+      geom_column.with_m.should == true
+      geom_column.srid.should == -1
+    end
+    
+
+    it "should create GEOMETRY columns with specified SRID" do
+      ActiveRecord::Schema.define do
+        create_table :migrated_geometry_models, :force => true do |t|
+          t.integer :extra
+          t.geometry :geom, :srid => 4326
+        end
+      end
+
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :geometry
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == false
+      geom_column.with_m.should == false
+      geom_column.srid.should == 4326
+    end
+  end
+
+  describe "adding columns" do
+    before :each do
+      ActiveRecord::Schema.define do
+        create_table :migrated_geometry_models, :force => true do |t|
+          t.integer :extra
+        end
+      end
+    end
+    
+    after :each do
+      @connection.drop_table "migrated_geometry_models"
+    end
+
+    SpatialAdapter.geometry_data_types.keys.each do |type|
+      it "should add #{type.to_s} columns" do
+        ActiveRecord::Schema.define do
+          add_column :migrated_geometry_models, :geom, type
+        end
+
+        geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+        geom_column.should be_a(SpatialAdapter::SpatialColumn)
+        geom_column.geometry_type.should == type
+        geom_column.type.should == :geometry
+        geom_column.with_z.should == false
+        geom_column.with_m.should == false
+        geom_column.srid.should == -1
+      end
+    end
+
+    it "should add 3d (xyz) geometry columns" do
+      ActiveRecord::Schema.define do
+        add_column :migrated_geometry_models, :geom, :point, :with_z => true
+      end
+
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.with_z.should == true
+      geom_column.with_m.should == false
+      geom_column.srid.should == -1
+    end
+
+
+    it "should add 3d (xym) geometry columns" do
+      ActiveRecord::Schema.define do
+        add_column :migrated_geometry_models, :geom, :point, :with_m => true
+      end
+
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :point
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == false
+      geom_column.with_m.should == true
+      geom_column.srid.should == -1
+    end
+
+
+    it "should add 4d (xyzm) geometry columns" do
+      ActiveRecord::Schema.define do
+        add_column :migrated_geometry_models, :geom, :point, :with_z => true, :with_m => true
+      end
+
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :point
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == true
+      geom_column.with_m.should == true
+      geom_column.srid.should == -1
+    end
+
+    it "should add GEOMETRY columns with specified SRID" do
+      ActiveRecord::Schema.define do
+        add_column :migrated_geometry_models, :geom, :geometry, :srid => 4326
+      end
+
+      geom_column = @connection.columns(:migrated_geometry_models).select{|c| c.name == 'geom'}.first
+      geom_column.should be_a(SpatialAdapter::SpatialColumn)
+      geom_column.geometry_type.should == :geometry
+      geom_column.type.should == :geometry
+      geom_column.with_z.should == false
+      geom_column.with_m.should == false
+      geom_column.srid.should == 4326
+    end
+  end
+  
+  describe "removing columns" do
+    before :each do
+    end
+    
+    after :each do
+      @connection.drop_table "migrated_geometry_models"
+    end
+
+    SpatialAdapter.geometry_data_types.keys.each do |type|
+      it "should remove #{type.to_s} columns using DropGeometryColumn" do
+        ActiveRecord::Schema.define do
+          create_table :migrated_geometry_models, :force => true do |t|
+            t.integer :extra
+            t.send(type, :geom)
+          end
+        end
+
+        @connection.should_receive(:execute).with(/DropGeometryColumn(.*migrated_geometry_models.*geom)/)
+        ActiveRecord::Schema.define do
+          remove_column :migrated_geometry_models, :geom
+        end
+        @connection.should_receive(:execute).with(anything())
+      end
+    end
+  end
+  
+end

data/spec/postgis/models_spec.rb

@@ -0,0 +1,136 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+require 'db/postgis_raw'
+require 'models/common'
+
+describe "Spatially-enabled Models" do
+  before :each do
+    postgis_connection
+    @connection = ActiveRecord::Base.connection
+  end
+  
+  describe "inserting records" do
+    it 'should save Point objects' do
+      model = PointModel.new(:extra => 'test', :geom => GeometryFactory.point)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.point.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save LineString objects' do
+      model = LineStringModel.new(:extra => 'test', :geom => GeometryFactory.line_string)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.line_string.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save Polygon objects' do
+      model = PolygonModel.new(:extra => 'test', :geom => GeometryFactory.polygon)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.polygon.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save MultiPoint objects' do
+      model = MultiPointModel.new(:extra => 'test', :geom => GeometryFactory.multi_point)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.multi_point.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save MultiLineString objects' do
+      model = MultiLineStringModel.new(:extra => 'test', :geom => GeometryFactory.multi_line_string)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.multi_line_string.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save MultiPolygon objects' do
+      model = MultiPolygonModel.new(:extra => 'test', :geom => GeometryFactory.multi_polygon)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.multi_polygon.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save GeometryCollection objects' do
+      model = GeometryCollectionModel.new(:extra => 'test', :geom => GeometryFactory.geometry_collection)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.geometry_collection.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save Geometry objects' do
+      model = GeometryModel.new(:extra => 'test', :geom => GeometryFactory.point)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.point.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save 3D Point (with Z coord) objects' do
+      model = PointzModel.new(:extra => 'test', :geom => GeometryFactory.pointz)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.pointz.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save 3D Point (with M coord) objects' do
+      model = PointmModel.new(:extra => 'test', :geom => GeometryFactory.pointm)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.pointm.as_hex_ewkb))
+      model.save.should == true
+    end
+  
+    it 'should save 4D Point objects' do
+      model = Point4Model.new(:extra => 'test', :geom => GeometryFactory.point4)
+      @connection.should_receive(:select_value).with(Regexp.new(GeometryFactory.point4.as_hex_ewkb))
+      model.save.should == true
+    end
+  end
+
+  describe "finding records" do
+    it 'should retrieve Point objects' do
+      model = PointModel.create(:extra => 'test', :geom => GeometryFactory.point)
+      PointModel.find(model.id).geom.should == GeometryFactory.point
+    end
+  
+    it 'should retrieve LineString objects' do
+      model = LineStringModel.create(:extra => 'test', :geom => GeometryFactory.line_string)
+      LineStringModel.find(model.id).geom.should == GeometryFactory.line_string
+    end
+  
+    it 'should retrieve Polygon objects' do
+      model = PolygonModel.create(:extra => 'test', :geom => GeometryFactory.polygon)
+      PolygonModel.find(model.id).geom.should == GeometryFactory.polygon
+    end
+  
+    it 'should retrieve MultiPoint objects' do
+      model = MultiPointModel.create(:extra => 'test', :geom => GeometryFactory.multi_point)
+      MultiPointModel.find(model.id).geom.should == GeometryFactory.multi_point
+    end
+  
+    it 'should retrieve MultiLineString objects' do
+      model = MultiLineStringModel.create(:extra => 'test', :geom => GeometryFactory.multi_line_string)
+      MultiLineStringModel.find(model.id).geom.should == GeometryFactory.multi_line_string
+    end
+  
+    it 'should retrieve MultiPolygon objects' do
+      model = MultiPolygonModel.create(:extra => 'test', :geom => GeometryFactory.multi_polygon)
+      MultiPolygonModel.find(model.id).geom.should == GeometryFactory.multi_polygon
+    end
+  
+    it 'should retrieve GeometryCollection objects' do
+      model = GeometryCollectionModel.create(:extra => 'test', :geom => GeometryFactory.geometry_collection)
+      GeometryCollectionModel.find(model.id).geom.should == GeometryFactory.geometry_collection
+    end
+  
+    it 'should retrieve Geometry objects' do
+      model = GeometryModel.create(:extra => 'test', :geom => GeometryFactory.point)
+      GeometryModel.find(model.id).geom.should == GeometryFactory.point
+    end
+  
+    it 'should retrieve 3D Point (with Z coord) objects' do
+      model = PointzModel.create(:extra => 'test', :geom => GeometryFactory.pointz)
+      PointzModel.find(model.id).geom.should == GeometryFactory.pointz
+    end
+  
+    it 'should retrieve 3D Point (with M coord) objects' do
+      model = PointmModel.create(:extra => 'test', :geom => GeometryFactory.pointm)
+      PointmModel.find(model.id).geom.should == GeometryFactory.pointm
+    end
+  
+    it 'should retrieve 4D Point objects' do
+      model = Point4Model.create(:extra => 'test', :geom => GeometryFactory.point4)
+      Point4Model.find(model.id).geom.should == GeometryFactory.point4
+    end
+  end
+end
+

data/spec/postgis/schema_dumper_spec.rb

@@ -0,0 +1,58 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe "Spatially-enabled Schema Dumps" do
+  before :all do
+    postgis_connection
+    @connection = ActiveRecord::Base.connection
+
+    # Create a new table
+    ActiveRecord::Schema.define do
+      create_table :migrated_geometry_models, :force => true do |t|
+        t.integer :extra
+        t.point   :geom, :with_m => true, :with_z => true, :srid => 4326
+      end
+      add_index :migrated_geometry_models, :geom, :spatial => true, :name => 'test_spatial_index'
+    end
+
+    File.open('schema.rb', "w") do |file|
+      ActiveRecord::SchemaDumper.dump(@connection, file)
+    end
+    
+    # Drop the original table
+    @connection.execute "drop table migrated_geometry_models"
+    
+    # Load the dumped schema
+    load('schema.rb')
+  end
+  
+  after :all do
+    # delete the schema file
+    File.delete('schema.rb')
+
+    # Drop the new table
+    @connection.execute "drop table migrated_geometry_models"
+  end
+  
+  it "should preserve spatial attributes of tables" do
+    columns = @connection.columns("migrated_geometry_models")
+    
+    columns.should have(3).items
+    geom_column = columns.select{|c| c.name == 'geom'}.first
+    geom_column.should be_a(SpatialAdapter::SpatialColumn)
+    geom_column.geometry_type.should == :point
+    geom_column.type.should == :geometry
+    geom_column.with_z.should == true
+    geom_column.with_m.should == true
+    geom_column.srid.should == 4326
+  end
+  
+  it "should preserve spatial indexes" do
+    indexes = @connection.indexes("migrated_geometry_models")
+    
+    indexes.should have(1).item
+    
+    indexes.first.name.should == 'test_spatial_index'
+    indexes.first.columns.should == ["geom"]
+    indexes.first.spatial.should == true
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,73 @@
+require 'rubygems'
+require 'spec'
+gem 'activerecord', '=2.3.5'
+
+$:.unshift((File.join(File.dirname(__FILE__), '..', 'lib')))
+require 'spatial_adapter'
+
+include GeoRuby::SimpleFeatures
+
+# Don't output migration logging
+ActiveRecord::Migration.verbose = false
+
+def postgis_connection
+  ActiveRecord::Base.establish_connection(
+    :adapter => 'postgresql',
+    :database => 'spatial_adapter'
+  )
+  # Turn off those annoying NOTICE messages
+  ActiveRecord::Base.connection.execute 'set client_min_messages = warning'
+end
+
+def mysql_connection
+  ActiveRecord::Base.establish_connection(
+    :adapter => 'mysql',
+    :database => 'spatial_adapter',
+    :username => 'root',
+    :host => 'localhost'
+  )
+end
+
+class GeometryFactory
+  class << self
+    def point
+      Point.from_x_y(1, 2, 4326)
+    end
+    
+    def line_string
+      LineString.from_coordinates([[1.4,2.5],[1.5,6.7]], 4326)
+    end
+    
+    def polygon
+      Polygon.from_coordinates([[[12.4,-45.3],[45.4,41.6],[4.456,1.0698],[12.4,-45.3]],[[2.4,5.3],[5.4,1.4263],[14.46,1.06],[2.4,5.3]]], 4326)
+    end
+    
+    def multi_point
+      MultiPoint.from_coordinates([[12.4,-123.3],[-65.1,123.4],[123.55555555,123]], 4326)
+    end
+    
+    def multi_line_string
+      MultiLineString.from_line_strings([LineString.from_coordinates([[1.5,45.2],[-54.12312,-0.012]]),LineString.from_coordinates([[1.5,45.2],[-54.12312,-0.012],[45.123,123.3]])], 4326)
+    end
+    
+    def multi_polygon
+      MultiPolygon.from_polygons([Polygon.from_coordinates([[[12.4,-45.3],[45.4,41.6],[4.456,1.0698],[12.4,-45.3]],[[2.4,5.3],[5.4,1.4263],[14.46,1.06],[2.4,5.3]]]),Polygon.from_coordinates([[[0,0],[4,0],[4,4],[0,4],[0,0]],[[1,1],[3,1],[3,3],[1,3],[1,1]]])], 4326)
+    end
+    
+    def geometry_collection
+      GeometryCollection.from_geometries([Point.from_x_y(4.67,45.4),LineString.from_coordinates([[5.7,12.45],[67.55,54]])], 4326)
+    end
+    
+    def pointz
+      Point.from_x_y_z(1, 2, 3, 4326)
+    end
+    
+    def pointm
+      Point.from_x_y_m(1, 2, 3, 4326)
+    end
+      
+    def point4
+      Point.from_x_y_z_m(1, 2, 3, 4, 4326)
+    end
+  end
+end