Module TestHelpers::Generation

# File railties/test/isolation/abstract_unit.rb, line 246
def add_to_config(str)
  environment = File.read("#{app_path}/config/application.rb")
  if environment =~ /(\n\s*end\s*end\s*)\Z/
    File.open("#{app_path}/config/application.rb", 'w') do |f|
      f.puts $` + "\n#{str}\n" + $1
    end
  end
end

# File railties/test/isolation/abstract_unit.rb, line 255
def add_to_env_config(env, str)
  environment = File.read("#{app_path}/config/environments/#{env}.rb")
  if environment =~ /(\n\s*end\s*)\Z/
    File.open("#{app_path}/config/environments/#{env}.rb", 'w') do |f|
      f.puts $` + "\n#{str}\n" + $1
    end
  end
end

# File railties/test/isolation/abstract_unit.rb, line 237
def add_to_top_of_config(str)
  environment = File.read("#{app_path}/config/application.rb")
  if environment =~ /(Rails::Application\s*)/
    File.open("#{app_path}/config/application.rb", 'w') do |f|
      f.puts $` + $1 + "\n#{str}\n" + $'
    end
  end
end

# File railties/test/isolation/abstract_unit.rb, line 271
def app_file(path, contents, mode = 'w')
  FileUtils.mkdir_p File.dirname("#{app_path}/#{path}")
  File.open("#{app_path}/#{path}", mode) do |f|
    f.puts contents
  end
end

# File railties/test/isolation/abstract_unit.rb, line 302
def boot_rails
  require File.expand_path('../../../../load_paths', __FILE__)
end

# File railties/test/isolation/abstract_unit.rb, line 95
def build_app(options = {})
  @prev_rails_env = ENV['RAILS_ENV']
  ENV['RAILS_ENV']       =   "development"
  ENV['SECRET_KEY_BASE'] ||= SecureRandom.hex(16)

  FileUtils.rm_rf(app_path)
  FileUtils.cp_r(app_template_path, app_path)

  # Delete the initializers unless requested
  unless options[:initializers]
    Dir["#{app_path}/config/initializers/*.rb"].each do |initializer|
      File.delete(initializer)
    end
  end

  gemfile_path = "#{app_path}/Gemfile"
  if options[:gemfile].blank? && File.exist?(gemfile_path)
    File.delete gemfile_path
  end

  routes = File.read("#{app_path}/config/routes.rb")
  if routes =~ /(\n\s*end\s*)\Z/
    File.open("#{app_path}/config/routes.rb", 'w') do |f|
      f.puts $` + "\nmatch ':controller(/:action(/:id))(.:format)', via: :all\n" + $1
    end
  end

  File.open("#{app_path}/config/database.yml", "w") do |f|
    f.puts <<-YAML
    default: &default
      adapter: sqlite3
      pool: 5
      timeout: 5000
    development:
      <<: *default
      database: db/development.sqlite3
    test:
      <<: *default
      database: db/test.sqlite3
    production:
      <<: *default
      database: db/production.sqlite3
    YAML
  end

  add_to_config <<-RUBY
    config.eager_load = false
    config.session_store :cookie_store, key: "_myapp_session"
    config.active_support.deprecation = :log
    config.active_support.test_order = :random
    config.action_controller.allow_forgery_protection = false
    config.log_level = :info
  RUBY
end

# File railties/test/isolation/abstract_unit.rb, line 288
def controller(name, contents)
  app_file("app/controllers/#{name}_controller.rb", contents)
end

# File railties/test/isolation/abstract_unit.rb, line 214
def engine(name)
  dir = "#{app_path}/random/#{name}"
  FileUtils.mkdir_p(dir)

  app = File.readlines("#{app_path}/config/application.rb")
  app.insert(2, "$:.unshift(\"#{dir}/lib\")")
  app.insert(3, "require #{name.inspect}")

  File.open("#{app_path}/config/application.rb", 'r+') do |f|
    f.puts app
  end

  Bukkit.new(dir).tap do |bukkit|
    yield bukkit if block_given?
  end
end

# File railties/test/isolation/abstract_unit.rb, line 278
def gsub_app_file(path, regexp, *args, &block)
  path = "#{app_path}/#{path}"
  content = File.read(path).gsub(regexp, *args, &block)
  File.open(path, 'wb') { |f| f.write(content) }
end

# File railties/test/isolation/abstract_unit.rb, line 156
def make_basic_app
  require "rails"
  require "action_controller/railtie"
  require "action_view/railtie"

  app = Class.new(Rails::Application)
  app.config.eager_load = false
  app.secrets.secret_key_base = "3b7cd727ee24e8444053437c36cc66c4"
  app.config.session_store :cookie_store, key: "_myapp_session"
  app.config.active_support.deprecation = :log
  app.config.active_support.test_order = :random
  app.config.log_level = :info

  yield app if block_given?
  app.initialize!

  app.routes.draw do
    get "/" => "omg#index"
  end

  require 'rack/test'
  extend ::Rack::Test::Methods
end

# File railties/test/isolation/abstract_unit.rb, line 284
def remove_file(path)
  FileUtils.rm_rf "#{app_path}/#{path}"
end

# File railties/test/isolation/abstract_unit.rb, line 264
def remove_from_config(str)
  file = "#{app_path}/config/application.rb"
  contents = File.read(file)
  contents.sub!(/#{str}/, "")
  File.open(file, "w+") { |f| f.puts contents }
end

# File railties/test/isolation/abstract_unit.rb, line 231
def script(script)
  Dir.chdir(app_path) do
    %x`#{Gem.ruby} #{app_path}/bin/rails #{script}`
  end
end

# File railties/test/isolation/abstract_unit.rb, line 180
def simple_controller
  controller :foo, <<-RUBY
    class FooController < ApplicationController
      def index
        render text: "foo"
      end
    end
  RUBY

  app_file 'config/routes.rb', <<-RUBY
    Rails.application.routes.draw do
      get ':controller(/:action)'
    end
  RUBY
end

# File railties/test/isolation/abstract_unit.rb, line 150
def teardown_app
  ENV['RAILS_ENV'] = @prev_rails_env if @prev_rails_env
end

# File railties/test/isolation/abstract_unit.rb, line 292
def use_frameworks(arr)
  to_remove = [:actionmailer, :activerecord] - arr

  if to_remove.include?(:activerecord)
    remove_from_config 'config.active_record.*'
  end

  $:.reject! {|path| path =~ %r'/(#{to_remove.join('|')})/' }
end