145 lines
5 KiB
Ruby
145 lines
5 KiB
Ruby
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
|
|
begin
|
|
assert_nothing_raised do
|
|
ActiveRecord::Base.establish_connection(ActiveRecord::Base.configurations['arunit'].except(:database))
|
|
ActiveRecord::Base.connection.execute "SELECT activerecord_unittest.pirates.*, activerecord_unittest2.courses.* FROM activerecord_unittest.pirates, activerecord_unittest2.courses"
|
|
end
|
|
ensure
|
|
ActiveRecord::Base.establish_connection 'arunit'
|
|
end
|
|
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
|