mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
3ea091e1cc
Signed-off-by: Pratik Naik <pratiknaik@gmail.com>
41 lines
1.3 KiB
Ruby
41 lines
1.3 KiB
Ruby
require 'abstract_unit'
|
|
require 'generators/generators_test_helper'
|
|
require 'generators/rails/session_migration/session_migration_generator'
|
|
|
|
module ActiveRecord
|
|
module SessionStore
|
|
class Session
|
|
class << self
|
|
attr_accessor :table_name
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
class SessionMigrationGeneratorTest < GeneratorsTestCase
|
|
|
|
def test_session_migration_with_default_name
|
|
run_generator
|
|
assert_migration "db/migrate/add_sessions_table.rb", /class AddSessionsTable < ActiveRecord::Migration/
|
|
end
|
|
|
|
def test_session_migration_with_given_name
|
|
run_generator ["create_session_table"]
|
|
assert_migration "db/migrate/create_session_table.rb", /class CreateSessionTable < ActiveRecord::Migration/
|
|
end
|
|
|
|
def test_session_migration_with_custom_table_name
|
|
ActiveRecord::SessionStore::Session.table_name = "custom_table_name"
|
|
run_generator
|
|
assert_migration "db/migrate/add_sessions_table.rb" do |migration|
|
|
assert_match /class AddSessionsTable < ActiveRecord::Migration/, migration
|
|
assert_match /create_table :custom_table_name/, migration
|
|
end
|
|
end
|
|
protected
|
|
|
|
def run_generator(args=[])
|
|
silence(:stdout) { Rails::Generators::SessionMigrationGenerator.start args, :destination_root => destination_root }
|
|
end
|
|
|
|
end
|