postgres_ext 0.3.1 → 0.4.0

data/spec/columns/ranges/int8range_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+describe 'int8range column' do
+  let!(:adapter) { ActiveRecord::Base.connection }
+  let!(:int8_range_column) { ActiveRecord::ConnectionAdapters::PostgreSQLColumn.new 'field', nil, 'int8range'}
+
+  describe '#type_class' do
+    it 'converts an end-inclusive PostgreSQL integer range to a Ruby range' do
+      int8_range_column.type_cast('[0,4]').should eq 0..4
+    end
+
+    it 'converts an end-exclusive PostgreSQL integer range to a Ruby range' do
+      int8_range_column.type_cast('[0,4)').should eq 0...4
+    end
+
+    it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+      int8_range_column.type_cast('(,4)').should eq -(1.0/0.0)...4
+      int8_range_column.type_cast('[0,)').should eq 0..(1.0/0.0)
+    end
+  end
+
+  describe 'int8 range to SQL statment conversion' do
+    it 'returns an end-inclusive PostgreSQL range' do
+      value = int8_range_column.type_cast('[0,4]')
+      adapter.type_cast(value, int8_range_column).should eq '[0,4]'
+    end
+    it 'returns an end-exclusive PostgreSQL range' do
+      value = int8_range_column.type_cast('[0,4)')
+      adapter.type_cast(value, int8_range_column).should eq '[0,4)'
+    end
+    it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+      value = int8_range_column.type_cast('(,4)')
+      adapter.type_cast(value, int8_range_column).should eq '(,4)'
+      value = int8_range_column.type_cast('[0,)')
+      adapter.type_cast(value, int8_range_column).should eq '[0,)'
+    end
+  end
+end

data/spec/columns/ranges/numrange_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+describe 'numrange column' do
+  let!(:adapter) { ActiveRecord::Base.connection }
+  let!(:numeric_range_column) { ActiveRecord::ConnectionAdapters::PostgreSQLColumn.new 'field', nil, 'numrange'}
+  describe '#type_class' do
+    it 'converts an end-inclusive PostgreSQL integer range to a Ruby range' do
+      numeric_range_column.type_cast('[0,4.1]').should eq 0.0..4.1
+    end
+
+    it 'converts an end-exclusive PostgreSQL integer range to a Ruby range' do
+      numeric_range_column.type_cast('[0,4)').should eq 0.0...4.0
+    end
+
+    it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+      numeric_range_column.type_cast('(,4)').should eq -(1.0/0.0)...4.0
+      numeric_range_column.type_cast('[0,)').should eq 0.0..(1.0/0.0)
+    end
+  end
+
+  describe 'Numeric range to SQL statment conversion' do
+    it 'returns an end-inclusive PostgreSQL range' do
+      value = numeric_range_column.type_cast('[0,4.0]')
+      adapter.type_cast(value, numeric_range_column).should eq '[0.0,4.0]'
+    end
+    it 'returns an end-exclusive PostgreSQL range' do
+      value = numeric_range_column.type_cast('[0,4.0)')
+      adapter.type_cast(value, numeric_range_column).should eq '[0.0,4.0)'
+    end
+    it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+      value = numeric_range_column.type_cast('(,4.0)')
+      adapter.type_cast(value, numeric_range_column).should eq '(,4.0)'
+      value = numeric_range_column.type_cast('[0,)')
+      adapter.type_cast(value, numeric_range_column).should eq '[0.0,)'
+    end
+  end
+end

