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:
parent
4a370f9d7c
commit
0f313fce1c
5 changed files with 131 additions and 11 deletions
|
@ -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
|
||||
|
|
|
@ -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|
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
80
activerecord/test/cases/tasks/sqlserver_rake_test.rb
Normal file
80
activerecord/test/cases/tasks/sqlserver_rake_test.rb
Normal 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
|
Loading…
Reference in a new issue