sequel 3.21.0 → 3.41.0

data/spec/extensions/null_dataset_spec.rb

@@ -0,0 +1,85 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
+
+describe "null_dataset extension" do
+  before do
+    @db = Sequel::mock(:fetch=>{:id=>1}, :autoid=>1, :numrows=>1, :columns=>[:id])
+    @ds = @db[:table].nullify
+    @i = 0
+    @pr = proc{|*a| @i += 1}
+  end
+  after do
+    @db.sqls.should == [] unless @skip_check
+  end
+
+  it "should make each be a noop" do
+    @ds.each(&@pr).should equal(@ds)
+    @i.should == 0
+  end
+
+  it "should make fetch_rows be a noop" do
+    @ds.fetch_rows("SELECT 1", &@pr).should == nil
+    @i.should == 0
+  end
+
+  it "should make insert be a noop" do
+    @ds.insert(1).should == nil
+  end
+
+  it "should make update be a noop" do
+    @ds.update(:a=>1).should == 0
+  end
+
+  it "should make delete be a noop" do
+    @ds.delete.should == 0
+  end
+
+  it "should make truncate be a noop" do
+    @ds.truncate.should == nil
+  end
+
+  it "should make execute_* be a noop" do
+    @ds.send(:execute_ddl,'FOO').should == nil
+    @ds.send(:execute_insert,'FOO').should == nil
+    @ds.send(:execute_dui,'FOO').should == nil
+    @ds.send(:execute,'FOO').should == nil
+  end
+
+  it "should have working columns" do
+    @skip_check = true
+    @ds.columns.should == [:id]
+    @db.sqls.should == ['SELECT * FROM table LIMIT 1']
+  end
+
+  it "should have count return 0" do
+    @ds.count.should == 0
+  end
+
+  it "should have empty return true" do
+    @ds.empty?.should == true
+  end
+
+  it "should make import a noop" do
+    @ds.import([:id], [[1], [2], [3]]).should == nil
+  end
+
+  it "should have nullify method returned modified receiver" do
+    @skip_check = true
+    ds = @db[:table]
+    ds.nullify.should_not equal(ds)
+    ds.each(&@pr)
+    @db.sqls.should == ['SELECT * FROM table']
+    @i.should == 1
+  end
+
+  it "should have null! method modify receiver" do
+    ds = @db[:table]
+    ds.nullify!.should equal(ds)
+    ds.each(&@pr)
+    @i.should == 0
+  end
+
+  it "should work with method chaining" do
+    @ds.where(:a=>1).select(:b).each(&@pr)
+    @i.should == 0
+  end
+end

data/spec/extensions/optimistic_locking_spec.rb

@@ -6,11 +6,8 @@ describe "optimistic_locking plugin" do
     end
     h = {1=>{:id=>1, :name=>'John', :lock_version=>2}}
     lv = @lv = "lock_version"
