1 files changed, 156 insertions, 0 deletions
diff --git a/railties/test/generators_test.rb b/railties/test/generators_test.rb
new file mode 100644
index 0000000000..89d52dd170
--- /dev/null
+++ b/railties/test/generators_test.rb
@@ -0,0 +1,156 @@
+require File.join(File.dirname(__FILE__), 'generators', 'generators_test_helper')
+require 'generators/rails/model/model_generator'
+require 'generators/test_unit/model/model_generator'
+require 'mocha'
+
+class GeneratorsTest < GeneratorsTestCase
+  def test_invoke_when_generator_is_not_found
+    output = capture(:stdout){ Rails::Generators.invoke :unknown }
+    assert_equal "Could not find generator unknown.\n", output
+  end
+
+  def test_help_when_a_generator_with_required_arguments_is_invoked_without_arguments
+    output = capture(:stdout){ Rails::Generators.invoke :model, [] }
+    assert_match /Description:/, output
+  end
+
+  def test_invoke_with_default_values
+    Rails::Generators::ModelGenerator.expects(:start).with(["Account"], {})
+    Rails::Generators.invoke :model, ["Account"]
+  end
+
+  def test_invoke_with_config_values
+    Rails::Generators::ModelGenerator.expects(:start).with(["Account"], :behavior => :skip)
+    Rails::Generators.invoke :model, ["Account"], :behavior => :skip
+  end
+
+  def test_find_by_namespace_without_base_or_context
+    assert_nil Rails::Generators.find_by_namespace(:model)
+  end
+
+  def test_find_by_namespace_with_base
+    klass = Rails::Generators.find_by_namespace(:model, :rails)
+    assert klass
+    assert_equal "rails:generators:model", klass.namespace
+  end
+
+  def test_find_by_namespace_with_context
+    klass = Rails::Generators.find_by_namespace(:test_unit, nil, :model)
+    assert klass
+    assert_equal "test_unit:generators:model", klass.namespace
+  end
+
+  def test_find_by_namespace_add_generators_to_raw_lookups
+    klass = Rails::Generators.find_by_namespace("test_unit:model")
+    assert klass
+    assert_equal "test_unit:generators:model", klass.namespace
+  end
+
+  def test_find_by_namespace_lookup_to_the_rails_root_folder
+    klass = Rails::Generators.find_by_namespace(:fixjour)
+    assert klass
+    assert_equal "fixjour", klass.namespace
+  end
+
+  def test_find_by_namespace_lookup_to_deep_rails_root_folders
+    klass = Rails::Generators.find_by_namespace(:fixjour, :active_record)
+    assert klass
+    assert_equal "active_record:generators:fixjour", klass.namespace
+  end
+
+  def test_find_by_namespace_lookup_traverse_folders
+    klass = Rails::Generators.find_by_namespace(:javascripts, :rails)
+    assert klass
+    assert_equal "rails:generators:javascripts", klass.namespace
+  end
+
+  def test_find_by_namespace_lookup_to_vendor_folders
+    klass = Rails::Generators.find_by_namespace(:mspec)
+    assert klass
+    assert_equal "mspec", klass.namespace
+  end
+
+  def test_builtin_generators
+    assert Rails::Generators.builtin.include? %w(rails model)
+  end
+
+  def test_rails_generators_help_with_builtin_information
+    output = capture(:stdout){ Rails::Generators.help }
+    assert_match /model/, output
+    assert_match /scaffold_controller/, output
+  end
+
+  def test_rails_generators_with_others_information
+    output = capture(:stdout){ Rails::Generators.help }.split("\n").last
+    assert_equal "Others: active_record:fixjour, fixjour, mspec, rails:javascripts, wrong.", output
+  end
+
+  def test_warning_is_shown_if_generator_cant_be_loaded
+    output = capture(:stderr){ Rails::Generators.find_by_namespace(:wrong) }
+    assert_match /\[WARNING\] Could not load generator at/, output
+    assert_match /Error: uninitialized constant Rails::Generator/, output
+  end
+
+  def test_no_color_sets_proper_shell
+    Rails::Generators.no_color!
+    assert_equal Thor::Shell::Basic, Thor::Base.shell
+  ensure
+    Thor::Base.shell = Thor::Shell::Color
+  end
+
+  def test_rails_root_templates
+    template = File.join(RAILS_ROOT, "lib", "templates", "active_record", "model", "model.rb")
+
+    # Create template
+    mkdir_p(File.dirname(template))
+    File.open(template, 'w'){ |f| f.write "empty" }
+
+    output = capture(:stdout) do
+      Rails::Generators.invoke :model, ["user"], :destination_root => destination_root
+    end
+
+    assert_file "app/models/user.rb" do |content|
+      assert_equal "empty", content
+    end
+  ensure
+    rm_rf File.dirname(template)
+  end
+
+  def test_fallbacks_for_generators_on_find_by_namespace
+    Rails::Generators.fallbacks[:remarkable] = :test_unit
+    klass = Rails::Generators.find_by_namespace(:plugin, :remarkable)
+    assert klass
+    assert_equal "test_unit:generators:plugin", klass.namespace
+  end
+
+  def test_fallbacks_for_generators_on_invoke
+    Rails::Generators.fallbacks[:shoulda] = :test_unit
+    TestUnit::Generators::ModelGenerator.expects(:start).with(["Account"], {})
+    Rails::Generators.invoke "shoulda:model", ["Account"]
+  end
+
+  def test_nested_fallbacks_for_generators
+    Rails::Generators.fallbacks[:super_shoulda] = :shoulda
+    TestUnit::Generators::ModelGenerator.expects(:start).with(["Account"], {})
+    Rails::Generators.invoke "super_shoulda:model", ["Account"]
+  end
+
+  def test_developer_options_are_overwriten_by_user_options
+    Rails::Generators.options[:new_generator] = { :generate => false }
+
+    klass = Class.new(Rails::Generators::Base) do
+      def self.name
+        "NewGenerator"
+      end
+
+      class_option :generate, :default => true
+    end
+
+    assert_equal false, klass.class_options[:generate].default
+  end
+
+  def test_source_paths_for_not_namespaced_generators
+    mspec = Rails::Generators.find_by_namespace :mspec
+    assert mspec.source_paths.include?(File.join(RAILS_ROOT, "lib", "templates", "mspec"))
+  end
+end