7263da542b
Replaced by `#lease_connection` to better reflect what it does. `ActiveRecord::Base#connection` is deprecated in the same way but without a removal timeline nor a deprecation warning. Inside the Active Record test suite, we do remove `Base.connection` to ensure it's not used internally. Some callsites have been converted to use `with_connection`, some other have been more simply migrated to `lease_connection` and will serve as a list of callsites to convert for https://github.com/rails/rails/pull/50793
68 lines
1.8 KiB
Ruby
68 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "test_helper"
|
|
require "database/setup"
|
|
|
|
class ActiveStorage::MigrationsTest < ActiveSupport::TestCase
|
|
setup do
|
|
@original_verbose = ActiveRecord::Migration.verbose
|
|
ActiveRecord::Migration.verbose = false
|
|
|
|
@connection = ActiveRecord::Base.lease_connection
|
|
@original_options = Rails.configuration.generators.options.deep_dup
|
|
end
|
|
|
|
teardown do
|
|
Rails.configuration.generators.options = @original_options
|
|
rerun_migration
|
|
ActiveRecord::Migration.verbose = @original_verbose
|
|
end
|
|
|
|
test "migration creates tables with default primary and foreign key types" do
|
|
rerun_migration
|
|
|
|
active_storage_tables.each do |table|
|
|
assert_equal :integer, primary_key(table).type
|
|
|
|
foreign_keys(table).each do |foreign_key|
|
|
assert_equal :integer, foreign_key.type
|
|
end
|
|
end
|
|
end
|
|
|
|
test "migration creates tables with configured primary and foreign key types" do
|
|
Rails.configuration.generators do |g|
|
|
g.orm :active_record, primary_key_type: :string
|
|
end
|
|
|
|
rerun_migration
|
|
|
|
active_storage_tables.each do |table|
|
|
assert_equal :string, primary_key(table).type
|
|
|
|
foreign_keys(table).each do |foreign_key|
|
|
assert_equal :string, foreign_key.type
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
def rerun_migration
|
|
CreateActiveStorageTables.migrate(:down)
|
|
CreateActiveStorageTables.migrate(:up)
|
|
end
|
|
|
|
def active_storage_tables
|
|
@active_storage_tables ||= ActiveStorage::Record.descendants.map { |klass| klass.table_name.to_sym }
|
|
end
|
|
|
|
def primary_key(table)
|
|
@connection.columns(table).find { |c| c.name == "id" }
|
|
end
|
|
|
|
def foreign_keys(table)
|
|
columns = @connection.foreign_keys(table).map(&:column)
|
|
@connection.columns(table).select { |c| columns.include?(c.name) }
|
|
end
|
|
end
|