Adding setup/teardown hooks to rspec, so we can use test/unit methods

1 files changed, 7 insertions, 0 deletions

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index aa56fd93e..a4171bb07 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -10,4 +10,11 @@ require 'puppettest'
 
 Spec::Runner.configure do |config|
   config.mock_with :mocha
+  config.prepend_before :each do
+      setup() if respond_to? :setup
+  end
+
+  config.prepend_after :each do
+      teardown() if respond_to? :teardown
+  end
 end