database_cleaner/examples/lib/activerecord_models.rb
2010-05-30 21:12:58 -06:00

41 lines
1.5 KiB
Ruby

require 'active_record'
databases_config = {
"one" => {"adapter" => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", "database" => "#{DB_DIR}/activerecord_one.db"},
"two" => {"adapter" => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", "database" => "#{DB_DIR}/activerecord_two.db"}
}
File.open("#{File.dirname(__FILE__)}/../config/database.yml", 'w') do |file|
file.write(YAML.dump(databases_config))
end
["two","one"].each do |db|
ActiveRecord::Base.establish_connection(databases_config[db])
ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widgets"')
ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widget_using_database_ones"')
ActiveRecord::Base.connection.execute('DROP TABLE IF EXISTS "active_record_widget_using_database_twos"')
ActiveRecord::Schema.define(:version => 1) do
create_table :active_record_widgets do |t|
t.string :name
end
create_table :active_record_widget_using_database_ones do |t|
t.string :name
end
create_table :active_record_widget_using_database_twos do |t|
t.string :name
end
end
end
class ActiveRecordWidget < ActiveRecord::Base
end
class ActiveRecordWidgetUsingDatabaseOne < ActiveRecord::Base
establish_connection(:adapter => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", :database => "#{DB_DIR}/activerecord_one.db")
end
class ActiveRecordWidgetUsingDatabaseTwo < ActiveRecord::Base
establish_connection(:adapter => "#{"jdbc" if defined?(JRUBY_VERSION)}sqlite3", :database => "#{DB_DIR}/activerecord_two.db")
end