1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00

Extract Sqlserver database tasks.

This commit is contained in:
kennyj 2013-03-28 03:07:51 +09:00
parent 4a370f9d7c
commit 0f313fce1c
5 changed files with 131 additions and 11 deletions

View file

@ -145,6 +145,7 @@ module ActiveRecord
'active_record/tasks/postgresql_database_tasks'
autoload :FirebirdDatabaseTasks, 'active_record/tasks/firebird_database_tasks'
autoload :SqlserverDatabaseTasks, 'active_record/tasks/sqlserver_database_tasks'
end
autoload :TestCase

View file

@ -278,8 +278,6 @@ db_namespace = namespace :db do
when 'oci', 'oracle'
ActiveRecord::Base.establish_connection(current_config)
File.open(filename, "w:utf-8") { |f| f << ActiveRecord::Base.connection.structure_dump }
when 'sqlserver'
`smoscript -s #{current_config['host']} -d #{current_config['database']} -u #{current_config['username']} -p #{current_config['password']} -f #{filename} -A -U`
else
ActiveRecord::Tasks::DatabaseTasks.structure_dump(current_config, filename)
end
@ -297,8 +295,6 @@ db_namespace = namespace :db do
current_config = ActiveRecord::Tasks::DatabaseTasks.current_config
filename = ENV['DB_STRUCTURE'] || File.join(Rails.root, "db", "structure.sql")
case current_config['adapter']
when 'sqlserver'
`sqlcmd -S #{current_config['host']} -d #{current_config['database']} -U #{current_config['username']} -P #{current_config['password']} -i #{filename}`
when 'oci', 'oracle'
ActiveRecord::Base.establish_connection(current_config)
IO.read(filename).split(";\n\n").each do |ddl|
@ -363,12 +359,6 @@ db_namespace = namespace :db do
task :purge => [:environment, :load_config] do
abcs = ActiveRecord::Base.configurations
case abcs['test']['adapter']
when 'sqlserver'
test = abcs.deep_dup['test']
test_database = test['database']
test['database'] = 'master'
ActiveRecord::Base.establish_connection(test)
ActiveRecord::Base.connection.recreate_database!(test_database)
when "oci", "oracle"
ActiveRecord::Base.establish_connection(:test)
ActiveRecord::Base.connection.structure_drop.split(";\n\n").each do |ddl|

View file

@ -18,7 +18,9 @@ module ActiveRecord
register_task(/mysql/, ActiveRecord::Tasks::MySQLDatabaseTasks)
register_task(/postgresql/, ActiveRecord::Tasks::PostgreSQLDatabaseTasks)
register_task(/sqlite/, ActiveRecord::Tasks::SQLiteDatabaseTasks)
register_task(/firebird/, ActiveRecord::Tasks::FirebirdDatabaseTasks)
register_task(/firebird/, ActiveRecord::Tasks::FirebirdDatabaseTasks)
register_task(/sqlserver/, ActiveRecord::Tasks::SqlserverDatabaseTasks)
def current_config(options = {})
options.reverse_merge! :env => Rails.env

View file

@ -0,0 +1,47 @@
require 'shellwords'
module ActiveRecord
module Tasks # :nodoc:
class SqlserverDatabaseTasks # :nodoc:
delegate :connection, :establish_connection, to: ActiveRecord::Base
def initialize(configuration)
@configuration = configuration
end
def create
$stderr.puts 'sorry, your database adapter is not supported yet, feel free to submit a patch'
end
def drop
$stderr.puts 'sorry, your database adapter is not supported yet, feel free to submit a patch'
end
def purge
test = configuration.deep_dup
test_database = test['database']
test['database'] = 'master'
establish_connection(test)
connection.recreate_database!(test_database)
end
def charset
$stderr.puts 'sorry, your database adapter is not supported yet, feel free to submit a patch'
end
def structure_dump(filename)
Kernel.system("smoscript -s #{configuration['host']} -d #{configuration['database']} -u #{configuration['username']} -p #{configuration['password']} -f #{filename} -A -U")
end
def structure_load(filename)
Kernel.system("sqlcmd -S #{configuration['host']} -d #{configuration['database']} -U #{configuration['username']} -P #{configuration['password']} -i #{filename}")
end
private
def configuration
@configuration
end
end
end
end

View file

@ -0,0 +1,80 @@
require 'cases/helper'
module ActiveRecord
module SqlserverSetupper
def setup
@database = 'db.sqlserver'
@connection = stub :connection
@configuration = {
'adapter' => 'sqlserver',
'database' => @database,
'host' => 'localhost',
'username' => 'username',
'password' => 'password',
}
ActiveRecord::Base.stubs(:connection).returns(@connection)
ActiveRecord::Base.stubs(:establish_connection).returns(true)
end
end
class SqlserverDBCreateTest < ActiveRecord::TestCase
include SqlserverSetupper
def test_db_retrieves_create
message = capture(:stderr) do
ActiveRecord::Tasks::DatabaseTasks.create @configuration
end
assert_match(/not supported/, message)
end
end
class SqlserverDBDropTest < ActiveRecord::TestCase
include SqlserverSetupper
def test_db_retrieves_drop
message = capture(:stderr) do
ActiveRecord::Tasks::DatabaseTasks.drop @configuration
end
assert_match(/not supported/, message)
end
end
class SqlserverDBCharsetAndCollationTest < ActiveRecord::TestCase
include SqlserverSetupper
def test_db_retrieves_collation
assert_raise NoMethodError do
ActiveRecord::Tasks::DatabaseTasks.collation @configuration
end
end
def test_db_retrieves_charset
message = capture(:stderr) do
ActiveRecord::Tasks::DatabaseTasks.charset @configuration
end
assert_match(/not supported/, message)
end
end
class SqlserverStructureDumpTest < ActiveRecord::TestCase
include SqlserverSetupper
def test_structure_dump
filename = "sqlserver.sql"
Kernel.expects(:system).with("smoscript -s localhost -d #{@database} -u username -p password -f #{filename} -A -U")
ActiveRecord::Tasks::DatabaseTasks.structure_dump(@configuration, filename)
end
end
class SqlserverStructureLoadTest < ActiveRecord::TestCase
include SqlserverSetupper
def test_structure_load
filename = "sqlserver.sql"
Kernel.expects(:system).with("sqlcmd -S localhost -d #{@database} -U username -P password -i #{filename}")
ActiveRecord::Tasks::DatabaseTasks.structure_load(@configuration, filename)
end
end
end