Change current sstructure.

railties/bin/gen

@@ -14,16 +14,18 @@ else ARGV.size == 1
   ARGV << "--help"
 end
 
-require File.dirname(__FILE__) + '/../lib/generator'
+require File.dirname(__FILE__) + '/../lib/generators'
 
-Rails::Generators.builtin.each do |name|
-  require "generator/generators/#{name}/#{name}_generator"
+Rails::Generators.builtin.each do |group, name|
+  require "generators/#{group}/#{name}/#{name}_generator"
 end
 
 name = ARGV.shift
 
 if klass = Thor::Util.find_by_namespace("rails:generators:#{name}")
   klass.start
+elsif klass = Thor::Util.find_by_namespace(name)
+  klass.start
 else
   puts "Could not find generator #{name}."
 end

railties/bin/rails

@@ -9,7 +9,7 @@ end
 
 ARGV << "--help" if ARGV.empty?
 
-require File.dirname(__FILE__) + '/../lib/generator'
-require 'generator/generators/app/app_generator'
+require File.dirname(__FILE__) + '/../lib/generators'
+require 'generators/rails/app/app_generator'
 
 Rails::Generators::AppGenerator.start

railties/lib/generators.rb

@@ -10,15 +10,16 @@ require 'thor'
 $:.unshift(File.dirname(__FILE__))
 require 'rails/version' unless defined?(Rails::VERSION)
 
-require 'generator/base'
-require 'generator/named_base'
+require 'generators/base'
+require 'generators/named_base'
 
 module Rails
   module Generators
     def self.builtin
-      Dir[File.dirname(__FILE__) + '/generator/generators/*'].collect do |file|
-        File.basename(file)
+      Dir[File.dirname(__FILE__) + '/generators/*/*'].collect do |file|
+        file.split('/')[-2, 2]
       end
     end
   end
 end
+

railties/lib/generators/base.rb

@@ -1,4 +1,4 @@
-require 'generator/actions'
+require 'generators/actions'
 
 module Rails
   module Generators
@@ -12,7 +12,7 @@ module Rails
       # Automatically sets the source root based on the class name.
       #
       def self.source_root
-        @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'generators', generator_name, 'templates'))
+        @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'rails', generator_name, 'templates'))
       end
 
       # Convenience method to get the namespace from the class name.

railties/lib/generators/named_base.rb

@@ -1,5 +1,5 @@
-require 'generator/base'
-require 'generator/generated_attribute'
+require 'generators/base'
+require 'generators/generated_attribute'
 
 module Rails
   module Generators

railties/test/generator/generator_test_helper.rb

@@ -1,55 +0,0 @@
-require 'test/unit'
-require 'fileutils'
-
-$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../lib"
-
-# For this while, let's load all generators by hand
-require 'generator/generators/app/app_generator'
-
-class GeneratorTestCase < Test::Unit::TestCase
-  include FileUtils
-
-  def destination_root
-    @destination_root ||= File.expand_path("#{File.dirname(__FILE__)}/../fixtures/tmp")
-  end
-
-  def setup
-    mkdir_p(destination_root)
-    rm_rf(destination_root)
-  end
-
-  def test_truth
-    # don't complain, test/unit
-  end
-
-  def capture(stream)
-    begin
-      stream = stream.to_s
-      eval "$#{stream} = StringIO.new"
-      yield
-      result = eval("$#{stream}").string
-    ensure 
-      eval("$#{stream} = #{stream.upcase}")
-    end
-
-    result
-  end
-  alias :silence :capture
-
-  def assert_file(relative, content=nil)
-    absolute = File.join(destination_root, relative)
-    assert File.exists?(absolute)
-
-    case content
-      when String
-        assert_equal content, File.read(absolute)
-      when Regexp
-        assert_match content, File.read(absolute)
-    end
-  end
-
-  def assert_no_file(relative, content=nil)
-    absolute = File.join(destination_root, relative)
-    assert !File.exists?(absolute)
-  end
-end

railties/test/generators/actions_test.rb

@@ -1,5 +1,5 @@
 require 'abstract_unit'
-require 'generator/generator_test_helper'
+require 'generators/generator_test_helper'
 
 class ActionsTest < GeneratorTestCase
   def setup
@@ -170,7 +170,7 @@ class ActionsTest < GeneratorTestCase
   protected
 
     def run_generator
