From f297003f96b37edd2dbbe529876af0567e409745 Mon Sep 17 00:00:00 2001
From: Mike Gunderloy <MikeG1@larkfarm.com>
Date: Thu, 17 Sep 2009 06:40:32 -0500
Subject: [PATCH] Require the appropriate mocking library

---
 lark_template.rb                     | 4 ++++
 patterns/default/test/test_helper.rb | 1 +
 2 files changed, 5 insertions(+)

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'