diff --git a/lark_template.rb b/lark_template.rb index 1a8ba2c..992e99c 100644 --- 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 index 62b56cb..6c51857 100644 --- 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'