-      silence(:stdout) { Rails::Generators::App.start [destination_root] }
+      silence(:stdout) { Rails::Generators::AppGenerator.start [destination_root] }
     end
 
     def generator(config={})

railties/test/generators/app_test.rb

@@ -1,5 +1,5 @@
 require 'abstract_unit'
-require 'generator/generator_test_helper'
+require 'generators/generator_test_helper'
 
 class AppTest < GeneratorTestCase
 

railties/test/generators/generator_test_helper.rb

@@ -1,303 +1,56 @@
 require 'test/unit'
 require 'fileutils'
 
-# Mock out what we need from AR::Base
-module ActiveRecord
-  class Base
-    class << self
-      attr_accessor :pluralize_table_names, :timestamped_migrations
-    end
-    self.pluralize_table_names = true
-    self.timestamped_migrations = true
-  end
-
-  module ConnectionAdapters
-    class Column
-      attr_reader :name, :default, :type, :limit, :null, :sql_type, :precision, :scale
-
-      def initialize(name, default, sql_type = nil)
-        @name = name
-        @default = default
-        @type = @sql_type = sql_type
-      end
-
-      def human_name
-        @name.humanize
-      end
-    end
-  end
-end
-
-# Mock up necessities from ActionView
-module ActionView
-  module Helpers
-    module ActionRecordHelper; end
-    class InstanceTag; end
-  end
-end
-
-# Set RAILS_ROOT appropriately fixture generation
-tmp_dir = "#{File.dirname(__FILE__)}/../fixtures/tmp"
-
-if defined? RAILS_ROOT
-  RAILS_ROOT.replace tmp_dir
-else
-  RAILS_ROOT = tmp_dir
-end
-FileUtils.mkdir_p RAILS_ROOT
-
 $LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../lib"
-require 'initializer'
 
-# Mocks out the configuration
-module Rails
-  def self.configuration
-    Rails::Configuration.new
-  end
-end
-
-require 'rails_generator'
+# For this while, let's load all generators by hand
+require 'generators'
+require 'generators/rails/app/app_generator'
 
 class GeneratorTestCase < Test::Unit::TestCase
   include FileUtils
 
-  def setup
-    ActiveRecord::Base.pluralize_table_names = true
-
-    mkdir_p "#{RAILS_ROOT}/app/views/layouts"
-    mkdir_p "#{RAILS_ROOT}/config"
-    mkdir_p "#{RAILS_ROOT}/db"
-    mkdir_p "#{RAILS_ROOT}/test/fixtures"
-    mkdir_p "#{RAILS_ROOT}/public/stylesheets"
-
-    File.open("#{RAILS_ROOT}/config/routes.rb", 'w') do |f|
-      f << "ActionController::Routing::Routes.draw do |map|\n\nend"
-    end
+  def destination_root
+    @destination_root ||= File.expand_path("#{File.dirname(__FILE__)}/../fixtures/tmp")
   end
 
-  def teardown
-    rm_rf "#{RAILS_ROOT}/app"
-    rm_rf "#{RAILS_ROOT}/test"
-    rm_rf "#{RAILS_ROOT}/config"
-    rm_rf "#{RAILS_ROOT}/db"
-    rm_rf "#{RAILS_ROOT}/public"
+  def setup
+    mkdir_p(destination_root)
+    rm_rf(destination_root)
   end
 
   def test_truth
     # don't complain, test/unit
   end
 
-  # Instantiates the Generator.
-  def build_generator(name, params)
-    Rails::Generator::Base.instance(name, params)
-  end
+  def capture(stream)
+    begin
+      stream = stream.to_s
+      eval "$#{stream} = StringIO.new"
+      yield
+      result = eval("$#{stream}").string
+    ensure 
+      eval("$#{stream} = #{stream.upcase}")
+    end
 
-  # Runs the +create+ command (like the command line does).
-  def run_generator(name, params)
-    silence_generator do
-      build_generator(name, params).command(:create).invoke!
+    result
+  end
+  alias :silence :capture
+
+  def assert_file(relative, content=nil)
+    absolute = File.join(destination_root, relative)
+    assert File.exists?(absolute)
+
+    case content
+      when String
+        assert_equal content, File.read(absolute)
+      when Regexp
+        assert_match content, File.read(absolute)
     end
   end
 
