mock_redis 0.19.0 → 0.20.0

data/spec/commands/mset_spec.rb

@@ -10,6 +10,14 @@ describe '#mset(key, value [, key, value, ...])' do
     @redises.mset(@key1, 1).should == 'OK'
   end
 
+  it "responds with 'OK' for passed Array with 1 item" do
+    @redises.mset([@key1, 1]).should == 'OK'
+  end
+
+  it "responds with 'OK' for passed Array with 2 items" do
+    @redises.mset([@key1, 1, @key2, 2]).should == 'OK'
+  end
+
   it 'sets the values' do
     @redises.mset(@key1, 'value1', @key2, 'value2')
     @redises.mget(@key1, @key2).should == %w[value1 value2]
@@ -26,4 +34,10 @@ describe '#mset(key, value [, key, value, ...])' do
       @redises.mset
     end.should raise_error(Redis::CommandError)
   end
+
+  it 'raises an error if given odd-sized array' do
+    lambda do
+      @redises.mset([@key1, 1, @key2])
+    end.should raise_error(Redis::CommandError)
+  end
 end

data/spec/commands/pipelined_spec.rb

@@ -39,4 +39,56 @@ describe '#pipelined' do
       future.class.should be MockRedis::Future
     end
   end
+
+  context 'with pipelined operations returning array replies' do
+    let(:key1) { 'colors' }
+    let(:key2) { 'seasons' }
+    let(:value1) { %w[blue yellow] }
+    let(:value2) { %w[summer winter] }
+
+    before do
+      @redises.rpush(key1, value1)
+      @redises.rpush(key2, value2)
+    end
+
+    after do
+      @redises.del key1
+      @redises.del key2
+    end
+
+    it 'returns an array of the array replies' do
+      results = @redises.pipelined do |_redis|
+        @redises.lrange(key1, 0, -1)
+        @redises.lrange(key2, 0, -1)
+      end
+
+      results.should == [value1, value2]
+    end
+  end
+
+  context 'with nested pipelines' do
+    let(:key1)   { 'hello' }
+    let(:key2)   { 'world' }
+    let(:value1) { 'foo' }
+    let(:value2) { 'bar' }
+
+    before do
+      @redises.set key1, value1
+      @redises.set key2, value2
+    end
+
+    it 'returns results of the nested pipelines' do
+      results = @redises.pipelined do |pr1|
+        pr1.pipelined do |pr2|
+          pr2.get key1
+        end
+
+        pr1.pipelined do |pr2|
+          pr2.get key2
+        end
+      end
+
+      results.should == [value1, value2]
+    end
+  end
 end

data/spec/commands/spop_spec.rb

@@ -21,5 +21,20 @@ describe '#spop(key)' do
     @redises.spop(@key).should be_nil
   end
 
+  it 'returns an array if count is not nil' do
+    @redises.sadd(@key, 'value2')
+    @redises.spop(@key, 2).should == %w[value value2]
+  end
+
+  it 'returns only whats in the set' do
+    @redises.spop(@key, 2).should == ['value']
+    @redises.smembers(@key).should == []
+  end
+
+  it 'returns an empty array if count is not nil and the set it empty' do
+    @redises.spop(@key)
+    @redises.spop(@key, 100).should == []
+  end
+
   it_should_behave_like 'a set-only command'
 end

data/spec/commands/watch_spec.rb

@@ -1,12 +1,17 @@
 require 'spec_helper'
 
-describe '#watch(key)' do
+describe '#watch(key [, key, ...)' do
+  before do
+    @key1 = 'mock-redis-test-watch1'
+    @key2 = 'mock-redis-test-watch2'
+  end
+
   it "returns 'OK'" do
-    @redises.watch('mock-redis-test').should == 'OK'
+    @redises.watch(@key1, @key2).should == 'OK'
   end
 
   it 'EXECs its MULTI on successes' do
-    @redises.watch 'foo' do
+    @redises.watch @key1, @key2 do
       @redises.multi do |multi|
         multi.set 'bar', 'baz'
       end

data/spec/commands/xadd_spec.rb

