2010-10-11 20:08:40 -04:00
|
|
|
require "cases/helper"
|
|
|
|
|
|
|
|
module ActiveRecord
|
|
|
|
module ConnectionAdapters
|
|
|
|
class QuotingTest < ActiveRecord::TestCase
|
2010-10-12 16:10:52 -04:00
|
|
|
class FakeColumn < ActiveRecord::ConnectionAdapters::Column
|
|
|
|
attr_accessor :type
|
|
|
|
|
|
|
|
def initialize type
|
|
|
|
@type = type
|
|
|
|
end
|
2010-10-12 14:29:28 -04:00
|
|
|
end
|
|
|
|
|
2010-10-11 20:08:40 -04:00
|
|
|
def setup
|
|
|
|
@quoter = Class.new { include Quoting }.new
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_true
|
|
|
|
assert_equal "'t'", @quoter.quoted_true
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_false
|
|
|
|
assert_equal "'f'", @quoter.quoted_false
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_column_name
|
|
|
|
assert_equal "foo", @quoter.quote_column_name('foo')
|
|
|
|
end
|
|
|
|
|
2010-10-11 20:40:36 -04:00
|
|
|
def test_quote_table_name
|
|
|
|
assert_equal "foo", @quoter.quote_table_name('foo')
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_table_name_calls_quote_column_name
|
|
|
|
@quoter.extend(Module.new {
|
|
|
|
def quote_column_name(string)
|
|
|
|
'lol'
|
|
|
|
end
|
|
|
|
})
|
|
|
|
assert_equal 'lol', @quoter.quote_table_name('foo')
|
|
|
|
end
|
|
|
|
|
2010-10-11 20:08:40 -04:00
|
|
|
def test_quote_string
|
|
|
|
assert_equal "''", @quoter.quote_string("'")
|
|
|
|
assert_equal "\\\\", @quoter.quote_string("\\")
|
|
|
|
assert_equal "hi''i", @quoter.quote_string("hi'i")
|
|
|
|
assert_equal "hi\\\\i", @quoter.quote_string("hi\\i")
|
|
|
|
end
|
2010-10-11 20:40:36 -04:00
|
|
|
|
|
|
|
def test_quoted_date
|
|
|
|
t = Date.today
|
|
|
|
assert_equal t.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_time_utc
|
|
|
|
before = ActiveRecord::Base.default_timezone
|
|
|
|
ActiveRecord::Base.default_timezone = :utc
|
|
|
|
t = Time.now
|
|
|
|
assert_equal t.getutc.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
ensure
|
|
|
|
ActiveRecord::Base.default_timezone = before
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_time_local
|
|
|
|
before = ActiveRecord::Base.default_timezone
|
|
|
|
ActiveRecord::Base.default_timezone = :local
|
|
|
|
t = Time.now
|
|
|
|
assert_equal t.getlocal.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
ensure
|
|
|
|
ActiveRecord::Base.default_timezone = before
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_time_crazy
|
|
|
|
before = ActiveRecord::Base.default_timezone
|
|
|
|
ActiveRecord::Base.default_timezone = :asdfasdf
|
|
|
|
t = Time.now
|
|
|
|
assert_equal t.getlocal.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
ensure
|
|
|
|
ActiveRecord::Base.default_timezone = before
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quoted_datetime_utc
|
|
|
|
before = ActiveRecord::Base.default_timezone
|
|
|
|
ActiveRecord::Base.default_timezone = :utc
|
|
|
|
t = DateTime.now
|
|
|
|
assert_equal t.getutc.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
ensure
|
|
|
|
ActiveRecord::Base.default_timezone = before
|
|
|
|
end
|
|
|
|
|
|
|
|
###
|
|
|
|
# DateTime doesn't define getlocal, so make sure it does nothing
|
|
|
|
def test_quoted_datetime_local
|
|
|
|
before = ActiveRecord::Base.default_timezone
|
|
|
|
ActiveRecord::Base.default_timezone = :local
|
|
|
|
t = DateTime.now
|
|
|
|
assert_equal t.to_s(:db), @quoter.quoted_date(t)
|
|
|
|
ensure
|
|
|
|
ActiveRecord::Base.default_timezone = before
|
|
|
|
end
|
2010-10-11 20:51:48 -04:00
|
|
|
|
|
|
|
def test_quote_with_quoted_id
|
|
|
|
assert_equal 1, @quoter.quote(Struct.new(:quoted_id).new(1), nil)
|
|
|
|
assert_equal 1, @quoter.quote(Struct.new(:quoted_id).new(1), 'foo')
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_nil
|
|
|
|
assert_equal 'NULL', @quoter.quote(nil, nil)
|
|
|
|
assert_equal 'NULL', @quoter.quote(nil, 'foo')
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_true
|
|
|
|
assert_equal @quoter.quoted_true, @quoter.quote(true, nil)
|
|
|
|
assert_equal '1', @quoter.quote(true, Struct.new(:type).new(:integer))
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_false
|
|
|
|
assert_equal @quoter.quoted_false, @quoter.quote(false, nil)
|
|
|
|
assert_equal '0', @quoter.quote(false, Struct.new(:type).new(:integer))
|
|
|
|
end
|
2010-10-11 20:57:47 -04:00
|
|
|
|
|
|
|
def test_quote_float
|
|
|
|
float = 1.2
|
|
|
|
assert_equal float.to_s, @quoter.quote(float, nil)
|
|
|
|
assert_equal float.to_s, @quoter.quote(float, Object.new)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_fixnum
|
|
|
|
fixnum = 1
|
|
|
|
assert_equal fixnum.to_s, @quoter.quote(fixnum, nil)
|
|
|
|
assert_equal fixnum.to_s, @quoter.quote(fixnum, Object.new)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_bignum
|
|
|
|
bignum = 1 << 100
|
|
|
|
assert_equal bignum.to_s, @quoter.quote(bignum, nil)
|
|
|
|
assert_equal bignum.to_s, @quoter.quote(bignum, Object.new)
|
|
|
|
end
|
2010-10-12 14:08:10 -04:00
|
|
|
|
|
|
|
def test_quote_bigdecimal
|
|
|
|
bigdec = BigDecimal.new((1 << 100).to_s)
|
|
|
|
assert_equal bigdec.to_s('F'), @quoter.quote(bigdec, nil)
|
|
|
|
assert_equal bigdec.to_s('F'), @quoter.quote(bigdec, Object.new)
|
|
|
|
end
|
2010-10-12 14:21:35 -04:00
|
|
|
|
|
|
|
def test_dates_and_times
|
|
|
|
@quoter.extend(Module.new { def quoted_date(value) 'lol' end })
|
|
|
|
assert_equal "'lol'", @quoter.quote(Date.today, nil)
|
|
|
|
assert_equal "'lol'", @quoter.quote(Date.today, Object.new)
|
|
|
|
assert_equal "'lol'", @quoter.quote(Time.now, nil)
|
|
|
|
assert_equal "'lol'", @quoter.quote(Time.now, Object.new)
|
|
|
|
assert_equal "'lol'", @quoter.quote(DateTime.now, nil)
|
|
|
|
assert_equal "'lol'", @quoter.quote(DateTime.now, Object.new)
|
|
|
|
end
|
2010-10-12 14:25:20 -04:00
|
|
|
|
|
|
|
def test_crazy_object
|
|
|
|
crazy = Class.new { def to_s; 'lol' end }.new
|
|
|
|
assert_equal "'lol'", @quoter.quote(crazy, nil)
|
|
|
|
assert_equal "'lol'", @quoter.quote(crazy, Object.new)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_crazy_object_calls_quote_string
|
|
|
|
crazy = Class.new { def to_s; 'lo\l' end }.new
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote(crazy, nil)
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote(crazy, Object.new)
|
|
|
|
end
|
2010-10-12 14:29:28 -04:00
|
|
|
|
|
|
|
def test_quote_string_no_column
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote('lo\l', nil)
|
|
|
|
end
|
|
|
|
|
2010-10-12 14:38:07 -04:00
|
|
|
def test_quote_as_mb_chars_no_column
|
|
|
|
string = ActiveSupport::Multibyte::Chars.new('lo\l')
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote(string, nil)
|
|
|
|
end
|
|
|
|
|
2010-10-12 14:29:28 -04:00
|
|
|
def test_quote_string_int_column
|
|
|
|
assert_equal "1", @quoter.quote('1', FakeColumn.new(:integer))
|
|
|
|
assert_equal "1", @quoter.quote('1.2', FakeColumn.new(:integer))
|
|
|
|
end
|
2010-10-12 14:30:34 -04:00
|
|
|
|
|
|
|
def test_quote_string_float_column
|
|
|
|
assert_equal "1.0", @quoter.quote('1', FakeColumn.new(:float))
|
|
|
|
assert_equal "1.2", @quoter.quote('1.2', FakeColumn.new(:float))
|
|
|
|
end
|
2010-10-12 14:34:21 -04:00
|
|
|
|
2010-10-12 14:38:07 -04:00
|
|
|
def test_quote_as_mb_chars_binary_column
|
|
|
|
string = ActiveSupport::Multibyte::Chars.new('lo\l')
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote(string, FakeColumn.new(:binary))
|
|
|
|
end
|
|
|
|
|
2010-10-12 14:34:21 -04:00
|
|
|
def test_quote_binary_without_string_to_binary
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote('lo\l', FakeColumn.new(:binary))
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_quote_binary_with_string_to_binary
|
|
|
|
col = Class.new(FakeColumn) {
|
2010-10-12 16:10:52 -04:00
|
|
|
def string_to_binary(value)
|
2010-10-12 14:34:21 -04:00
|
|
|
'foo'
|
|
|
|
end
|
|
|
|
}.new(:binary)
|
|
|
|
assert_equal "'foo'", @quoter.quote('lo\l', col)
|
|
|
|
end
|
2010-10-12 14:38:07 -04:00
|
|
|
|
|
|
|
def test_quote_as_mb_chars_binary_column_with_string_to_binary
|
|
|
|
col = Class.new(FakeColumn) {
|
2010-10-12 16:10:52 -04:00
|
|
|
def string_to_binary(value)
|
2010-10-12 14:38:07 -04:00
|
|
|
'foo'
|
|
|
|
end
|
|
|
|
}.new(:binary)
|
|
|
|
string = ActiveSupport::Multibyte::Chars.new('lo\l')
|
|
|
|
assert_equal "'foo'", @quoter.quote(string, col)
|
|
|
|
end
|
2010-10-12 14:43:50 -04:00
|
|
|
|
|
|
|
def test_string_with_crazy_column
|
|
|
|
assert_equal "'lo\\\\l'", @quoter.quote('lo\l', FakeColumn.new(:foo))
|
|
|
|
end
|
2010-10-11 20:08:40 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|