Fix hset for array of key + value

sds · Nov 14, 2024 · 28f6dc3 · 28f6dc3
1 parent 30eb1bc
commit 28f6dc3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/mock_redis/hash_methods.rb b/lib/mock_redis/hash_methods.rb
@@ -137,6 +137,7 @@ def hscan_each(key, opts = {}, &block)
 
     def hset(key, *args)
       added = 0
+      args = args.flatten
       with_hash_at(key) do |hash|
         if args.length == 1 && args[0].is_a?(Hash)
           args = args[0].to_a.flatten

diff --git a/spec/commands/hset_spec.rb b/spec/commands/hset_spec.rb
@@ -38,5 +38,17 @@
     expect(@redises.hset(@key, { 'k1' => 'v1', 'k2' => 'v2' })).to eq(2)
   end
 
+  it 'stores array values correctly' do
+    @redises.hset(@key, ['k1', 'v1', 'k2', 'v2'])
+    expect(@redises.hget(@key, 'k1')).to eq('v1')
+    expect(@redises.hget(@key, 'k2')).to eq('v2')
+  end
+
+  it 'stores multiple arguments correctly' do
+    @redises.hset(@key, 'k1', 'v1', 'k2', 'v2')
+    expect(@redises.hget(@key, 'k1')).to eq('v1')
+    expect(@redises.hget(@key, 'k2')).to eq('v2')
+  end
+
   it_should_behave_like 'a hash-only command'
 end