Fixing a unit test for node integration

1 files changed, 2 insertions, 9 deletions

diff --git a/spec/integration/node.rb b/spec/integration/node.rb
index 87ff448e4..631d4403e 100755
--- a/spec/integration/node.rb
+++ b/spec/integration/node.rb
@@ -11,14 +11,11 @@ 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
+        @terminus = Puppet::Node.indirection.terminus(:memory)
+        Puppet::Node.indirection.stubs(:terminus).returns @terminus
         @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
@@ -44,8 +41,4 @@ describe Puppet::Node, " when using the memory terminus" do
     it "should fail when asked to destroy a node that does not exist" do
         proc { Puppet::Node.destroy(@node) }.should raise_error(ArgumentError)
     end
-
-    after do
-        Puppet.settings.clear
-    end
 end