-  # Silences the logger temporarily and returns the output as a String.
-  def silence_generator
-    logger_original = Rails::Generator::Base.logger
-    myout = StringIO.new
-    Rails::Generator::Base.logger = Rails::Generator::SimpleLogger.new(myout)
-    yield if block_given?
-    Rails::Generator::Base.logger = logger_original
-    myout.string
-  end
-
-  # Asserts that the given controller was generated.
-  # It takes a name or symbol without the <tt>_controller</tt> part and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_controller_for(name, parent = "ApplicationController")
-    assert_generated_class "app/controllers/#{name.to_s.underscore}_controller", parent do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given model was generated.
-  # It takes a name or symbol and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_model_for(name, parent = "ActiveRecord::Base")
-    assert_generated_class "app/models/#{name.to_s.underscore}", parent do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given helper was generated.
-  # It takes a name or symbol without the <tt>_helper</tt> part.
-  # The contents of the module source file is passed to a block.
-  def assert_generated_helper_for(name)
-    assert_generated_module "app/helpers/#{name.to_s.underscore}_helper" do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given functional test was generated.
-  # It takes a name or symbol without the <tt>_controller_test</tt> part and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_functional_test_for(name, parent = "ActionController::TestCase")
-    assert_generated_class "test/functional/#{name.to_s.underscore}_controller_test",parent do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given helper test test was generated.
-  # It takes a name or symbol without the <tt>_helper_test</tt> part and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_helper_test_for(name, parent = "ActionView::TestCase")
-    path = "test/unit/helpers/#{name.to_s.underscore}_helper_test"
-    # Have to pass the path without the "test/" part so that class_name_from_path will return a correct result
-    class_name = class_name_from_path(path.gsub(/^test\//, ''))
-
-    assert_generated_class path,parent,class_name do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given unit test was generated.
-  # It takes a name or symbol without the <tt>_test</tt> part and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_unit_test_for(name, parent = "ActiveSupport::TestCase")
-    assert_generated_class "test/unit/#{name.to_s.underscore}_test", parent do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given file was generated.
-  # The contents of the file is passed to a block.
-  def assert_generated_file(path)
-    assert_file_exists(path)
-    File.open("#{RAILS_ROOT}/#{path}") do |f|
-      yield f.read if block_given?
-    end
-  end
-
-  # asserts that the given file exists
-  def assert_file_exists(path)
-    assert File.exist?("#{RAILS_ROOT}/#{path}"),
-      "The file '#{RAILS_ROOT}/#{path}' should exist"
-  end
-
-  # Asserts that the given class source file was generated.
-  # It takes a path without the <tt>.rb</tt> part and an optional super class.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_class(path, parent = nil, class_name = class_name_from_path(path))
-    assert_generated_file("#{path}.rb") do |body|
-      assert_match /class #{class_name}#{parent.nil? ? '':" < #{parent}"}/, body, "the file '#{path}.rb' should be a class"
-      yield body if block_given?
-    end
-  end
-
-  def class_name_from_path(path)
-    # FIXME: Sucky way to detect namespaced classes
-    if path.split('/').size > 3
-      path =~ /\/?(\d+_)?(\w+)\/(\w+)$/
-      "#{$2.camelize}::#{$3.camelize}"
-    else
-      path =~ /\/?(\d+_)?(\w+)$/
-      $2.camelize
-    end
-  end
-
-  # Asserts that the given module source file was generated.
-  # It takes a path without the <tt>.rb</tt> part.
-  # The contents of the class source file is passed to a block.
-  def assert_generated_module(path)
-    # FIXME: Sucky way to detect namespaced modules
-    if path.split('/').size > 3
-      path =~ /\/?(\w+)\/(\w+)$/
-      module_name = "#{$1.camelize}::#{$2.camelize}"
-    else
-      path =~ /\/?(\w+)$/
-      module_name = $1.camelize
-    end
-
-    assert_generated_file("#{path}.rb") do |body|
-      assert_match /module #{module_name}/, body, "the file '#{path}.rb' should be a module"
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given CSS stylesheet file was generated.
-  # It takes a path without the <tt>.css</tt> part.
-  # The contents of the stylesheet source file is passed to a block.
-  def assert_generated_stylesheet(path)
-    assert_generated_file("public/stylesheets/#{path}.css") do |body|
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given YAML file was generated.
-  # It takes a path without the <tt>.yml</tt> part.
-  # The parsed YAML tree is passed to a block.
-  def assert_generated_yaml(path)
-    assert_generated_file("#{path}.yml") do |body|
-      yaml = YAML.load(body)
-      assert yaml, 'YAML data missing'
-      yield yaml if block_given?
-    end
-  end
-
-  # Asserts that the given fixtures YAML file was generated.
-  # It takes a fixture name without the <tt>.yml</tt> part.
-  # The parsed YAML tree is passed to a block.
-  def assert_generated_fixtures_for(name)
-    assert_generated_yaml "test/fixtures/#{name.to_s.underscore}" do |yaml|
-      yield yaml if block_given?
-    end
-  end
-
-  # Asserts that the given views were generated.
-  # It takes a controller name and a list of views (including extensions).
-  # The body of each view is passed to a block.
-  def assert_generated_views_for(name, *actions)
-    actions.each do |action|
-      assert_generated_file("app/views/#{name.to_s.underscore}/#{action}") do |body|
-        yield body if block_given?
-      end
-    end
-  end
-
-  def assert_generated_migration(name, parent = "ActiveRecord::Migration")
-    file = Dir.glob("#{RAILS_ROOT}/db/migrate/*_#{name.to_s.underscore}.rb").first
-    file = file.match(/db\/migrate\/[0-9]+_\w+/).to_s
-    assert_generated_class file, parent do |body|
-      assert_match /timestamps/, body, "should have timestamps defined"
-      yield body if block_given?
-    end
-  end
-
-  # Asserts that the given migration file was not generated.
-  # It takes the name of the migration as a parameter.
-  def assert_skipped_migration(name)
-    migration_file = "#{RAILS_ROOT}/db/migrate/001_#{name.to_s.underscore}.rb"
-    assert !File.exist?(migration_file), "should not create migration #{migration_file}"
-  end
-
-  # Asserts that the given resource was added to the routes.
-  def assert_added_route_for(name)
-    assert_generated_file("config/routes.rb") do |body|
-      assert_match /map.resources :#{name.to_s.underscore}/, body,
-        "should add route for :#{name.to_s.underscore}"
-    end
-  end
-
-  # Asserts that the given methods are defined in the body.
-  # This does assume standard rails code conventions with regards to the source code.
-  # The body of each individual method is passed to a block.
-  def assert_has_method(body, *methods)
-    methods.each do |name|
-      assert body =~ /^  def #{name}(\(.+\))?\n((\n|   .*\n)*)  end/, "should have method #{name}"
-      yield(name, $2) if block_given?
-    end
-  end
-
-  # Asserts that the given column is defined in the migration.
-  def assert_generated_column(body, name, type)
-    assert_match /t\.#{type.to_s} :#{name.to_s}/, body, "should have column #{name.to_s} defined"
+  def assert_no_file(relative, content=nil)
+    absolute = File.join(destination_root, relative)
+    assert !File.exists?(absolute)
   end
 end

railties/test/rails_generator/generator_test_helper.rb

@@ -0,0 +1,303 @@
+require 'test/unit'
+require 'fileutils'
+
+# Mock out what we need from AR::Base
+module ActiveRecord
+  class Base
+    class << self
+      attr_accessor :pluralize_table_names, :timestamped_migrations
+    end
+    self.pluralize_table_names = true
+    self.timestamped_migrations = true
+  end
+
+  module ConnectionAdapters
+    class Column
+      attr_reader :name, :default, :type, :limit, :null, :sql_type, :precision, :scale
+
+      def initialize(name, default, sql_type = nil)
+        @name = name
+        @default = default
+        @type = @sql_type = sql_type
+      end
+
+      def human_name
+        @name.humanize
+      end
+    end
+  end
+end
+
+# Mock up necessities from ActionView
+module ActionView
+  module Helpers
+    module ActionRecordHelper; end
+    class InstanceTag; end
+  end
+end
+
+# Set RAILS_ROOT appropriately fixture generation
+tmp_dir = "#{File.dirname(__FILE__)}/../fixtures/tmp"
+
+if defined? RAILS_ROOT
+  RAILS_ROOT.replace tmp_dir
+else
+  RAILS_ROOT = tmp_dir
+end
+FileUtils.mkdir_p RAILS_ROOT
+
+$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../lib"
+require 'initializer'
+
+# Mocks out the configuration
+module Rails
+  def self.configuration
+    Rails::Configuration.new
+  end
+end
+
+require 'rails_generator'
+
+class GeneratorTestCase < Test::Unit::TestCase
+  include FileUtils
+
+  def setup
+    ActiveRecord::Base.pluralize_table_names = true
+
+    mkdir_p "#{RAILS_ROOT}/app/views/layouts"
+    mkdir_p "#{RAILS_ROOT}/config"
+    mkdir_p "#{RAILS_ROOT}/db"
+    mkdir_p "#{RAILS_ROOT}/test/fixtures"
+    mkdir_p "#{RAILS_ROOT}/public/stylesheets"
+
+    File.open("#{RAILS_ROOT}/config/routes.rb", 'w') do |f|
+      f << "ActionController::Routing::Routes.draw do |map|\n\nend"
+    end
+  end
+
+  def teardown
+    rm_rf "#{RAILS_ROOT}/app"
+    rm_rf "#{RAILS_ROOT}/test"
+    rm_rf "#{RAILS_ROOT}/config"
+    rm_rf "#{RAILS_ROOT}/db"
+    rm_rf "#{RAILS_ROOT}/public"
+  end
+
+  def test_truth
+    # don't complain, test/unit
+  end
+
+  # Instantiates the Generator.
+  def build_generator(name, params)
+    Rails::Generator::Base.instance(name, params)
+  end
+
+  # Runs the +create+ command (like the command line does).
+  def run_generator(name, params)
+    silence_generator do
+      build_generator(name, params).command(:create).invoke!
+    end
+  end
+
+  # Silences the logger temporarily and returns the output as a String.
+  def silence_generator
+    logger_original = Rails::Generator::Base.logger
+    myout = StringIO.new
+    Rails::Generator::Base.logger = Rails::Generator::SimpleLogger.new(myout)
+    yield if block_given?
+    Rails::Generator::Base.logger = logger_original
+    myout.string
+  end
+
+  # Asserts that the given controller was generated.
+  # It takes a name or symbol without the <tt>_controller</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_controller_for(name, parent = "ApplicationController")
+    assert_generated_class "app/controllers/#{name.to_s.underscore}_controller", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given model was generated.
+  # It takes a name or symbol and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_model_for(name, parent = "ActiveRecord::Base")
+    assert_generated_class "app/models/#{name.to_s.underscore}", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given helper was generated.
+  # It takes a name or symbol without the <tt>_helper</tt> part.
+  # The contents of the module source file is passed to a block.
+  def assert_generated_helper_for(name)
+    assert_generated_module "app/helpers/#{name.to_s.underscore}_helper" do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given functional test was generated.
+  # It takes a name or symbol without the <tt>_controller_test</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_functional_test_for(name, parent = "ActionController::TestCase")
+    assert_generated_class "test/functional/#{name.to_s.underscore}_controller_test",parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given helper test test was generated.
+  # It takes a name or symbol without the <tt>_helper_test</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_helper_test_for(name, parent = "ActionView::TestCase")
+    path = "test/unit/helpers/#{name.to_s.underscore}_helper_test"
+    # Have to pass the path without the "test/" part so that class_name_from_path will return a correct result
+    class_name = class_name_from_path(path.gsub(/^test\//, ''))
+
+    assert_generated_class path,parent,class_name do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given unit test was generated.
+  # It takes a name or symbol without the <tt>_test</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_unit_test_for(name, parent = "ActiveSupport::TestCase")
+    assert_generated_class "test/unit/#{name.to_s.underscore}_test", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given file was generated.
+  # The contents of the file is passed to a block.
+  def assert_generated_file(path)
+    assert_file_exists(path)
+    File.open("#{RAILS_ROOT}/#{path}") do |f|
+      yield f.read if block_given?
+    end
+  end
+
+  # asserts that the given file exists
+  def assert_file_exists(path)
+    assert File.exist?("#{RAILS_ROOT}/#{path}"),
+      "The file '#{RAILS_ROOT}/#{path}' should exist"
+  end
+
+  # Asserts that the given class source file was generated.
+  # It takes a path without the <tt>.rb</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_class(path, parent = nil, class_name = class_name_from_path(path))
+    assert_generated_file("#{path}.rb") do |body|
+      assert_match /class #{class_name}#{parent.nil? ? '':" < #{parent}"}/, body, "the file '#{path}.rb' should be a class"
+      yield body if block_given?
+    end
+  end
+
+  def class_name_from_path(path)
+    # FIXME: Sucky way to detect namespaced classes
+    if path.split('/').size > 3
+      path =~ /\/?(\d+_)?(\w+)\/(\w+)$/
+      "#{$2.camelize}::#{$3.camelize}"
+    else
+      path =~ /\/?(\d+_)?(\w+)$/
+      $2.camelize
+    end
+  end
+
+  # Asserts that the given module source file was generated.
+  # It takes a path without the <tt>.rb</tt> part.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_module(path)
+    # FIXME: Sucky way to detect namespaced modules
+    if path.split('/').size > 3
+      path =~ /\/?(\w+)\/(\w+)$/
+      module_name = "#{$1.camelize}::#{$2.camelize}"
+    else
+      path =~ /\/?(\w+)$/
+      module_name = $1.camelize
+    end
+
+    assert_generated_file("#{path}.rb") do |body|
+      assert_match /module #{module_name}/, body, "the file '#{path}.rb' should be a module"
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given CSS stylesheet file was generated.
+  # It takes a path without the <tt>.css</tt> part.
+  # The contents of the stylesheet source file is passed to a block.
+  def assert_generated_stylesheet(path)
+    assert_generated_file("public/stylesheets/#{path}.css") do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given YAML file was generated.
+  # It takes a path without the <tt>.yml</tt> part.
+  # The parsed YAML tree is passed to a block.
+  def assert_generated_yaml(path)
+    assert_generated_file("#{path}.yml") do |body|
+      yaml = YAML.load(body)
+      assert yaml, 'YAML data missing'
+      yield yaml if block_given?
+    end
+  end
+
+  # Asserts that the given fixtures YAML file was generated.
+  # It takes a fixture name without the <tt>.yml</tt> part.
+  # The parsed YAML tree is passed to a block.
+  def assert_generated_fixtures_for(name)
+    assert_generated_yaml "test/fixtures/#{name.to_s.underscore}" do |yaml|
+      yield yaml if block_given?
+    end
+  end
+
+  # Asserts that the given views were generated.
+  # It takes a controller name and a list of views (including extensions).
+  # The body of each view is passed to a block.
+  def assert_generated_views_for(name, *actions)
+    actions.each do |action|
+      assert_generated_file("app/views/#{name.to_s.underscore}/#{action}") do |body|
+        yield body if block_given?
+      end
+    end
+  end
+
+  def assert_generated_migration(name, parent = "ActiveRecord::Migration")
+    file = Dir.glob("#{RAILS_ROOT}/db/migrate/*_#{name.to_s.underscore}.rb").first
+    file = file.match(/db\/migrate\/[0-9]+_\w+/).to_s
+    assert_generated_class file, parent do |body|
+      assert_match /timestamps/, body, "should have timestamps defined"
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given migration file was not generated.
+  # It takes the name of the migration as a parameter.
+  def assert_skipped_migration(name)
+    migration_file = "#{RAILS_ROOT}/db/migrate/001_#{name.to_s.underscore}.rb"
+    assert !File.exist?(migration_file), "should not create migration #{migration_file}"
+  end
+
+  # Asserts that the given resource was added to the routes.
+  def assert_added_route_for(name)
+    assert_generated_file("config/routes.rb") do |body|
+      assert_match /map.resources :#{name.to_s.underscore}/, body,
+        "should add route for :#{name.to_s.underscore}"
+    end
+  end
+
+  # Asserts that the given methods are defined in the body.
+  # This does assume standard rails code conventions with regards to the source code.
+  # The body of each individual method is passed to a block.
+  def assert_has_method(body, *methods)
+    methods.each do |name|
+      assert body =~ /^  def #{name}(\(.+\))?\n((\n|   .*\n)*)  end/, "should have method #{name}"
+      yield(name, $2) if block_given?
+    end
+  end
+
+  # Asserts that the given column is defined in the migration.
+  def assert_generated_column(body, name, type)
+    assert_match /t\.#{type.to_s} :#{name.to_s}/, body, "should have column #{name.to_s} defined"
+  end
+end