Fixing the format_handler tests so that they clean up after themselves.

Signed-off-by: Luke Kanies <luke@madstop.com>

1 files changed, 7 insertions, 0 deletions

diff --git a/spec/unit/network/format_handler.rb b/spec/unit/network/format_handler.rb
index 0b885ca8d..d142c3177 100755
--- a/spec/unit/network/format_handler.rb
+++ b/spec/unit/network/format_handler.rb
@@ -29,6 +29,13 @@ class FormatTester
 end
 
 describe Puppet::Network::FormatHandler do
+    after do
+        formats = Puppet::Network::FormatHandler.instance_variable_get("@formats")
+        formats.each do |name, format|
+            formats.delete(name) unless format.is_a?(Puppet::Network::Format)
+        end
+    end
+
     it "should be able to test whether a format is supported" do
         FormatTester.should respond_to(:support_format?)
     end