data/spec/columns/ranges/tsrange_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+describe 'tsrange column' do
+  let!(:adapter) { ActiveRecord::Base.connection }
+  let!(:ts_range_column) { ActiveRecord::ConnectionAdapters::PostgreSQLColumn.new 'field', nil, 'tsrange'}
+
+  describe '#type_class' do
+    it 'converts an end-inclusive PostgreSQL integer range to a Ruby range' do
+      ts_range_column.type_cast('[2011-01-01 12:34:00,2012-01-31 08:00:01]').should eq Time.new(2011, 01, 01, 12, 34)..Time.new(2012, 01, 31, 8, 0, 1)
+    end
+
+    it 'converts an end-exclusive PostgreSQL integer range to a Ruby range' do
+      ts_range_column.type_cast('[2011-01-01 12:34:00,2012-01-31 08:00:01)').should eq Time.new(2011, 01, 01, 12, 34)...Time.new(2012, 01, 31, 8, 0, 1)
+    end
+
+    #it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+    ## Cannot have a range from -Infinity to a ts
+    ## ts_range_column.type_cast('(,2011-01-01)').should eq -(1.0/0.0)...4
+    #ts_range_column.type_cast('[2011-01-01 12:34:00,)').should eq Time.new(2011,01,01, 12, 34)..(1.0/0.0)
+    #end
+  end
+
+  describe 'Time range to SQL statment conversion' do
+    it 'returns an end-inclusive PostgreSQL range' do
+      value = ts_range_column.type_cast('[2011-01-01 12:34:00,2012-01-31 08:00:01]')
+      adapter.type_cast(value, ts_range_column).should eq '[2011-01-01 12:34:00,2012-01-31 08:00:01]'
+    end
+    it 'returns an end-exclusive PostgreSQL range' do
+      value = ts_range_column.type_cast('[2011-01-01 12:34:00,2012-01-31 08:00:01)')
+      adapter.type_cast(value, ts_range_column).should eq '[2011-01-01 12:34:00,2012-01-31 08:00:01)'
+    end
+    #it 'converts an infinite PostgreSQL integer range to a Ruby range' do
+    #value = ts_range_column.type_cast('[2011-01-01 12:34:00,)')
+    #adapter.type_cast(value, ts_range_column).should eq '[2011-01-01 12:34:00,)'
+    #end
+  end
+end

data/spec/dummy/app/models/person.rb

@@ -1,3 +1,3 @@
 class Person < ActiveRecord::Base
-  attr_accessible :ip, :tags
+  attr_accessible :ip, :tags, :tag_ids, :biography, :lucky_number, :int_range
 end

data/spec/dummy/config/application.rb

@@ -27,7 +27,7 @@ module Dummy
 
     # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
     # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
-    # config.time_zone = 'Central Time (US & Canada)'
+    config.time_zone = 'Eastern Time (US & Canada)'
 
     # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
     # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]

data/spec/dummy/db/migrate/20120501163758_create_people.rb

@@ -7,6 +7,7 @@ class CreatePeople < ActiveRecord::Migration
       t.string  :tags, :array => true
       t.text    :biography
       t.integer :lucky_number
+      t.numrange :num_range
 
       t.timestamps
     end

data/spec/dummy/db/schema.rb

@@ -14,14 +14,15 @@
 ActiveRecord::Schema.define(:version => 20120501163758) do
 
   create_table "people", :force => true do |t|
-    t.inet     "ip"
-    t.cidr     "subnet"
-    t.integer  "tag_ids",                      :array => true
-    t.string   "tags",                         :array => true
-    t.text     "biography"
-    t.integer  "lucky_number"
-    t.datetime "created_at",   :null => false
-    t.datetime "updated_at",   :null => false
+    t.inet          "ip"
+    t.cidr          "subnet"
+    t.integer       "tag_ids",                      :array => true
+    t.string        "tags",                         :array => true
+    t.text          "biography"
+    t.integer       "lucky_number"
+    t.num_rang      "num_range"
+    t.datetime      "created_at",   :null => false
+    t.datetime      "updated_at",   :null => false
   end
 
 end

data/spec/migrations/array_spec.rb

@@ -82,6 +82,26 @@ describe 'Array migrations' do
     end
   end
 
+  context 'Change Column' do
+    after { connection.drop_table :data_types }
+    it 'updates the column definitions' do
+      lambda do
+        connection.create_table :data_types do |t|
+          t.integer :array_1, :array => true, :default => []
+        end
+
+        connection.change_column :data_types, :array_1, :integer, :array => true, :default => [], :null => false
+      end.should_not raise_exception
+
+      columns = connection.columns(:data_types)
+
+      array_1 = columns.detect { |c| c.name == 'array_1'}
+      array_1.sql_type.should eq 'integer[]'
+      array_1.default.should  eq []
+      array_1.null.should     be_false
+    end
+  end
+
   context 'Default Values' do
     describe 'String defaults' do
       after { connection.drop_table :default_strings }

