Clean up class collisions check and a class method helper.

1 files changed, 7 insertions, 8 deletions

diff --git a/railties/lib/generators/rails/performance_test/performance_test_generator.rb b/railties/lib/generators/rails/performance_test/performance_test_generator.rb
index bf83b354a8..b8efda8df3 100644
--- a/railties/lib/generators/rails/performance_test/performance_test_generator.rb
+++ b/railties/lib/generators/rails/performance_test/performance_test_generator.rb
@@ -1,12 +1,11 @@
 module Rails
   module Generators
-    class PerformanceTestGenerator < NamedBase

-      def check_class_collisions
-        class_collisions "#{class_name}Test"
-      end
+    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

-    end
  end
+      def create_test_files
+        template 'performance_test.rb', File.join('test/performance', class_path, "#{file_name}_test.rb")
+      end
+    end
+  end
 end