-    @c.dataset.quote_identifiers = false
-    @c.dataset.meta_def(:h){h}
-    @c.dataset.meta_def(:lv){lv}
-    @c.dataset.meta_def(:update) do |opts|
-      case update_sql(opts)
+    @c.instance_dataset.numrows = @c.dataset.numrows = proc do |sql|
+      case sql
       when /UPDATE people SET (name|#{lv}) = ('Jim'|'Bob'|\d+), (?:name|#{lv}) = ('Jim'|'Bob'|\d+) WHERE \(\(id = (\d+)\) AND \(#{lv} = (\d+)\)\)/
         name, nlv = $1 == 'name' ? [$2, $3] : [$3, $2]
         m = h[$4.to_i]
@@ -28,20 +25,6 @@ describe "optimistic_locking plugin" do
         else
           0
         end
-      else
-        puts update_sql(opts)
-      end
-    end
-    @c.dataset.instance_eval do
-      def fetch_rows(sql)
-        m = h[1].dup
-        v = m.delete(:lock_version)
-        m[lv.to_sym] = v
-        yield(m)
-      end
-    end
-    @c.dataset.meta_def(:delete) do
-      case delete_sql
       when /DELETE FROM people WHERE \(\(id = (\d+)\) AND \(#{lv} = (\d+)\)\)/
         m = h[$1.to_i]
         if m && m[lv.to_sym] == $2.to_i
@@ -51,9 +34,15 @@ describe "optimistic_locking plugin" do
           0
         end
       else
-        puts delete_sql
+        puts sql
       end
     end
+    @c.instance_dataset._fetch = @c.dataset._fetch = proc do |sql|
+      m = h[1].dup
+      v = m.delete(:lock_version)
+      m[lv.to_sym] = v
+      m
+    end
     @c.columns :id, :name, :lock_version
     @c.plugin :optimistic_locking
   end
@@ -106,6 +95,20 @@ describe "optimistic_locking plugin" do
     proc{p2.update(:name=>'Bob')}.should raise_error(Sequel::Plugins::OptimisticLocking::Error)
   end
 
+  specify "should work correctly if attempting to refresh and save again after a failed save" do
+    p1 = @c[1]
+    p2 = @c[1]
+    p1.update(:name=>'Jim')
+    begin
+      p2.update(:name=>'Bob')
+    rescue Sequel::Plugins::OptimisticLocking::Error
+      p2.refresh
+      @c.db.sqls
+      proc{p2.update(:name=>'Bob')}.should_not raise_error
+    end
+    @c.db.sqls.first.should =~ /UPDATE people SET (name = 'Bob', lock_version = 4|lock_version = 4, name = 'Bob') WHERE \(\(id = 1\) AND \(lock_version = 3\)\)/
+  end
+
   specify "should increment the lock column when #modified! even if no columns are changed" do
     p1 = @c[1]
     p1.modified!
@@ -115,7 +118,7 @@ describe "optimistic_locking plugin" do
   end
 
   specify "should not increment the lock column when the update fails" do
-    @c.dataset.meta_def(:update) { raise Exception }
+    @c.instance_dataset.meta_def(:update) { raise Exception }
     p1 = @c[1]
     p1.modified!
     lv = p1.lock_version

data/spec/extensions/pg_array_ops_spec.rb

@@ -0,0 +1,112 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
+
+describe "Sequel::Postgres::ArrayOp" do
+  before do
+    @db = Sequel.connect('mock://postgres', :quote_identifiers=>false)
+    @a = Sequel.pg_array_op(:a)
+  end
+
+  it "should support the standard mathematical operators" do
+    @db.literal(@a < @a).should == "(a < a)"
+    @db.literal(@a <= @a).should == "(a <= a)"
+    @db.literal(@a > @a).should == "(a > a)"
+    @db.literal(@a >= @a).should == "(a >= a)"
+  end
+
+  it "#[] should support subscript access" do
+    @db.literal(@a[1]).should == "a[1]"
+    @db.literal(@a[1][2]).should == "a[1][2]"
+  end
+
+  it "#any should use the ANY method" do
+    @db.literal(1=>@a.any).should == "(1 = ANY(a))"
+  end
+
+  it "#all should use the ALL method" do
+    @db.literal(1=>@a.all).should == "(1 = ALL(a))"
+  end
+
+  it "#contains should use the @> operator" do
+    @db.literal(@a.contains(:b)).should == "(a @> b)"
+  end
+
+  it "#contained_by should use the <@ operator" do
+    @db.literal(@a.contained_by(:b)).should == "(a <@ b)"
+  end
+
+  it "#overlaps should use the && operator" do
+    @db.literal(@a.overlaps(:b)).should == "(a && b)"
+  end
+
+  it "#push/concat should use the || operator in append mode" do
+    @db.literal(@a.push(:b)).should == "(a || b)"
+    @db.literal(@a.concat(:b)).should == "(a || b)"
+  end
+
+  it "#unshift should use the || operator in prepend mode" do
+    @db.literal(@a.unshift(:b)).should == "(b || a)"
+  end
+
+  it "#dims should use the array_dims function" do
+    @db.literal(@a.dims).should == "array_dims(a)"
+  end
+
+  it "#length should use the array_length function" do
+    @db.literal(@a.length).should == "array_length(a, 1)"
+    @db.literal(@a.length(2)).should == "array_length(a, 2)"
+  end
+
+  it "#length should use the array_lower function" do
+    @db.literal(@a.lower).should == "array_lower(a, 1)"
+    @db.literal(@a.lower(2)).should == "array_lower(a, 2)"
+  end
+
+  it "#to_string/join should use the array_to_string function" do
+    @db.literal(@a.to_string).should == "array_to_string(a, '', NULL)"
+    @db.literal(@a.join).should == "array_to_string(a, '', NULL)"
+    @db.literal(@a.join(':')).should == "array_to_string(a, ':', NULL)"
+    @db.literal(@a.join(':', '*')).should == "array_to_string(a, ':', '*')"
+  end
+
+  it "#unnest should use the unnest function" do
+    @db.literal(@a.unnest).should == "unnest(a)"
+  end
+
+  it "#pg_array should return self" do
+    @a.pg_array.should equal(@a)
+  end
+
+  it "Sequel.pg_array_op should return arg for ArrayOp" do
+    Sequel.pg_array_op(@a).should equal(@a)
+  end
+
+  it "should be able to turn expressions into array ops using pg_array" do
+    @db.literal(Sequel.qualify(:b, :a).pg_array.push(3)).should == "(b.a || 3)"
+    @db.literal(Sequel.function(:a, :b).pg_array.push(3)).should == "(a(b) || 3)"
+  end
+
+  it "should be able to turn literal strings into array ops using pg_array" do
+    @db.literal(Sequel.lit('a').pg_array.unnest).should == "unnest(a)"
+  end
+
+  it "should be able to turn symbols into array ops using Sequel.pg_array_op" do
+    @db.literal(Sequel.pg_array_op(:a).unnest).should == "unnest(a)"
+  end
+
+  it "should be able to turn symbols into array ops using Sequel.pg_array" do
+    @db.literal(Sequel.pg_array(:a).unnest).should == "unnest(a)"
+  end
+
+  it "should allow transforming PGArray instances into ArrayOp instances" do
+    @db.literal(Sequel.pg_array([1,2]).op.push(3)).should == "(ARRAY[1,2] || 3)"
+  end
+
+  it "should wrap array arguments in PGArrays" do
+    @db.literal(@a.contains([1, 2])).should == "(a @> ARRAY[1,2])"
+    @db.literal(@a.contained_by([1, 2])).should == "(a <@ ARRAY[1,2])"
+    @db.literal(@a.overlaps([1, 2])).should == "(a && ARRAY[1,2])"
+    @db.literal(@a.push([1, 2])).should == "(a || ARRAY[1,2])"
+    @db.literal(@a.concat([1, 2])).should == "(a || ARRAY[1,2])"
+    @db.literal(@a.unshift([1, 2])).should == "(ARRAY[1,2] || a)"
+  end
+end

data/spec/extensions/pg_array_spec.rb

@@ -0,0 +1,324 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
+
+describe "pg_array extension" do
+  before(:all) do
+    @pg_types = Sequel::Postgres::PG_TYPES.dup
+  end
+  after(:all) do
+    Sequel::Postgres::PG_TYPES.replace(@pg_types)
+  end
+
+  before do
+    @db = Sequel.connect('mock://postgres', :quote_identifiers=>false)
+    @db.extend_datasets(Module.new{def supports_timestamp_timezones?; false; end; def supports_timestamp_usecs?; false; end})
+    @db.extension(:pg_array)
+    @m = Sequel::Postgres
+    @converter = @m::PG_TYPES
+  end
+
+  it "should parse single dimensional text arrays" do
+    c = @converter[1009]
+    c.call("{a}").to_a.first.should be_a_kind_of(String)
+    c.call("{}").to_a.should == []
+    c.call("{a}").to_a.should == ['a']
+    c.call('{"a b"}').to_a.should == ['a b']
+    c.call('{a,b}').to_a.should == ['a', 'b']
+  end
+
+  it "should parse multi-dimensional text arrays" do
+    c = @converter[1009]
+    c.call("{{}}").to_a.should == [[]]
+    c.call("{{a},{b}}").to_a.should == [['a'], ['b']]
+    c.call('{{"a b"},{c}}').to_a.should == [['a b'], ['c']]
+    c.call('{{{a},{b}},{{c},{d}}}').to_a.should == [[['a'], ['b']], [['c'], ['d']]]
+    c.call('{{{a,e},{b,f}},{{c,g},{d,h}}}').to_a.should == [[['a', 'e'], ['b', 'f']], [['c', 'g'], ['d', 'h']]]
+  end
+
+  it "should parse text arrays with embedded deliminaters" do
+    c = @converter[1009]
+    c.call('{{"{},","\\",\\,\\\\\\"\\""}}').to_a.should == [['{},', '",,\\""']]
+  end
+
+  it "should parse single dimensional integer arrays" do
+    c = @converter[1007]
+    c.call("{1}").to_a.first.should be_a_kind_of(Integer)
+    c.call("{}").to_a.should == []
+    c.call("{1}").to_a.should == [1]
+    c.call('{2,3}').to_a.should == [2, 3]
+    c.call('{3,4,5}').to_a.should == [3, 4, 5]
+  end
+
+  it "should parse multiple dimensional integer arrays" do
+    c = @converter[1007]
+    c.call("{{}}").to_a.should == [[]]
+    c.call("{{1}}").to_a.should == [[1]]
+    c.call('{{2},{3}}').to_a.should == [[2], [3]]
+    c.call('{{{1,2},{3,4}},{{5,6},{7,8}}}').to_a.should == [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
+  end
+
+  it "should parse single dimensional float arrays" do
+    c = @converter[1022]
+    c.call("{}").to_a.should == []
+    c.call("{1.5}").to_a.should == [1.5]
+    c.call('{2.5,3.5}').to_a.should == [2.5, 3.5]
+    c.call('{3.5,4.5,5.5}').to_a.should == [3.5, 4.5, 5.5]
+  end
+
+  it "should parse multiple dimensional float arrays" do
+    c = @converter[1022]
+    c.call("{{}}").to_a.should == [[]]
+    c.call("{{1.5}}").to_a.should == [[1.5]]
+    c.call('{{2.5},{3.5}}').to_a.should == [[2.5], [3.5]]
+    c.call('{{{1.5,2.5},{3.5,4.5}},{{5.5,6.5},{7.5,8.5}}}').to_a.should == [[[1.5, 2.5], [3.5, 4.5]], [[5.5, 6.5], [7.5, 8.5]]]
+  end
+
+  it "should parse integers in float arrays as floats" do
+    c = @converter[1022]
+    c.call("{1}").to_a.first.should be_a_kind_of(Float)
+    c.call("{1}").to_a.should == [1.0]
+    c.call('{{{1,2},{3,4}},{{5,6},{7,8}}}').to_a.should == [[[1.0, 2.0], [3.0, 4.0]], [[5.0, 6.0], [7.0, 8.0]]]
+  end
+
+  it "should parse single dimensional decimal arrays" do
+    c = @converter[1231]
+    c.call("{}").to_a.should == []
+    c.call("{1.5}").to_a.should == [BigDecimal.new('1.5')]
+    c.call('{2.5,3.5}').to_a.should == [BigDecimal.new('2.5'), BigDecimal.new('3.5')]
+    c.call('{3.5,4.5,5.5}').to_a.should == [BigDecimal.new('3.5'), BigDecimal.new('4.5'), BigDecimal.new('5.5')]
+  end
+
+  it "should parse multiple dimensional decimal arrays" do
+    c = @converter[1231]
+    c.call("{{}}").to_a.should == [[]]
+    c.call("{{1.5}}").to_a.should == [[BigDecimal.new('1.5')]]
+    c.call('{{2.5},{3.5}}').to_a.should == [[BigDecimal.new('2.5')], [BigDecimal.new('3.5')]]
+    c.call('{{{1.5,2.5},{3.5,4.5}},{{5.5,6.5},{7.5,8.5}}}').to_a.should == [[[BigDecimal.new('1.5'), BigDecimal.new('2.5')], [BigDecimal.new('3.5'), BigDecimal.new('4.5')]], [[BigDecimal.new('5.5'), BigDecimal.new('6.5')], [BigDecimal.new('7.5'), BigDecimal.new('8.5')]]]
+  end
+
+  it "should parse decimal values with arbitrary precision" do
+    c = @converter[1231]
+    c.call("{1.000000000000000000005}").to_a.should == [BigDecimal.new('1.000000000000000000005')]
+    c.call("{{1.000000000000000000005,2.000000000000000000005},{3.000000000000000000005,4.000000000000000000005}}").to_a.should == [[BigDecimal.new('1.000000000000000000005'), BigDecimal.new('2.000000000000000000005')], [BigDecimal.new('3.000000000000000000005'), BigDecimal.new('4.000000000000000000005')]]
+  end
+
+  it "should parse integers in decimal arrays as BigDecimals" do
+    c = @converter[1231]
+    c.call("{1}").to_a.first.should be_a_kind_of(BigDecimal)
+    c.call("{1}").to_a.should == [BigDecimal.new('1')]
+    c.call('{{{1,2},{3,4}},{{5,6},{7,8}}}').to_a.should == [[[BigDecimal.new('1'), BigDecimal.new('2')], [BigDecimal.new('3'), BigDecimal.new('4')]], [[BigDecimal.new('5'), BigDecimal.new('6')], [BigDecimal.new('7'), BigDecimal.new('8')]]]
+  end
+
+  it "should parse arrays with NULL values" do
+    @converter.values_at(1007, 1009, 1022, 1231).each do |c|
+      c.call("{NULL}").should == [nil]
+      c.call("{NULL,NULL}").should == [nil,nil]
+      c.call("{{NULL,NULL},{NULL,NULL}}").should == [[nil,nil],[nil,nil]]
+    end
+  end
+
+  it 'should parse arrays with "NULL" values' do
+    c = @converter[1009]
+    c.call('{NULL,"NULL",NULL}').to_a.should == [nil, "NULL", nil]
+    c.call('{NULLA,"NULL",NULL}').to_a.should == ["NULLA", "NULL", nil]
+  end
+
+  it "should literalize arrays without types correctly" do
+    @db.literal(@m::PGArray.new([])).should == 'ARRAY[]'
+    @db.literal(@m::PGArray.new([1])).should == 'ARRAY[1]'
+    @db.literal(@m::PGArray.new([nil])).should == 'ARRAY[NULL]'
+    @db.literal(@m::PGArray.new([nil, 1])).should == 'ARRAY[NULL,1]'
+    @db.literal(@m::PGArray.new([1.0, 2.5])).should == 'ARRAY[1.0,2.5]'
+    @db.literal(@m::PGArray.new([BigDecimal.new('1'), BigDecimal.new('2.000000000000000000005')])).should == 'ARRAY[1.0,2.000000000000000000005]'
+    @db.literal(@m::PGArray.new([nil, "NULL"])).should == "ARRAY[NULL,'NULL']"
+    @db.literal(@m::PGArray.new([nil, "{},[]'\""])).should == "ARRAY[NULL,'{},[]''\"']"
+  end
+
+  it "should literalize multidimensional arrays correctly" do
+    @db.literal(@m::PGArray.new([[]])).should == 'ARRAY[[]]'
+    @db.literal(@m::PGArray.new([[1, 2]])).should == 'ARRAY[[1,2]]'
+    @db.literal(@m::PGArray.new([[3], [5]])).should == 'ARRAY[[3],[5]]'
+    @db.literal(@m::PGArray.new([[[1.0]], [[2.5]]])).should == 'ARRAY[[[1.0]],[[2.5]]]'
+    @db.literal(@m::PGArray.new([[[["NULL"]]]])).should == "ARRAY[[[['NULL']]]]"
+    @db.literal(@m::PGArray.new([["a", "b"], ["{},[]'\"", nil]])).should == "ARRAY[['a','b'],['{},[]''\"',NULL]]"
+  end
+
+  it "should literalize with types correctly" do
+    @db.literal(@m::PGArray.new([1], :int4)).should == 'ARRAY[1]::int4[]'
+    @db.literal(@m::PGArray.new([nil], :text)).should == 'ARRAY[NULL]::text[]'
+    @db.literal(@m::PGArray.new([nil, 1], :int8)).should == 'ARRAY[NULL,1]::int8[]'
+    @db.literal(@m::PGArray.new([1.0, 2.5], :real)).should == 'ARRAY[1.0,2.5]::real[]'
+    @db.literal(@m::PGArray.new([BigDecimal.new('1'), BigDecimal.new('2.000000000000000000005')], :decimal)).should == 'ARRAY[1.0,2.000000000000000000005]::decimal[]'
+    @db.literal(@m::PGArray.new([nil, "NULL"], :varchar)).should == "ARRAY[NULL,'NULL']::varchar[]"
+    @db.literal(@m::PGArray.new([nil, "{},[]'\""], :"varchar(255)")).should == "ARRAY[NULL,'{},[]''\"']::varchar(255)[]"
+  end
+
+  it "should have Sequel.pg_array method for easy PGArray creation" do
+    @db.literal(Sequel.pg_array([1])).should == 'ARRAY[1]'
+    @db.literal(Sequel.pg_array([1, 2], :int4)).should == 'ARRAY[1,2]::int4[]'
+    @db.literal(Sequel.pg_array([[[1], [2]], [[3], [4]]], :real)).should == 'ARRAY[[[1],[2]],[[3],[4]]]::real[]'
+  end
+
+  it "should have Sequel.pg_array return existing PGArrays as-is" do
+    a = Sequel.pg_array([1])
+    Sequel.pg_array(a).should equal(a)
+  end
+
+  it "should have Sequel.pg_array create a new PGArrays if type of existing does not match" do
+    a = Sequel.pg_array([1], :int4)
+    b = Sequel.pg_array(a, :int8)
+    a.should == b
+    a.should_not equal(b)
+    a.array_type.should == :int4
+    b.array_type.should == :int8
+  end
+
+  it "should support using arrays as bound variables" do
+    @db.bound_variable_arg(1, nil).should == 1
+    @db.bound_variable_arg(Sequel.pg_array([1,2]), nil).should == '{1,2}'
+    @db.bound_variable_arg([1,2], nil).should == '{1,2}'
+    @db.bound_variable_arg([[1,2]], nil).should == '{{1,2}}'
+    @db.bound_variable_arg([1.0,2.0], nil).should == '{1.0,2.0}'
+    @db.bound_variable_arg([Sequel.lit('a'), Sequel.blob("a\0'\"")], nil).should == '{a,"a\\\\000\\\\047\\""}'
+    @db.bound_variable_arg(["\\ \"", 'NULL', nil], nil).should == '{"\\\\ \\"","NULL",NULL}'
+  end
+
+  it "should parse array types from the schema correctly" do
+    @db.fetch = [{:name=>'id', :db_type=>'integer'}, {:name=>'i', :db_type=>'integer[]'}, {:name=>'f', :db_type=>'real[]'}, {:name=>'d', :db_type=>'numeric[]'}, {:name=>'t', :db_type=>'text[]'}]
+    @db.schema(:items).map{|e| e[1][:type]}.should == [:integer, :integer_array, :float_array, :decimal_array, :string_array]
+  end
+
+  it "should support typecasting of the various array types" do
+    {
+      :integer=>{:class=>Integer, :convert=>['1', 1, '1']},
+      :float=>{:db_type=>'double precision',  :class=>Float, :convert=>['1.1', 1.1, '1.1']},
+      :decimal=>{:db_type=>'numeric', :class=>BigDecimal, :convert=>['1.00000000000000000000000001', BigDecimal.new('1.00000000000000000000000001'), '1.00000000000000000000000001']},
+      :string=>{:db_type=>'text', :class=>String, :convert=>[1, '1', "'1'"]},
+      :bigint=>{:class=>Integer, :convert=>['1', 1, '1']},
+      :boolean=>{:class=>TrueClass, :convert=>['t', true, 'true']},
+      :blob=>{:db_type=>'bytea', :class=>Sequel::SQL::Blob, :convert=>['1', '1', "'1'"]},
+      :date=>{:class=>Date, :convert=>['2011-10-12', Date.new(2011, 10, 12), "'2011-10-12'"]},
+      :time=>{:db_type=>'time without time zone', :class=>Sequel::SQLTime, :convert=>['01:02:03', Sequel::SQLTime.create(1, 2, 3), "'01:02:03'"]},
+      :datetime=>{:db_type=>'timestamp without time zone', :class=>Time, :convert=>['2011-10-12 01:02:03', Time.local(2011, 10, 12, 1, 2, 3), "'2011-10-12 01:02:03'"]},
+      :time_timezone=>{:db_type=>'time with time zone', :class=>Sequel::SQLTime, :convert=>['01:02:03', Sequel::SQLTime.create(1, 2, 3), "'01:02:03'"]},
+      :datetime_timezone=>{:db_type=>'timestamp with time zone', :class=>Time, :convert=>['2011-10-12 01:02:03', Time.local(2011, 10, 12, 1, 2, 3), "'2011-10-12 01:02:03'"]},
+    }.each do |type, h|
+      meth = :"#{type}_array"
+      db_type = h[:db_type]||type
+      klass = h[:class]
+      array_in, value, output = h[:convert]
+
+      [[array_in]].each do |input|
+        v = @db.typecast_value(meth, input)
+        v.should == [value]
+        v.first.should be_a_kind_of(klass)
+        v.array_type.should_not be_nil
+        @db.typecast_value(meth, Sequel.pg_array([value])).should == v
+        @db.typecast_value(meth, v).should equal(v)
+      end
+
+      [[[array_in]]].each do |input|
+        v = @db.typecast_value(meth, input)
+        v.should == [[value]]
+        v.first.first.should be_a_kind_of(klass)
+        v.array_type.should_not be_nil
+        @db.typecast_value(meth, Sequel.pg_array([[value]])).should == v
+        @db.typecast_value(meth, v).should equal(v)
+      end
+
+      @db.literal(@db.typecast_value(meth, [array_in])).should == "ARRAY[#{output}]::#{db_type}[]"
+      @db.literal(@db.typecast_value(meth, [])).should == "ARRAY[]::#{db_type}[]"
+    end
+    proc{@db.typecast_value(:integer_array, {})}.should raise_error(Sequel::InvalidValue)
+  end
+
+  it "should support SQL::AliasMethods" do
+    @db.select(Sequel.pg_array([1], :integer).as(:col1)).sql.should == 'SELECT ARRAY[1]::integer[] AS col1'
+  end
+
+  it "should support registering custom array types" do
+    Sequel::Postgres::PGArray.register('foo')
+    @db.typecast_value(:foo_array, []).should be_a_kind_of(Sequel::Postgres::PGArray)
+    @db.fetch = [{:name=>'id', :db_type=>'foo[]'}]
+    @db.schema(:items).map{|e| e[1][:type]}.should == [:foo_array]
+  end
+
+  it "should support registering custom types with :type_symbol option" do
+    Sequel::Postgres::PGArray.register('foo', :type_symbol=>:bar)
+    @db.typecast_value(:bar_array, []).should be_a_kind_of(Sequel::Postgres::PGArray)
+    @db.fetch = [{:name=>'id', :db_type=>'foo[]'}]
+    @db.schema(:items).map{|e| e[1][:type]}.should == [:bar_array]
+  end
+
+  it "should support using a block as a custom conversion proc given as block" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>1234){|s| (s*2).to_i}
+    @converter[1234].call('{1}').should == [11]
+  end
+
+  it "should support using a block as a custom conversion proc given as :converter option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>1234, :converter=>proc{|s| (s*2).to_i})
+    @converter[1234].call('{1}').should == [11]
+  end
+
+  it "should support using an existing scaler conversion proc via the :scalar_oid option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>1234, :scalar_oid=>16)
+    @converter[1234].call('{t}').should == [true]
+  end
+
+  it "should support using a given conversion procs hash via the :type_procs option" do
+    h = {16=>proc{|s| "!#{s}"}}
+    Sequel::Postgres::PGArray.register('foo', :oid=>1234, :scalar_oid=>16, :type_procs=>h)
+    h[1234].call('{t}').should == ["!t"]
+  end
+
+  it "should support adding methods to the given module via the :typecast_methods_module option" do
+    m = Module.new
+    Sequel::Postgres::PGArray.register('foo15', :scalar_typecast=>:boolean, :typecast_methods_module=>m)
+    @db.typecast_value(:foo15_array, ['t']).should == ['t']
+    @db.extend(m)
+    @db.typecast_value(:foo15_array, ['t']).should == [true]
+  end
+
+  it "should raise an error if using :scalar_oid option with unexisting scalar conversion proc" do
+    proc{Sequel::Postgres::PGArray.register('foo', :scalar_oid=>0)}.should raise_error(Sequel::Error)
+  end
+
+  it "should raise an error if using :converter option and a block argument" do
+    proc{Sequel::Postgres::PGArray.register('foo', :converter=>proc{}){}}.should raise_error(Sequel::Error)
+  end
+
+  it "should raise an error if using :scalar_oid option and a block argument" do
+    proc{Sequel::Postgres::PGArray.register('foo', :scalar_oid=>16){}}.should raise_error(Sequel::Error)
+  end
+
+  it "should support registering custom types with :oid option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>1)
+    Sequel::Postgres::PG_TYPES[1].call('{1}').should be_a_kind_of(Sequel::Postgres::PGArray)
+  end
+
+  it "should support registering custom types with :parser=>:json option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>2, :parser=>:json)
+    Sequel::Postgres::PG_TYPES[2].should be_a_kind_of(Sequel::Postgres::PGArray::JSONCreator)
+  end
+
+  it "should support registering converters with :parser=>:json option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>4, :parser=>:json){|s| s * 2}
+    Sequel::Postgres::PG_TYPES[4].call('{{1, 2}, {3, 4}}').should == [[2, 4], [6, 8]]
+  end
+
+  it "should support registering custom types with :array_type option" do
+    Sequel::Postgres::PGArray.register('foo', :oid=>3, :array_type=>:blah)
+    @db.literal(Sequel::Postgres::PG_TYPES[3].call('{}')).should == 'ARRAY[]::blah[]'
+  end
+
+  it "should use and not override existing database typecast method if :typecast_method option is given" do
+    Sequel::Postgres::PGArray.register('foo', :typecast_method=>:float)
+    @db.fetch = [{:name=>'id', :db_type=>'foo[]'}]
+    @db.schema(:items).map{|e| e[1][:type]}.should == [:float_array]
+  end
+
+  it "should set appropriate timestamp conversion procs when getting conversion procs" do
+    procs = @db.conversion_procs
+    procs[1185].call('{"2011-10-20 11:12:13"}').should == [Time.local(2011, 10, 20, 11, 12, 13)]
+    procs[1115].call('{"2011-10-20 11:12:13"}').should == [Time.local(2011, 10, 20, 11, 12, 13)]
+  end
+end

data/spec/extensions/pg_auto_parameterize_spec.rb

@@ -0,0 +1,65 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "spec_helper")
+
+describe "pg_auto_parameterize extension" do
+  before do
+    @db = Sequel.connect('mock://postgres', :quote_identifiers=>false)
+    @db.synchronize{|c| def c.escape_bytea(v) v*2 end}
+    @db.extend_datasets{def use_cursor(*) clone end}
+    @db.extension :pg_auto_parameterize
+  end
+
+  it "should automatically parameterize queries strings, blobs, numerics, dates, and times" do
+    pr = proc do |ds, sql, *args|
+      arg = args[0]
+      parg = args[1] || arg
+      s = ds.filter(:a=>arg).sql
+      s.should == sql
+      s.args.should == (parg == :nil ? nil : [parg])
+    end
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::int4)', 1)
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::int8)', 18446744073709551616)
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::double precision)', 1.1)
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::numeric)', BigDecimal.new('1.01'))
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1)', "a")
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::bytea)', Sequel.blob("a\0b"))
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = 1)', Sequel.lit('1'), :nil)
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::time)', Sequel::SQLTime.create(1, 2, 3, 500000))
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::date)', Date.today)
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::timestamp)', DateTime.new(2012, 1, 2, 3, 4, 5))
+    pr.call(@db[:table], 'SELECT * FROM table WHERE (a = $1::timestamp)', Time.utc(2012, 1, 2, 3, 4, 5))
+  end
+
+  it "should extract parameters from query string when executing" do
+    @db.fetch = {:a=>1}
+    @db.numrows = 1
+    @db.autoid = 1
+
+    @db[:table].filter(:a=>1).all.should == [{:a=>1}]
+    @db.sqls.should == ['SELECT * FROM table WHERE (a = $1::int4) -- args: [1]']
+
+    @db[:table].filter(:a=>1).update(:b=>'a').should == 1
+    @db.sqls.should == ['UPDATE table SET b = $1 WHERE (a = $2::int4) -- args: ["a", 1]']
+
+    @db[:table].filter(:a=>1).delete.should == 1
+    @db.sqls.should == ['DELETE FROM table WHERE (a = $1::int4) -- args: [1]']
+
+    @db[:table].insert(:a=>1).should == 1
+    @db.sqls.should == ['INSERT INTO table (a) VALUES ($1::int4) RETURNING id -- args: [1]']
+  end
+
+  it "should not automatically paramiterize if no_auto_parameterize is used" do
+    @db[:table].no_auto_parameterize.filter(:a=>1).sql.should == 'SELECT * FROM table WHERE (a = 1)'
+  end
+
+  it "should not automatically parameterize prepared statements" do
+    @db[:table].filter(:a=>1, :b=>:$b).prepare(:select).sql.should =~ /SELECT \* FROM table WHERE \(\((a = 1|b = \$b)\) AND \((a = 1|b = \$b)\)\)/
+  end
+
+  it "should show args with string when inspecting SQL " do
+    @db[:table].filter(:a=>1).sql.inspect.should == '"SELECT * FROM table WHERE (a = $1::int4); [1]"'
+  end
+
+  it "should not auto parameterize when using cursors" do
+    @db[:table].filter(:a=>1).use_cursor.opts[:no_auto_parameterize].should be_true
+  end
+end