Require the appropriate mocking library

jmcervera · Sep 17, 2009 · f297003 · f297003
1 parent 99f3d27
commit f297003
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lark_template.rb b/lark_template.rb
@@ -538,9 +538,13 @@ def generate_expectation(object_name, method_name, parameter = nil)
 
 # testing
 file 'test/exemplars/sample_exemplar.rb', load_pattern('test/exemplars/sample_exemplar.rb')
+mock_require = ""
 mock_include = ""
 if @mocking == "rr"
+  mock_require = "require 'rr'"
   mock_include = "  include RR::Adapters::TestUnit"
+elsif @mocking == "mocha"
+  mock_require = "require 'mocha'"
 end
 file 'test/test_helper.rb', load_pattern('test/test_helper.rb', 'default', binding)
 

diff --git a/patterns/default/test/test_helper.rb b/patterns/default/test/test_helper.rb
@@ -8,6 +8,7 @@
 require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
 require 'test_help'
 require 'shoulda'
+#{mock_require}
 require 'authlogic/test_case'
 require 'webrat'