From a3702f8974c1a08c66ca303c6b7ab4414c311159 Mon Sep 17 00:00:00 2001
From: Fernando Guillen <fguillen.mail@gmail.com>
Date: Wed, 1 Feb 2012 17:25:45 +0100
Subject: [PATCH] testing reset methods

---
 lib/time_window_drop_collector/wrappers/rails_cache.rb | 2 +-
 lib/time_window_drop_collector/wrappers/redis.rb       | 4 ++++
 test/mock_wrapper_test.rb                              | 6 ++++++
 test/rails_cache_wrapper_test.rb                       | 6 ++++++
 test/redis_wrapper_test.rb                             | 9 +++++++++
 5 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/lib/time_window_drop_collector/wrappers/rails_cache.rb b/lib/time_window_drop_collector/wrappers/rails_cache.rb
index cdf45d5..f3ffcc0 100644
--- a/lib/time_window_drop_collector/wrappers/rails_cache.rb
+++ b/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
diff --git a/lib/time_window_drop_collector/wrappers/redis.rb b/lib/time_window_drop_collector/wrappers/redis.rb
index 5b8020c..ba9fe1c 100644
--- a/lib/time_window_drop_collector/wrappers/redis.rb
+++ b/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
diff --git a/test/mock_wrapper_test.rb b/test/mock_wrapper_test.rb
index 9d3ddb3..65289b4 100644
--- a/test/mock_wrapper_test.rb
+++ b/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
\ No newline at end of file
diff --git a/test/rails_cache_wrapper_test.rb b/test/rails_cache_wrapper_test.rb
index fc286bd..e1881bd 100644
--- a/test/rails_cache_wrapper_test.rb
+++ b/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
\ No newline at end of file
diff --git a/test/redis_wrapper_test.rb b/test/redis_wrapper_test.rb
index ed611db..1c692ea 100644
--- a/test/redis_wrapper_test.rb
+++ b/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
\ No newline at end of file