~michaelforrest/use-case-mapper/trunk

require "cases/helper"

 

class AdapterTest < ActiveRecord::TestCase

  def setup

    @connection = ActiveRecord::Base.connection

  end

 

  def test_tables

    tables = @connection.tables

    assert tables.include?("accounts")

    assert tables.include?("authors")

    assert tables.include?("tasks")

    assert tables.include?("topics")

  end

 

  def test_table_exists?

    assert @connection.table_exists?("accounts")

    assert !@connection.table_exists?("nonexistingtable")

  end

 

  def test_indexes

    idx_name = "accounts_idx"

 

    if @connection.respond_to?(:indexes)

      indexes = @connection.indexes("accounts")

      assert indexes.empty?

 

      @connection.add_index :accounts, :firm_id, :name => idx_name

      indexes = @connection.indexes("accounts")

      assert_equal "accounts", indexes.first.table

      # OpenBase does not have the concept of a named index

      # Indexes are merely properties of columns.

      assert_equal idx_name, indexes.first.name unless current_adapter?(:OpenBaseAdapter)

      assert !indexes.first.unique

      assert_equal ["firm_id"], indexes.first.columns

    else

      warn "#{@connection.class} does not respond to #indexes"

    end

 

  ensure

    @connection.remove_index(:accounts, :name => idx_name) rescue nil

  end

 

  def test_current_database

    if @connection.respond_to?(:current_database)

      assert_equal ENV['ARUNIT_DB_NAME'] || "activerecord_unittest", @connection.current_database

    end

  end

 

  if current_adapter?(:MysqlAdapter)

    def test_charset

      assert_not_nil @connection.charset

      assert_not_equal 'character_set_database', @connection.charset

      assert_equal @connection.show_variable('character_set_database'), @connection.charset

    end

 

    def test_collation

      assert_not_nil @connection.collation

      assert_not_equal 'collation_database', @connection.collation

      assert_equal @connection.show_variable('collation_database'), @connection.collation

    end

 

    def test_show_nonexistent_variable_returns_nil

      assert_nil @connection.show_variable('foo_bar_baz')

    end

 

    def test_not_specifying_database_name_for_cross_database_selects

      assert_nothing_raised do

        ActiveRecord::Base.establish_connection({

          :adapter  => 'mysql',

          :username => 'rails'

        })

        ActiveRecord::Base.connection.execute "SELECT activerecord_unittest.pirates.*, activerecord_unittest2.courses.* FROM activerecord_unittest.pirates, activerecord_unittest2.courses"

      end

 

      ActiveRecord::Base.establish_connection 'arunit'

    end

  end

 

  if current_adapter?(:PostgreSQLAdapter)

    def test_encoding

      assert_not_nil @connection.encoding

    end

  end

 

  def test_table_alias

    def @connection.test_table_alias_length() 10; end

    class << @connection

      alias_method :old_table_alias_length, :table_alias_length

      alias_method :table_alias_length,     :test_table_alias_length

    end

 

    assert_equal 'posts',      @connection.table_alias_for('posts')

    assert_equal 'posts_comm', @connection.table_alias_for('posts_comments')

    assert_equal 'dbo_posts',  @connection.table_alias_for('dbo.posts')

 

    class << @connection

      remove_method :table_alias_length

      alias_method :table_alias_length, :old_table_alias_length

    end

  end

 

  # test resetting sequences in odd tables in postgreSQL

  if ActiveRecord::Base.connection.respond_to?(:reset_pk_sequence!)

    require 'models/movie'

    require 'models/subscriber'

 

    def test_reset_empty_table_with_custom_pk

      Movie.delete_all

      Movie.connection.reset_pk_sequence! 'movies'

      assert_equal 1, Movie.create(:name => 'fight club').id

    end

 

    if ActiveRecord::Base.connection.adapter_name != "FrontBase"

      def test_reset_table_with_non_integer_pk

        Subscriber.delete_all

        Subscriber.connection.reset_pk_sequence! 'subscribers'

        sub = Subscriber.new(:name => 'robert drake')

        sub.id = 'bob drake'

        assert_nothing_raised { sub.save! }

      end

    end

  end

 

  def test_add_limit_offset_should_sanitize_sql_injection_for_limit_without_comas

    sql_inject = "1 select * from schema"

      assert_equal " LIMIT 1", @connection.add_limit_offset!("", :limit=>sql_inject)

    if current_adapter?(:MysqlAdapter)

      assert_equal " LIMIT 7, 1", @connection.add_limit_offset!("", :limit=>sql_inject, :offset=>7)

    else

      assert_equal " LIMIT 1 OFFSET 7", @connection.add_limit_offset!("", :limit=>sql_inject, :offset=>7)

    end

  end

 

  def test_add_limit_offset_should_sanitize_sql_injection_for_limit_with_comas

    sql_inject = "1, 7 procedure help()"

    if current_adapter?(:MysqlAdapter)

      assert_equal " LIMIT 1,7", @connection.add_limit_offset!("", :limit=>sql_inject)

      assert_equal " LIMIT 7, 1", @connection.add_limit_offset!("", :limit=> '1 ; DROP TABLE USERS', :offset=>7)

    else

      assert_equal " LIMIT 1,7", @connection.add_limit_offset!("", :limit=>sql_inject)

      assert_equal " LIMIT 1,7 OFFSET 7", @connection.add_limit_offset!("", :limit=>sql_inject, :offset=>7)

    end

  end

end