Add hooks for integration and performance tests.

railties/lib/generators.rb

@@ -36,9 +36,11 @@ module Generators
       :fixture => true,
       :force_plural => false,
       :helper => true,
+      :integration_tool => :test_unit,
       :layout => true,
       :migration => true,
       :orm => :active_record,
+      :performance_tool => :test_unit,
       :resource_controller => :controller,
       :scaffold_controller => :scaffold_controller,
       :singleton => false,

railties/lib/generators/rails/integration_test/integration_test_generator.rb

 module Rails
   module Generators
     class IntegrationTestGenerator < NamedBase
-      check_class_collision :suffix => "Test"
-
-      def create_test_files
-        template 'integration_test.rb', File.join('test/integration', class_path, "#{file_name}_test.rb")
-      end
+      hook_for :integration_tool, :as => :integration
     end
   end
 end

railties/lib/generators/rails/performance_test/performance_test_generator.rb

 module Rails
   module Generators
     class PerformanceTestGenerator < NamedBase
-      check_class_collision :suffix => "Test"
-
-      def create_test_files
-        template 'performance_test.rb', File.join('test/performance', class_path, "#{file_name}_test.rb")
-      end
+      hook_for :performance_tool, :as => :performance
     end
   end
 end

railties/lib/generators/test_unit/integration/integration_generator.rb

+require 'generators/test_unit'
+
+module TestUnit
+  module Generators
+    class IntegrationGenerator < Base
+      check_class_collision :suffix => "Test"
+
+      def create_test_files
+        template 'integration_test.rb', File.join('test/integration', class_path, "#{file_name}_test.rb")
+      end
+    end
+  end
+end

railties/lib/generators/test_unit/performance/performance_generator.rb

+require 'generators/test_unit'
+
+module TestUnit
+  module Generators
+    class PerformanceGenerator < Base
+      check_class_collision :suffix => "Test"
+
+      def create_test_files
+        template 'performance_test.rb', File.join('test/performance', class_path, "#{file_name}_test.rb")
+      end
+    end
+  end
+end