@@ -0,0 +1,102 @@
+require 'spec_helper'
+
+describe '#xadd("mystream", { f1: "v1", f2: "v2" }, id: "0-0", maxlen: 1000, approximate: true)' do
+  before :all do
+    sleep 1 - (Time.now.to_f % 1)
+    @key = 'mock-redis-test:xadd'
+  end
+
+  before :each do
+    @redises._gsub(/\d{3}-\d/, '...-.')
+  end
+
+  it 'returns an id based on the timestamp' do
+    t = Time.now.to_i
+    expect(@redises.xadd(@key, key: 'value')).to match(/#{t}\d{3}-0/)
+  end
+
+  it 'adds data with symbols' do
+    @redises.xadd(@key, symbol_key: :symbol_value)
+    expect(@redises.xrange(@key, '-', '+').last[1])
+      .to eq('symbol_key' => 'symbol_value')
+  end
+
+  it 'increments the sequence number with the same timestamp' do
+    Timecop.freeze do
+      @redises.xadd(@key, key: 'value')
+      expect(@redises.xadd(@key, key: 'value')).to match(/\d+-1/)
+    end
+  end
+
+  it 'sets the id if it is given' do
+    expect(@redises.xadd(@key, { key: 'value' }, id: '1234567891234-2'))
+      .to eq '1234567891234-2'
+  end
+
+  it 'accepts is as an integer' do
+    expect(@redises.xadd(@key, { key: 'value' }, id: 1_234_567_891_234))
+      .to eq '1234567891234-0'
+  end
+
+  it 'raises exception if id is less that stream top item' do
+    @redises.xadd(@key, { key: 'value' }, id: '1234567891234-0')
+    expect { @redises.xadd(@key, { key: 'value' }, id: '1234567891233-0') }
+      .to raise_error(
+        Redis::CommandError,
+        'ERR The ID specified in XADD is equal or smaller than the target ' \
+        'stream top item'
+      )
+  end
+
+  it 'raises exception if id of 0 is added to an empty stream' do
+    expect { @redises.xadd('mock-redis-test:unknown-stream', { key: 'value' }, id: '0') }
+      .to raise_error(
+        Redis::CommandError,
+        'ERR The ID specified in XADD is equal or smaller than the target ' \
+        'stream top item'
+      )
+  end
+
+  it 'does not raise exception on empty stream with id of 0 and positive sequence number' do
+    expect { @redises.xadd('mock-redis-test:unknown-stream', { key: 'value' }, id: '0-1') }
+      .to_not raise_error
+  end
+
+  it 'caters for the current time being before the last time' do
+    t = (Time.now.to_f * 1000).to_i + 2000
+    @redises.xadd(@key, { key: 'value' }, id: "#{t}-0")
+    expect(@redises.xadd(@key, key: 'value')).to match(/#{t}-1/)
+  end
+
+  it 'appends a sequence number if it is missing' do
+    expect(@redises.xadd(@key, { key: 'value' }, id: '1234567891234'))
+      .to eq '1234567891234-0'
+  end
+
+  it 'raises an invalid stream id error' do
+    expect { @redises.xadd(@key, { key: 'value' }, id: 'X') }
+      .to raise_error(
+        Redis::CommandError,
+        'ERR Invalid stream ID specified as stream command argument'
+      )
+  end
+
+  it 'caps the stream to 5 elements' do
+    @redises.xadd(@key, { key1: 'value1' }, id: '1234567891234-0')
+    @redises.xadd(@key, { key2: 'value2' }, id: '1234567891245-0')
+    @redises.xadd(@key, { key3: 'value3' }, id: '1234567891245-1')
+    @redises.xadd(@key, { key4: 'value4' }, id: '1234567891278-0')
+    @redises.xadd(@key, { key5: 'value5' }, id: '1234567891278-1')
+    @redises.xadd(@key, { key6: 'value6' }, id: '1234567891299-0')
+    @redises.xadd(@key, { key7: 'value7' }, id: '1234567891300-0', maxlen: 5)
+    expect(@redises.xrange(@key, '-', '+')).to eq(
+      [
+        ['1234567891245-1', { 'key3' => 'value3' }],
+        ['1234567891278-0', { 'key4' => 'value4' }],
+        ['1234567891278-1', { 'key5' => 'value5' }],
+        ['1234567891299-0', { 'key6' => 'value6' }],
+        ['1234567891300-0', { 'key7' => 'value7' }]
+      ]
+    )
+  end
+end

data/spec/commands/xlen_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe '#xlen(key)' do
+  before :all do
+    sleep 1 - (Time.now.to_f % 1)
+    @key = 'mock-redis-test:xlen'
+  end
+
+  before :each do
+    @redises._gsub(/\d{3}-\d/, '...-.')
+  end
+
+  it 'returns the number of items in the stream' do
+    expect(@redises.xlen(@key)).to eq 0
+    @redises.xadd(@key, key: 'value')
+    expect(@redises.xlen(@key)).to eq 1
+    3.times { @redises.xadd(@key, key: 'value') }
+    expect(@redises.xlen(@key)).to eq 4
+  end
+end

data/spec/commands/xrange_spec.rb

@@ -0,0 +1,141 @@
+require 'spec_helper'
+
+describe '#xrange("mystream", first: "0-1", last: "0-3", count: 10)' do
+  before { @key = 'mock-redis-test:xrange' }
+
+  it 'finds an empty range' do
+    expect(@redises.xrange(@key, '-', '+')).to eq []
+  end
+
+  it 'finds a single entry with a full range' do
+    @redises.xadd(@key, { key: 'value' }, id: '1234567891234-0')
+    expect(@redises.xrange(@key, '-', '+'))
+      .to eq [['1234567891234-0', { 'key' => 'value' }]]
+  end
+
+  it 'finds multiple key/value pairs with a full range' do
+    @redises.xadd(@key, { key1: 'value1', key2: 'value2', key3: 'value3' }, id: '1234567891234-0')
+    expect(@redises.xrange(@key, '-', '+'))
+      .to eq [['1234567891234-0', { 'key1' => 'value1', 'key2' => 'value2', 'key3' => 'value3' }]]
+  end
+
+  context 'six items on the list' do
+    before :each do
+      @redises.xadd(@key, { key1: 'value1' }, id: '1234567891234-0')
+      @redises.xadd(@key, { key2: 'value2' }, id: '1234567891245-0')
+      @redises.xadd(@key, { key3: 'value3' }, id: '1234567891245-1')
+      @redises.xadd(@key, { key4: 'value4' }, id: '1234567891278-0')
+      @redises.xadd(@key, { key5: 'value5' }, id: '1234567891278-1')
+      @redises.xadd(@key, { key6: 'value6' }, id: '1234567891299-0')
+    end
+
+    it 'returns entries in sequential order' do
+      expect(@redises.xrange(@key, '-', '+')).to eq(
+        [
+          ['1234567891234-0', { 'key1' => 'value1' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891299-0', { 'key6' => 'value6' }]
+        ]
+      )
+    end
+
+    it 'returns entries with a lower limit' do
+      expect(@redises.xrange(@key, '1234567891239-0', '+')).to eq(
+        [
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891299-0', { 'key6' => 'value6' }]
+        ]
+      )
+    end
+
+    it 'returns entries with an upper limit' do
+      expect(@redises.xrange(@key, '-', '1234567891285-0')).to eq(
+        [
+          ['1234567891234-0', { 'key1' => 'value1' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }]
+        ]
+      )
+    end
+
+    it 'returns entries with both a lower and an upper limit' do
+      expect(@redises.xrange(@key, '1234567891239-0', '1234567891285-0')).to eq(
+        [
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }]
+        ]
+      )
+    end
+
+    it 'finds the list with sequence numbers' do
+      expect(@redises.xrange(@key, '1234567891245-1', '1234567891278-0')).to eq(
+        [
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }]
+        ]
+      )
+    end
+
+    it 'finds the list with lower bound without sequence numbers' do
+      expect(@redises.xrange(@key, '1234567891245', '+')).to eq(
+        [
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891299-0', { 'key6' => 'value6' }]
+        ]
+      )
+    end
+
+    it 'finds the list with upper bound without sequence numbers' do
+      expect(@redises.xrange(@key, '-', '1234567891278')).to eq(
+        [
+          ['1234567891234-0', { 'key1' => 'value1' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }]
+        ]
+      )
+    end
+
+    it 'accepts limits as integers' do
+      expect(@redises.xrange(@key, 1_234_567_891_245, 1_234_567_891_278)).to eq(
+        [
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891278-1', { 'key5' => 'value5' }]
+        ]
+      )
+    end
+
+    it 'returns a limited number of items' do
+      expect(@redises.xrange(@key, '-', '+', count: 2)).to eq(
+        [
+          ['1234567891234-0', { 'key1' => 'value1' }],
+          ['1234567891245-0', { 'key2' => 'value2' }]
+        ]
+      )
+    end
+  end
+
+  it 'raises an invalid stream id error' do
+    expect { @redises.xrange(@key, 'X', '+') }
+      .to raise_error(
+        Redis::CommandError,
+        'ERR Invalid stream ID specified as stream command argument'
+      )
+  end
+end

