Merge pull request #309 from anomalousthought/reset_method

Add reset method to mock classes

lib/fog/compute/bluebox.rb

@@ -33,6 +33,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/go_grid.rb

@@ -36,6 +36,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/linode.rb

@@ -31,6 +31,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/new_servers.rb

@@ -25,6 +25,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/rackspace.rb

@@ -53,6 +53,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/slicehost.rb

@@ -34,6 +34,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           unless options.delete(:provider)
             location = caller.first

lib/fog/compute/storm_on_demand.rb

@@ -56,6 +56,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def self.reset_data(keys=data.keys)
           for key in [*keys]
             data.delete(key)

lib/fog/compute/voxel.rb

@@ -48,6 +48,10 @@ module Fog
           end
         end
 
+        def self.reset
+          @data = nil
+        end
+
         def initialize(options={})
           @voxel_api_key = options[:voxel_api_key]
           @data = self.class.data[@voxel_api_key]

lib/fog/core/mock.rb

@@ -63,6 +63,16 @@ module Fog
       selection
     end
 
+    def self.reset      
+      providers = Fog.providers.map{|p| eval("Fog::#{p}")}
+      providers.select!{|m| m.constants.include?(:Compute)}
+
+      providers.each do |provider|
+        next unless provider::Compute::Mock.respond_to?(:reset)
+        provider::Compute::Mock.reset
+      end
+    end
+
   end
 
 end