shoulda_test_gen.rb in padrino-gen-0.1.2

- old
+ new

@@ -1,10 +1,10 @@
 module Padrino
   module Generators
     module Components
       module Tests
-        
+
         module ShouldaGen
           SHOULDA_SETUP = (<<-TEST).gsub(/^ {10}/, '')
           class Test::Unit::TestCase
             include Rack::Test::Methods
 
@@ -16,11 +16,34 @@
 
           def setup_test
             require_dependencies 'test/unit', 'shoulda', :env => :testing
             insert_test_suite_setup SHOULDA_SETUP
           end
+
+          SHOULDA_CONTROLLER_TEST = (<<-TEST).gsub(/^ {10}/, '')
+          require File.dirname(__FILE__) + '/../test_config.rb'
+
+          class !NAME!ControllerTest < Test::Unit::TestCase
+            context "!NAME!Controller" do
+              setup do
+                get '/'
+              end
+
+              should "return hello world text" do
+                assert_equal "Hello World", last_response.body
+              end
+            end
+          end
+          TEST
+
+          # Generates a controller test given the controllers name
+          def generate_controller_test(name, root)
+            shoulda_contents = SHOULDA_CONTROLLER_TEST.gsub(/!NAME!/, name.to_s.camelize)
+            create_file File.join(root, "test/controllers/#{name}_controller_test.rb"), shoulda_contents
+          end
+
         end
-        
+
       end
     end
   end
 end