87 lines
3.6 KiB
Ruby
87 lines
3.6 KiB
Ruby
|
require "cases/helper"
|
||
|
|
||
|
class ActiveSchemaTest < ActiveRecord::TestCase
|
||
|
def setup
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
alias_method :execute_without_stub, :execute
|
||
|
def execute(sql, name = nil) return sql end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def teardown
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
remove_method :execute
|
||
|
alias_method :execute, :execute_without_stub
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_drop_table
|
||
|
assert_equal "DROP TABLE `people`", drop_table(:people)
|
||
|
end
|
||
|
|
||
|
if current_adapter?(:MysqlAdapter)
|
||
|
def test_create_mysql_database_with_encoding
|
||
|
assert_equal "CREATE DATABASE `matt` DEFAULT CHARACTER SET `utf8`", create_database(:matt)
|
||
|
assert_equal "CREATE DATABASE `aimonetti` DEFAULT CHARACTER SET `latin1`", create_database(:aimonetti, {:charset => 'latin1'})
|
||
|
assert_equal "CREATE DATABASE `matt_aimonetti` DEFAULT CHARACTER SET `big5` COLLATE `big5_chinese_ci`", create_database(:matt_aimonetti, {:charset => :big5, :collation => :big5_chinese_ci})
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_add_column
|
||
|
assert_equal "ALTER TABLE `people` ADD `last_name` varchar(255)", add_column(:people, :last_name, :string)
|
||
|
end
|
||
|
|
||
|
def test_add_column_with_limit
|
||
|
assert_equal "ALTER TABLE `people` ADD `key` varchar(32)", add_column(:people, :key, :string, :limit => 32)
|
||
|
end
|
||
|
|
||
|
def test_drop_table_with_specific_database
|
||
|
assert_equal "DROP TABLE `otherdb`.`people`", drop_table('otherdb.people')
|
||
|
end
|
||
|
|
||
|
def test_add_timestamps
|
||
|
#we need to actually modify some data, so we make execute to point to the original method
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
alias_method :execute_with_stub, :execute
|
||
|
alias_method :execute, :execute_without_stub
|
||
|
end
|
||
|
ActiveRecord::Base.connection.create_table :delete_me do |t|
|
||
|
end
|
||
|
ActiveRecord::Base.connection.add_timestamps :delete_me
|
||
|
assert_equal ActiveRecord::Base.connection.execute("SHOW FIELDS FROM delete_me where FIELD='updated_at' AND TYPE='datetime'").num_rows, 1
|
||
|
assert_equal ActiveRecord::Base.connection.execute("SHOW FIELDS FROM delete_me where FIELD='created_at' AND TYPE='datetime'").num_rows, 1
|
||
|
ensure
|
||
|
ActiveRecord::Base.connection.drop_table :delete_me rescue nil
|
||
|
#before finishing, we restore the alias to the mock-up method
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
alias_method :execute, :execute_with_stub
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def test_remove_timestamps
|
||
|
#we need to actually modify some data, so we make execute to point to the original method
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
alias_method :execute_with_stub, :execute
|
||
|
alias_method :execute, :execute_without_stub
|
||
|
end
|
||
|
ActiveRecord::Base.connection.create_table :delete_me do |t|
|
||
|
t.timestamps
|
||
|
end
|
||
|
ActiveRecord::Base.connection.remove_timestamps :delete_me
|
||
|
assert_equal ActiveRecord::Base.connection.execute("SHOW FIELDS FROM delete_me where FIELD='updated_at' AND TYPE='datetime'").num_rows, 0
|
||
|
assert_equal ActiveRecord::Base.connection.execute("SHOW FIELDS FROM delete_me where FIELD='created_at' AND TYPE='datetime'").num_rows, 0
|
||
|
ensure
|
||
|
ActiveRecord::Base.connection.drop_table :delete_me rescue nil
|
||
|
#before finishing, we restore the alias to the mock-up method
|
||
|
ActiveRecord::ConnectionAdapters::MysqlAdapter.class_eval do
|
||
|
alias_method :execute, :execute_with_stub
|
||
|
end
|
||
|
end
|
||
|
|
||
|
|
||
|
private
|
||
|
def method_missing(method_symbol, *arguments)
|
||
|
ActiveRecord::Base.connection.send(method_symbol, *arguments)
|
||
|
end
|
||
|
end
|