1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activerecord/test/cases/migration/helper.rb

39 lines
1,019 B
Ruby

require "cases/helper"
module ActiveRecord
class Migration
class << self; attr_accessor :message_count; end
self.message_count = 0
module TestHelper
attr_reader :connection, :table_name
CONNECTION_METHODS = %w[add_column remove_column rename_column add_index change_column rename_table column_exists? index_exists? add_reference add_belongs_to remove_reference remove_references remove_belongs_to]
class TestModel < ActiveRecord::Base
self.table_name = :test_models
end
def setup
super
@connection = ActiveRecord::Base.connection
connection.create_table :test_models do |t|
t.timestamps null: true
end
TestModel.reset_column_information
end
def teardown
super
TestModel.reset_table_name
TestModel.reset_sequence_name
connection.drop_table :test_models, if_exists: true
end
private
delegate(*CONNECTION_METHODS, to: :connection)
end
end
end