Fix srem to allow passing an array of integers as argument

sds · Jun 4, 2019 · 90f6a5a · 90f6a5a
1 parent 40f73e6
commit 90f6a5a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/mock_redis/set_methods.rb b/lib/mock_redis/set_methods.rb
@@ -117,6 +117,7 @@ def srem(key, members)
       with_set_at(key) do |s|
         if members.is_a?(Array)
           orig_size = s.size
+          members = members.map(&:to_s)
           s.delete_if { |m| members.include?(m) }
           orig_size - s.size
         else

diff --git a/spec/commands/srem_spec.rb b/spec/commands/srem_spec.rb
@@ -36,5 +36,10 @@
     @redises.get(@key).should be_nil
   end
 
+  it 'allow passing an array of integers as argument' do
+    @redises.sadd(@key, %w[1 2])
+    @redises.srem(@key, [1, 2]).should == 2
+  end
+
   it_should_behave_like 'a set-only command'
 end