Updating first portion of the Network RESTServer spec with example code, getting the added examples to pass.

1 files changed, 18 insertions, 0 deletions

diff --git a/lib/puppet/network/rest_server.rb b/lib/puppet/network/rest_server.rb
index e415e8bcb..0ffa8ff16 100644
--- a/lib/puppet/network/rest_server.rb
+++ b/lib/puppet/network/rest_server.rb
@@ -1,2 +1,20 @@
 class Puppet::Network::RESTServer # :nodoc:
+  @@routes = {}
+  
+  def self.register(*indirections)
+    raise ArgumentError, "indirection names are required" if indirections.empty?
+    indirections.flatten.each { |i| @@routes[i.to_sym] = true }
+  end
+  
+  def self.unregister(*indirections)
+    raise ArgumentError, "indirection names are required" if indirections.empty?
+    indirections.flatten.each do |i|
+      raise(ArgumentError, "indirection [%s] is not known" % i) unless @@routes[i.to_sym]
+      @@routes.delete(i.to_sym)
+    end
+  end
+  
+  def self.reset
+    self.unregister(@@routes.keys) unless @@routes.keys.empty?
+  end
 end