Allow mget to accept a block (#210)

sds · May 1, 2021 · 622d0b1 · 622d0b1
1 parent c129012
commit 622d0b1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/lib/mock_redis/string_methods.rb b/lib/mock_redis/string_methods.rb
@@ -154,14 +154,16 @@ def incrbyfloat(key, n)
       new_value
     end
 
-    def mget(*keys)
+    def mget(*keys, &blk)
       keys.flatten!
 
       assert_has_args(keys, 'mget')
 
-      keys.map do |key|
+      data = keys.map do |key|
         get(key) if stringy?(key)
       end
+
+      blk ? blk.call(data) : data
     end
 
     def mapped_mget(*keys)

diff --git a/spec/commands/mget_spec.rb b/spec/commands/mget_spec.rb
@@ -56,4 +56,10 @@
       end.should raise_error(Redis::CommandError)
     end
   end
+
+  context 'emulate block' do
+    it 'returns an array of values' do
+      @redises.mget(@key1, @key2) { |values| values.map(&:to_i) }.should == [1, 2]
+    end
+  end
 end