require "cases/helper"
require 'bigdecimal'
require 'yaml'
require 'securerandom'

module ActiveRecord
  module ConnectionAdapters
    class SQLite3Adapter
      class QuotingTest < ActiveRecord::TestCase
        def setup
          @conn = Base.sqlite3_connection :database => ':memory:',
            :adapter => 'sqlite3',
            :timeout => 100
        end

        def test_type_cast_binary_encoding_without_logger
          @conn.extend(Module.new { def logger; end })
          binary = SecureRandom.hex
          expected = binary.dup.encode!(Encoding::UTF_8)
          assert_equal expected, @conn.type_cast(binary)
        end

        def test_type_cast_symbol
          assert_equal 'foo', @conn.type_cast(:foo)
        end

        def test_type_cast_date
          date = Date.today
          expected = @conn.quoted_date(date)
          assert_equal expected, @conn.type_cast(date)
        end

        def test_type_cast_time
          time = Time.now
          expected = @conn.quoted_date(time)
          assert_equal expected, @conn.type_cast(time)
        end

        def test_type_cast_numeric
          assert_equal 10, @conn.type_cast(10)
          assert_equal 2.2, @conn.type_cast(2.2)
        end

        def test_type_cast_nil
          assert_equal nil, @conn.type_cast(nil)
        end

        def test_type_cast_true
          assert_equal 't', @conn.type_cast(true)
        end

        def test_type_cast_false
          assert_equal 'f', @conn.type_cast(false)
        end

        def test_type_cast_bigdecimal
          bd = BigDecimal.new '10.0'
          assert_equal bd.to_f, @conn.type_cast(bd)
        end

        def test_type_cast_unknown_should_raise_error
          obj = Class.new.new
          assert_raise(TypeError) { @conn.type_cast(obj) }
        end

        def test_type_cast_object_which_responds_to_quoted_id
          quoted_id_obj = Class.new {
            def quoted_id
              "'zomg'"
            end

            def id
              10
            end
          }.new
          assert_equal 10, @conn.type_cast(quoted_id_obj)

          quoted_id_obj = Class.new {
            def quoted_id
              "'zomg'"
            end
          }.new
          assert_raise(TypeError) { @conn.type_cast(quoted_id_obj) }
        end

        def test_quoting_binary_strings
          value = "hello".encode('ascii-8bit')
          type = SQLite3String.new

          assert_equal "'hello'", @conn.quote(type.type_cast_for_database(value))
        end
      end
    end
  end
end