data/spec/migrations/ranges/daterange_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'daterange migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an daterange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.daterange :date_range_1
+        t.daterange :date_range_2, :date_range_3
+        t.column :date_range_4, :daterange
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    date_range_1 = columns.detect { |c| c.name == 'date_range_1'}
+    date_range_2 = columns.detect { |c| c.name == 'date_range_2'}
+    date_range_3 = columns.detect { |c| c.name == 'date_range_3'}
+    date_range_4 = columns.detect { |c| c.name == 'date_range_4'}
+
+    date_range_1.sql_type.should eq 'daterange'
+    date_range_2.sql_type.should eq 'daterange'
+    date_range_3.sql_type.should eq 'daterange'
+    date_range_4.sql_type.should eq 'daterange'
+  end
+end

data/spec/migrations/ranges/int4range_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'int4range migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an numrange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.int4range :range_1
+        t.int4range :range_2, :range_3
+        t.column :range_4, :int4range
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    range_1 = columns.detect { |c| c.name == 'range_1'}
+    range_2 = columns.detect { |c| c.name == 'range_2'}
+    range_3 = columns.detect { |c| c.name == 'range_3'}
+    range_4 = columns.detect { |c| c.name == 'range_4'}
+
+    range_1.sql_type.should eq 'int4range'
+    range_2.sql_type.should eq 'int4range'
+    range_3.sql_type.should eq 'int4range'
+    range_4.sql_type.should eq 'int4range'
+  end
+end

data/spec/migrations/ranges/int8range_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'int8range migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an numrange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.int8range :range_1
+        t.int8range :range_2, :range_3
+        t.column :range_4, :int8range
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    range_1 = columns.detect { |c| c.name == 'range_1'}
+    range_2 = columns.detect { |c| c.name == 'range_2'}
+    range_3 = columns.detect { |c| c.name == 'range_3'}
+    range_4 = columns.detect { |c| c.name == 'range_4'}
+
+    range_1.sql_type.should eq 'int8range'
+    range_2.sql_type.should eq 'int8range'
+    range_3.sql_type.should eq 'int8range'
+    range_4.sql_type.should eq 'int8range'
+  end
+end

data/spec/migrations/ranges/numrange_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'numrange migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an numrange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.numrange :num_range_1
+        t.numrange :num_range_2, :num_range_3
+        t.column :num_range_4, :numrange
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    num_range_1 = columns.detect { |c| c.name == 'num_range_1'}
+    num_range_2 = columns.detect { |c| c.name == 'num_range_2'}
+    num_range_3 = columns.detect { |c| c.name == 'num_range_3'}
+    num_range_4 = columns.detect { |c| c.name == 'num_range_4'}
+
+    num_range_1.sql_type.should eq 'numrange'
+    num_range_2.sql_type.should eq 'numrange'
+    num_range_3.sql_type.should eq 'numrange'
+    num_range_4.sql_type.should eq 'numrange'
+  end
+end

data/spec/migrations/ranges/tsrange_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'tsrange migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an tsrange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.tsrange :ts_range_1
+        t.tsrange :ts_range_2, :ts_range_3
+        t.column :ts_range_4, :tsrange
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    ts_range_1 = columns.detect { |c| c.name == 'ts_range_1'}
+    ts_range_2 = columns.detect { |c| c.name == 'ts_range_2'}
+    ts_range_3 = columns.detect { |c| c.name == 'ts_range_3'}
+    ts_range_4 = columns.detect { |c| c.name == 'ts_range_4'}
+
+    ts_range_1.sql_type.should eq 'tsrange'
+    ts_range_2.sql_type.should eq 'tsrange'
+    ts_range_3.sql_type.should eq 'tsrange'
+    ts_range_4.sql_type.should eq 'tsrange'
+  end
+end

