Fixing an integration test so it cleans up after itself

1 files changed, 5 insertions, 0 deletions

diff --git a/spec/integration/node.rb b/spec/integration/node.rb
index e4a311998..87ff448e4 100755
--- a/spec/integration/node.rb
+++ b/spec/integration/node.rb
@@ -10,10 +10,15 @@ require 'puppet/node'
 describe Puppet::Node, " when using the memory terminus" do
     before do
         @name = "me"
+        @old_terminus = Puppet::Node.indirection.terminus_class
         Puppet::Node.terminus_class = :memory
         @node = Puppet::Node.new(@name)
     end
 
+    after do
+        Puppet::Node.terminus_class = @old_terminus
+    end
+
     it "should find no nodes by default" do
         Puppet::Node.find(@name).should be_nil
     end