data/spec/commands/xrevrange_spec.rb

@@ -0,0 +1,130 @@
+require 'spec_helper'
+
+describe '#xrevrange(key, start, end)' do
+  before { @key = 'mock-redis-test:xrevrange' }
+
+  it 'finds an empty range' do
+    expect(@redises.xrevrange(@key, '-', '+')).to eq []
+  end
+
+  it 'finds a single entry with a full range' do
+    @redises.xadd(@key, { key: 'value' }, id: '1234567891234-0')
+    expect(@redises.xrevrange(@key, '+', '-'))
+      .to eq [['1234567891234-0', { 'key' => 'value' }]]
+  end
+
+  it 'finds multiple key/value pairs with a full range' do
+    @redises.xadd(@key, { key1: 'value1', key2: 'value2', key3: 'value3' }, id: '1234567891234-0')
+    expect(@redises.xrevrange(@key, '+', '-'))
+      .to eq [['1234567891234-0', { 'key1' => 'value1', 'key2' => 'value2', 'key3' => 'value3' }]]
+  end
+
+  context 'six items on the list' do
+    before :each do
+      @redises.xadd(@key, { key1: 'value1' }, id: '1234567891234-0')
+      @redises.xadd(@key, { key2: 'value2' }, id: '1234567891245-0')
+      @redises.xadd(@key, { key3: 'value3' }, id: '1234567891245-1')
+      @redises.xadd(@key, { key4: 'value4' }, id: '1234567891278-0')
+      @redises.xadd(@key, { key5: 'value5' }, id: '1234567891278-1')
+      @redises.xadd(@key, { key6: 'value6' }, id: '1234567891299-0')
+    end
+
+    it 'returns entries in sequential order' do
+      expect(@redises.xrevrange(@key, '+', '-')).to eq(
+        [
+          ['1234567891299-0', { 'key6' => 'value6' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891234-0', { 'key1' => 'value1' }],
+        ]
+      )
+    end
+
+    it 'returns entries with a lower limit' do
+      expect(@redises.xrevrange(@key, '+', '1234567891239-0')).to eq(
+        [
+          ['1234567891299-0', { 'key6' => 'value6' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+        ]
+      )
+    end
+
+    it 'returns entries with an upper limit' do
+      expect(@redises.xrevrange(@key, '1234567891285-0', '-')).to eq(
+        [
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891234-0', { 'key1' => 'value1' }],
+        ]
+      )
+    end
+
+    it 'returns entries with both a lower and an upper limit' do
+      expect(@redises.xrevrange(@key, '1234567891285-0', '1234567891239-0')).to eq(
+        [
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+        ]
+      )
+    end
+
+    it 'finds the list with sequence numbers' do
+      expect(@redises.xrevrange(@key, '1234567891278-0', '1234567891245-1')).to eq(
+        [
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+        ]
+      )
+    end
+
+    it 'finds the list with lower bound without sequence numbers' do
+      expect(@redises.xrevrange(@key, '+', '1234567891245')).to eq(
+        [
+          ['1234567891299-0', { 'key6' => 'value6' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+        ]
+      )
+    end
+
+    it 'finds the list with upper bound without sequence numbers' do
+      expect(@redises.xrevrange(@key, '1234567891278', '-')).to eq(
+        [
+          ['1234567891278-1', { 'key5' => 'value5' }],
+          ['1234567891278-0', { 'key4' => 'value4' }],
+          ['1234567891245-1', { 'key3' => 'value3' }],
+          ['1234567891245-0', { 'key2' => 'value2' }],
+          ['1234567891234-0', { 'key1' => 'value1' }],
+        ]
+      )
+    end
+
+    it 'returns a limited number of items' do
+      expect(@redises.xrevrange(@key, count: 2)).to eq(
+        [
+          ['1234567891299-0', { 'key6' => 'value6' }],
+          ['1234567891278-1', { 'key5' => 'value5' }],
+        ]
+      )
+    end
+  end
+
+  it 'raises an invalid stream id error' do
+    expect { @redises.xrevrange(@key, 'X', '-') }
+      .to raise_error(
+        Redis::CommandError,
+        'ERR Invalid stream ID specified as stream command argument'
+      )
+  end
+end