data/spec/migrations/ranges/tstzrange_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe 'tstzrange migrations' do
+  let!(:connection) { ActiveRecord::Base.connection }
+  after { connection.drop_table :data_types }
+
+  it 'creates an tstzrange column' do
+    lambda do
+      connection.create_table :data_types do |t|
+        t.tstzrange :tstz_range_1
+        t.tstzrange :tstz_range_2, :tstz_range_3
+        t.column :tstz_range_4, :tstzrange
+      end
+    end.should_not raise_exception
+
+    columns = connection.columns(:data_types)
+    tstz_range_1 = columns.detect { |c| c.name == 'tstz_range_1'}
+    tstz_range_2 = columns.detect { |c| c.name == 'tstz_range_2'}
+    tstz_range_3 = columns.detect { |c| c.name == 'tstz_range_3'}
+    tstz_range_4 = columns.detect { |c| c.name == 'tstz_range_4'}
+
+    tstz_range_1.sql_type.should eq 'tstzrange'
+    tstz_range_2.sql_type.should eq 'tstzrange'
+    tstz_range_3.sql_type.should eq 'tstzrange'
+    tstz_range_4.sql_type.should eq 'tstzrange'
+  end
+end

data/spec/models/ranges/daterange_spec.rb

@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe 'Models with daterange columns' do
+  let!(:adapter) { ActiveRecord::Base.connection }
+
+  context 'no default value, range' do
+    before do
+      adapter.create_table :date_rangers, :force => true do |t|
+        t.daterange :best_estimate
+      end
+      class DateRanger < ActiveRecord::Base
+        attr_accessible :best_estimate
+      end
+    end
+
+    after do
+      adapter.drop_table :date_rangers
+      Object.send(:remove_const, :DateRanger)
+    end
+
+    describe '#create' do
+      it 'creates an record when there is no assignment' do
+        range = DateRanger.create()
+        range.reload
+        range.best_estimate.should eq nil
+      end
+
+      it 'creates an record with a range' do
+        date_range = Date.new(2011, 01, 01)..Date.new(2012, 01, 31)
+        range = DateRanger.create( :best_estimate => date_range)
+        range.reload
+        range.best_estimate.should eq Date.new(2011, 01, 01)...Date.new(2012, 02, 01)
+      end
+    end
+
+    describe 'range assignment' do
+      it 'updates an record with an range string' do
+        date_range = Date.new(2011, 01, 01)..Date.new(2012, 01, 31)
+        new_date_range = Date.new(2012, 01, 01)...Date.new(2012, 02, 01)
+        range = DateRanger.create( :best_estimate => date_range)
+        range.best_estimate = new_date_range
+        range.save
+
+        range.reload
+        range.best_estimate.should eq new_date_range
+      end
+
+      it 'converts empty strings to nil' do
+        range = DateRanger.create
+        range.best_estimate = ''
+        range.save
+
+        range.reload
+        range.best_estimate.should eq nil
+      end
+    end
+  end
+
+  context 'default value, date range' do
+    before do
+      adapter.create_table :date_default_rangers, :force => true do |t|
+        t.daterange :best_estimate, :default => Date.new(2011, 01, 01)..Date.new(2011, 01, 31)
+      end
+      class DateDefaultRanger < ActiveRecord::Base
+        attr_accessible :best_estimate
+      end
+    end
+
+    after do
+      adapter.drop_table :date_default_rangers
+      Object.send(:remove_const, :DateDefaultRanger)
+    end
+
+    describe '#create' do
+      it 'creates an record when there is no assignment' do
+        range = DateDefaultRanger.create()
+        range.reload
+        range.best_estimate.should eq Date.new(2011, 01, 01)...Date.new(2011, 02, 01)
+      end
+
+      it 'creates an record with a range' do
+        range = DateDefaultRanger.create(:best_estimate => Date.new(2012, 01, 01)..Date.new(2012, 12, 31))
+        range.reload
+        range.best_estimate.should eq Date.new(2012, 01, 01)...Date.new(2013, 01, 01)
+      end
+    end
+  end
+end