Ensure that commands respects symbols or strings for keys

Change-Id: Ied52a1dbc2af72dbfe80e19cbf55ad02695fe211 Reviewed-on: https://gerrit.causes.com/17025 Tested-by: jenkins <[email protected]> Reviewed-by: Alex Wallace <[email protected]>
sds · Jan 3, 2013 · 8a78a2d · 8a78a2d
1 parent bdc98d0
commit 8a78a2d
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/lib/mock_redis/database.rb b/lib/mock_redis/database.rb
@@ -6,6 +6,7 @@
 require 'mock_redis/string_methods'
 require 'mock_redis/zset_methods'
 require 'mock_redis/sort_method'
+require 'mock_redis/indifferent_hash'
 
 class MockRedis
   class Database
@@ -20,7 +21,7 @@ class Database
 
     def initialize(base, *args)
       @base = base
-      @data = {}
+      @data = MockRedis::IndifferentHash.new
       @expire_times = []
     end
 

diff --git a/lib/mock_redis/indifferent_hash.rb b/lib/mock_redis/indifferent_hash.rb
@@ -0,0 +1,11 @@
+class MockRedis
+  class IndifferentHash < Hash
+    def [](key)
+      super(key.to_s)
+    end
+
+    def []=(key, value)
+      super(key.to_s, value)
+    end
+  end
+end
diff --git a/spec/commands/get_spec.rb b/spec/commands/get_spec.rb
@@ -19,5 +19,12 @@
     @redises.get(@key).should == "100"
   end
 
+  it "stringifies key" do
+    key = :a_symbol
+
+    @redises.set(key, 'hello')
+    @redises.get(key.to_s).should == 'hello'
+  end
+
   it_should_behave_like "a string-only command"
 end