0
@@ -25,23 +25,31 @@ elsif rails_path = current_file.rails_path_for(choice.to_sym)
0
   end
0
 
0
   if !rails_path.exists? 
0
-    if !TextMate.message_ok_cancel("Create missing #{rails_path.basename}?")
0
+    if choice.to_sym != :view && !TextMate.message_ok_cancel("Create missing #{rails_path.basename}?")
0
       TextMate.exit_discard
0
-    end
0
-    rails_path.touch    
0
-    if choice.to_sym == :controller
0
-      generated_code = <<-RUBY
0
-class #{rails_path.controller_name.camelize}Controller < ApplicationController
0
-end
0
-RUBY
0
-      rails_path.append generated_code
0
-    elsif choice.to_sym == :helper
0
-      generated_code = <<-RUBY
0
-module #{rails_path.controller_name.camelize}Helper
0
-end
0
-RUBY
0
-      rails_path.append generated_code
0
-    end
0
+    end                             
0
+    generated_code = case choice.to_sym
0
+    when :controller 
0
+      "class #{rails_path.controller_name.camelize}Controller < ApplicationController\nend"
0
+    when :helper
0
+      "module #{rails_path.controller_name.camelize}Helper\nend"
0
+    when :unit_test
0
+      "require File.dirname(__FILE__) + '/../test_helper'
0
+
0
+class #{Inflector.singularize(rails_path.controller_name).camelize}Test < ActiveSupport::TestCase
0
+  # Replace this with your real tests.
0
+  def test_truth
0
+    assert true
0
+  end
0
+end"   
0
+    when :functional_test
0
+      "require File.dirname(__FILE__) + '/../test_helper'
0
+
0
+class #{rails_path.controller_name.camelize}ControllerTest < ActionController::TestCase     
0
+end"
0
+    end 
0
+    rails_path.touch
0
+    rails_path.append generated_code if generated_code
0
     TextMate.refresh_project_drawer
0
   end
0