testing reset methods

lib/time_window_drop_collector/wrappers/rails_cache.rb

@@ -18,7 +18,7 @@ def get( keys )
 			end
 
       def reset
-        client.quit
+        client.reset
       end
 		end
 	end

lib/time_window_drop_collector/wrappers/redis.rb

@@ -27,6 +27,10 @@ def get( keys )
 
         result
   		end
+
+      def reset
+        client.quit
+      end
   	end
   end
 end

test/mock_wrapper_test.rb

@@ -19,4 +19,10 @@ def test_values_for
     wrapper.client.expects( :get_multi ).with( "keys" ).returns( "keys_values" )
     assert_equal( "keys_values", wrapper.get( "keys" ))
   end
+
+  def test_reset
+    wrapper = TimeWindowDropCollector::Wrappers::Mock.new( ["arg1"] )
+    wrapper.client.expects( :flush )
+    wrapper.reset
+  end
 end

test/rails_cache_wrapper_test.rb

@@ -25,4 +25,10 @@ def test_values_for
     wrapper.client.expects( :read_multi ).with( "keys" ).returns( "keys_values" )
     assert_equal( "keys_values", wrapper.get( "keys" ))
   end
+
+  def test_reset
+    wrapper = TimeWindowDropCollector::Wrappers::RailsCache.new( ["arg1"] )
+    wrapper.client.expects( :reset )
+    wrapper.reset
+  end
 end

test/redis_wrapper_test.rb

@@ -47,4 +47,13 @@ def test_get
 
     assert_equal( { "key1" => 1, "key2" => 2 }, wrapper.get( ["key1", "key2"] ))
   end
+
+  def test_reset
+    client = mock
+    Redis.expects( :new ).returns( client )
+    wrapper = TimeWindowDropCollector::Wrappers::Redis.new( nil )
+    client.expects( :quit )
+    wrapper.reset
+  end
+
 end