Spec'd a reset() method for clearing out known routes. Uses the unregister method so that any hooks there will be run. Probably a violation of YAGNI, but I'm willing to suffer it :-)

1 files changed, 16 insertions, 0 deletions

diff --git a/spec/unit/network/rest_server.rb b/spec/unit/network/rest_server.rb
index d18086fff..986f95239 100644
--- a/spec/unit/network/rest_server.rb
+++ b/spec/unit/network/rest_server.rb
@@ -50,6 +50,22 @@ describe Puppet::Network::RESTServer, "in general" do
     Puppet::Network::RESTServer.unregister(:foo)    
     Proc.new { Puppet::Network::RESTServer.unregister(:foo) }.should raise_error(ArgumentError)
   end
+  
+  it "should allow clearing out the list of all indirection accessible to clients" do
+    Puppet::Network::RESTServer.register(:foo, :bar)
+    Puppet::Network::RESTServer.reset
+    [ :foo, :bar, :baz].each do |indirection|
+      Proc.new { Puppet::Network::RESTServer.unregister(indirection) }.should raise_error(ArgumentError)
+    end
+  end
+  
+  it "should use the normal means of turning off indirections accessible to clients when clearing all indirections" do
+    Puppet::Network::RESTServer.register(:foo, :bar, :baz)
+    Puppet::Network::RESTServer.expects(:unregister).with do |args|
+      args.include?(:foo) && args.include?(:bar) && args.include?(:baz)
+    end
+    Puppet::Network::RESTServer.reset
+  end
 end
 
 describe Puppet::Network::RESTServer, "when listening is not turned on" do