require "cases/helper"
module ActiveRecord
module ConnectionAdapters
class QuotingTest < ActiveRecord::TestCase
class FakeColumn < ActiveRecord::ConnectionAdapters::Column
attr_accessor :type
def initialize type
@type = type
end
end
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
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
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
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
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
def test_quote_float
float = 1.2
assert_equal float.to_s, @quoter.quote(float, nil)
assert_equal float.to_s, @quoter.quote(float, FakeColumn.new(:float))
end
def test_quote_fixnum
fixnum = 1
assert_equal fixnum.to_s, @quoter.quote(fixnum, nil)
assert_equal fixnum.to_s, @quoter.quote(fixnum, FakeColumn.new(:integer))
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, FakeColumn.new(:integer))
end
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, FakeColumn.new(:decimal))
end
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, FakeColumn.new(:date))
assert_equal "'lol'", @quoter.quote(Time.now, nil)
assert_equal "'lol'", @quoter.quote(Time.now, FakeColumn.new(:time))
assert_equal "'lol'", @quoter.quote(DateTime.now, nil)
assert_equal "'lol'", @quoter.quote(DateTime.now, FakeColumn.new(:datetime))
end
def test_crazy_object
crazy = Class.new.new
expected = "'#{YAML.dump(crazy)}'"
assert_equal expected, @quoter.quote(crazy, nil)
assert_equal expected, @quoter.quote(crazy, Object.new)
end
def test_crazy_object_calls_quote_string
crazy = Class.new { def initialize; @lol = 'lo\l' end }.new
assert_match "lo\\\\l", @quoter.quote(crazy, nil)
assert_match "lo\\\\l", @quoter.quote(crazy, Object.new)
end
def test_quote_string_no_column
assert_equal "'lo\\\\l'", @quoter.quote('lo\l', nil)
end
def test_quote_as_mb_chars_no_column
string = ActiveSupport::Multibyte::Chars.new('lo\l')
assert_equal "'lo\\\\l'", @quoter.quote(string, nil)
end
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
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
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
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) {
def string_to_binary(value)
'foo'
end
}.new(:binary)
assert_equal "'foo'", @quoter.quote('lo\l', col)
end
def test_quote_as_mb_chars_binary_column_with_string_to_binary
col = Class.new(FakeColumn) {
def string_to_binary(value)
'foo'
end
}.new(:binary)
string = ActiveSupport::Multibyte::Chars.new('lo\l')
assert_equal "'foo'", @quoter.quote(string, col)
end
def test_string_with_crazy_column
assert_equal "'lo\\\\l'", @quoter.quote('lo\l', FakeColumn.new(:foo))
end
def test_quote_duration
assert_equal "1800", @quoter.quote(30.minutes)
end
def test_quote_duration_int_column
assert_equal "7200", @quoter.quote(2.hours, FakeColumn.new(:integer))
end
end
end
end
