mock_redis 0.5.4 → 0.31.0

data/spec/commands/sort_set_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe '#sort(key, options)' do
+  before do
+    @key = 'mock-redis-test:set_sort'
+
+    @redises.sadd(@key, '1')
+    @redises.sadd(@key, '2')
+
+    @redises.set('mock-redis-test:values_1', 'a')
+    @redises.set('mock-redis-test:values_2', 'b')
+
+    @redises.set('mock-redis-test:weight_1', '2')
+    @redises.set('mock-redis-test:weight_2', '1')
+
+    @redises.hset('mock-redis-test:hash_1', 'key', 'x')
+    @redises.hset('mock-redis-test:hash_2', 'key', 'y')
+  end
+
+  it_should_behave_like 'a sortable'
+end

data/spec/commands/sort_zset_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe '#sort(key, options)' do
+  before do
+    @key = 'mock-redis-test:zset_sort'
+
+    @redises.zadd(@key, 100, '1')
+    @redises.zadd(@key, 99, '2')
+
+    @redises.set('mock-redis-test:values_1', 'a')
+    @redises.set('mock-redis-test:values_2', 'b')
+
+    @redises.set('mock-redis-test:weight_1', '2')
+    @redises.set('mock-redis-test:weight_2', '1')
+
+    @redises.hset('mock-redis-test:hash_1', 'key', 'x')
+    @redises.hset('mock-redis-test:hash_2', 'key', 'y')
+  end
+
+  it_should_behave_like 'a sortable'
+end

data/spec/commands/spop_spec.rb

@@ -7,19 +7,34 @@ describe '#spop(key)' do
     @redises.sadd(@key, 'value')
   end
 
-  it "returns a member of the set" do
+  it 'returns a member of the set' do
     @redises.spop(@key).should == 'value'
   end
 
-  it "removes a member of the set" do
+  it 'removes a member of the set' do
     @redises.spop(@key)
     @redises.smembers(@key).should == []
   end
 
-  it "returns nil if the set is empty" do
+  it 'returns nil if the set is empty' do
     @redises.spop(@key)
     @redises.spop(@key).should be_nil
   end
 
-  it_should_behave_like "a set-only command"
+  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/srandmember_spec.rb

@@ -7,19 +7,43 @@ describe '#srandmember(key)' do
     @redises.sadd(@key, 'value')
   end
 
-  it "returns a member of the set" do
+  it 'returns a member of the set' do
     @redises.srandmember(@key).should == 'value'
   end
 
-  it "does not modify the set" do
+  it 'does not modify the set' do
     @redises.srandmember(@key)
     @redises.smembers(@key).should == ['value']
   end
 
-  it "returns nil if the set is empty" do
+  it 'returns nil if the set is empty' do
     @redises.spop(@key)
     @redises.srandmember(@key).should be_nil
   end
 
-  it_should_behave_like "a set-only command"
+  context 'when count argument is specified' do
+    before do
+      @redises.sadd(@key, 'value2')
+      @redises.sadd(@key, 'value3')
+    end
+
+    # NOTE: We disable result checking since MockRedis and Redis will likely
+    # return a different random set (since the selection is, well, random)
+    it 'returns the whole set if count is greater than the set size' do
+      @redises.send_without_checking(:srandmember, @key, 5).should =~ %w[value value2 value3]
+    end
+
+    it 'returns random members up to count from the set when count is smaller than the set size' do
+      @redises.send_without_checking(:srandmember, @key, 2).size.should == 2
+    end
+
+    it 'returns random members up to count from the set when count is negative even if count.abs is greater than the set size' do # rubocop:disable Layout/LineLength
+      @redises.send_without_checking(:srandmember, @key, -5).size.should == 5
+    end
+
+    it 'returns nil if the set is empty' do
+      @redises.srem(@key, %w[value value2 value3])
+      @redises.srandmember(@key, 2).should be_empty
+    end
+  end
 end

data/spec/commands/srem_spec.rb

@@ -8,33 +8,38 @@ describe '#srem(key, member)' do
     @redises.sadd(@key, 'ernie')
   end
 
-  it "returns true if member is in the set" do
-    @redises.srem(@key, 'bert').should be_true
+  it 'returns true if member is in the set' do
+    @redises.srem(@key, 'bert').should == true
   end
 
-  it "returns false if member is not in the set" do
-    @redises.srem(@key, 'cookiemonster').should be_false
+  it 'returns false if member is not in the set' do
+    @redises.srem(@key, 'cookiemonster').should == false
   end
 
-  it "removes member from the set" do
+  it 'removes member from the set' do
     @redises.srem(@key, 'ernie')
     @redises.smembers(@key).should == ['bert']
   end
 
-  it "stringifies member" do
+  it 'stringifies member' do
     @redises.sadd(@key, '1')
-    @redises.srem(@key, 1).should be_true
+    @redises.srem(@key, 1).should == true
   end
 
-  it "cleans up empty sets" do
-    @redises.smembers(@key).each {|m| @redises.srem(@key, m)}
+  it 'cleans up empty sets' do
+    @redises.smembers(@key).each { |m| @redises.srem(@key, m) }
     @redises.get(@key).should be_nil
   end
 
-  it "supports a variable number of arguments" do
-    @redises.srem(@key, ['bert', 'ernie']).should == 2
+  it 'supports a variable number of arguments' do
+    @redises.srem(@key, %w[bert ernie]).should == 2
     @redises.get(@key).should be_nil
   end
 
-  it_should_behave_like "a set-only command"
+  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

data/spec/commands/sscan_each_spec.rb

@@ -0,0 +1,48 @@
+require 'spec_helper'
+
+describe '#sscan_each' do
+  subject { MockRedis::Database.new(self) }
+
+  let(:key) { 'mock-redis-test:sscan_each' }
+
+  before do
+    allow(subject).to receive(:smembers).and_return(collection)
+  end
+
+  context 'when no keys are found' do
+    let(:collection) { [] }
+
+    it 'does not iterate over any elements' do
+      results = subject.sscan_each(key).map do |k|
+        k
+      end
+      expect(results).to match_array(collection)
+    end
+  end
+
+  context 'when keys are found' do
+    context 'when no match filter is supplied' do
+      let(:collection) { Array.new(20) { |i| "k#{i}" } }
+
+      it 'iterates over each item in the collection' do
+        results = subject.sscan_each(key).map do |k|
+          k
+        end
+        expect(results).to match_array(collection)
+      end
+    end
+
+    context 'when giving a custom match filter' do
+      let(:match) { 'k1*' }
+      let(:collection) { Array.new(12) { |i| "k#{i}" } }
+      let(:expected) { %w[k1 k10 k11] }
+
+      it 'iterates over each item in the filtered collection' do
+        results = subject.sscan_each(key, match: match).map do |k|
+          k
+        end
+        expect(results).to match_array(expected)
+      end
+    end
+  end
+end

data/spec/commands/sscan_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+describe '#sscan' do
+  let(:count) { 10 }
+  let(:match) { '*' }
+  let(:key) { 'mock-redis-test:sscan' }
+
+  before do
+    # The return order of the members is non-deterministic, so we sort them to compare
+    expect_any_instance_of(Redis).to receive(:sscan).and_wrap_original do |m, *args|
+      result = m.call(*args)
+      [result[0], result[1].sort]
+    end
+    expect_any_instance_of(MockRedis).to receive(:sscan).and_wrap_original do |m, *args|
+      result = m.call(*args)
+      [result[0], result[1].sort]
+    end
+  end
+
+  context 'when the set does not exist' do
+    it 'returns a 0 cursor and an empty collection' do
+      expect(@redises.sscan(key, 0, count: count, match: match)).to eq(['0', []])
+    end
+  end
+
+  context 'when the set exists' do
+    before do
+      @redises.sadd(key, 'Hello')
+      @redises.sadd(key, 'World')
+      @redises.sadd(key, 'Test')
+    end
+
+    let(:expected) { ['0', %w[Hello Test World]] }
+
+    it 'returns a 0 cursor and the collection' do
+      expect(@redises.sscan(key, 0, count: count)).to eq(expected)
+    end
+  end
+end

data/spec/commands/strlen_spec.rb

@@ -1,19 +1,18 @@
-# -*- coding: utf-8 -*-
 require 'spec_helper'
 
-describe "#strlen(key)" do
+describe '#strlen(key)' do
   before do
     @key = 'mock-redis-test:73288'
-    @redises.set(@key, "5 ∈ (0..10)")
+    @redises.set(@key, '5 ∈ (0..10)')
   end
 
   it "returns the string's length in bytes" do
     @redises.strlen(@key).should == 13
   end
 
-  it "returns 0 for a nonexistent value" do
+  it 'returns 0 for a nonexistent value' do
     @redises.strlen('mock-redis-test:does-not-exist').should == 0
   end
 
-  it_should_behave_like "a string-only command"
+  it_should_behave_like 'a string-only command'
 end

data/spec/commands/sunion_spec.rb

@@ -5,36 +5,38 @@ describe '#sunion(key [, key, ...])' do
     @evens   = 'mock-redis-test:sunion:evens'
     @primes  = 'mock-redis-test:sunion:primes'
 
-    [2, 4, 6, 8, 10].each {|i| @redises.sadd(@evens, i) }
-    [2, 3, 5, 7].each {|i| @redises.sadd(@primes, i) }
+    [2, 4, 6, 8, 10].each { |i| @redises.sadd(@evens, i) }
+    [2, 3, 5, 7].each { |i| @redises.sadd(@primes, i) }
   end
 
-  it "returns the elements in the resulting set" do
+  it 'returns the elements in the resulting set' do
     @redises.sunion(@evens, @primes).should == %w[2 4 6 8 10 3 5 7]
   end
 
-  it "treats missing keys as empty sets" do
+  it 'treats missing keys as empty sets' do
     @redises.sunion(@primes, 'mock-redis-test:nonesuch').
       should == %w[2 3 5 7]
   end
 
-  it "raises an error if given 0 sets" do
+  it 'allows Array as argument' do
+    @redises.sunion([@evens, @primes]).should == %w[2 4 6 8 10 3 5 7]
+  end
+
+  it 'raises an error if given 0 sets' do
     lambda do
       @redises.sunion
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
   end
 
-  it "raises an error if any argument is not a a set" do
+  it 'raises an error if any argument is not a a set' do
     @redises.set('mock-redis-test:notset', 1)
 
     lambda do
       @redises.sunion(@numbers, 'mock-redis-test:notset')
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
 
     lambda do
       @redises.sunion('mock-redis-test:notset', @numbers)
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
   end
 end
-
-

data/spec/commands/sunionstore_spec.rb

@@ -6,54 +6,54 @@ describe '#sunionstore(destination, key [, key, ...])' do
     @primes      = 'mock-redis-test:sunionstore:primes'
     @destination = 'mock-redis-test:sunionstore:destination'
 
-    [2, 4, 6, 8, 10].each {|i| @redises.sadd(@evens, i) }
-    [2, 3, 5, 7].each {|i| @redises.sadd(@primes, i) }
+    [2, 4, 6, 8, 10].each { |i| @redises.sadd(@evens, i) }
+    [2, 3, 5, 7].each { |i| @redises.sadd(@primes, i) }
   end
 
-  it "returns the number of elements in the resulting set" do
+  it 'returns the number of elements in the resulting set' do
     @redises.sunionstore(@destination, @primes, @evens).should == 8
   end
 
-  it "stores the resulting set" do
+  it 'stores the resulting set' do
     @redises.sunionstore(@destination, @primes, @evens)
     @redises.smembers(@destination).should == %w[10 8 6 4 7 5 3 2]
   end
 
-  it "does not store empty sets" do
+  it 'does not store empty sets' do
     @redises.sunionstore(@destination,
       'mock-redis-test:nonesuch',
       'mock-redis-test:nonesuch2').should == 0
     @redises.get(@destination).should be_nil
   end
 
-  it "removes existing elements in destination" do
+  it 'removes existing elements in destination' do
     @redises.sadd(@destination, 42)
 
     @redises.sunionstore(@destination, @primes)
     @redises.smembers(@destination).should == %w[7 5 3 2]
   end
 
-  it "correctly unions and stores when the destination is empty and is one of the arguments" do
+  it 'correctly unions and stores when the destination is empty and is one of the arguments' do
     @redises.sunionstore(@destination, @destination, @primes)
 
     @redises.smembers(@destination).should == %w[7 5 3 2]
   end
 
-  it "raises an error if given 0 sets" do
+  it 'raises an error if given 0 sets' do
     lambda do
       @redises.sunionstore(@destination)
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
   end
 
-  it "raises an error if any argument is not a a set" do
+  it 'raises an error if any argument is not a a set' do
     @redises.set('mock-redis-test:notset', 1)
 
     lambda do
       @redises.sunionstore(@destination, @primes, 'mock-redis-test:notset')
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
 
     lambda do
       @redises.sunionstore(@destination, 'mock-redis-test:notset', @primes)
-    end.should raise_error(RuntimeError)
+    end.should raise_error(Redis::CommandError)
   end
 end

data/spec/commands/ttl_spec.rb

@@ -1,20 +1,25 @@
 require 'spec_helper'
 
-describe "#ttl(key)" do
+describe '#ttl(key)' do
   before do
     @key = 'mock-redis-test:persist'
     @redises.set(@key, 'spork')
   end
 
-  it "returns -1 for a key with no expiration" do
+  it 'returns -1 for a key with no expiration' do
     @redises.ttl(@key).should == -1
   end
 
-  it "returns -1 for a key that does not exist" do
-    @redises.ttl('mock-redis-test:nonesuch').should == -1
+  it 'returns -2 for a key that does not exist' do
+    @redises.ttl('mock-redis-test:nonesuch').should == -2
   end
 
-  context "[mock only]" do
+  it 'stringifies key' do
+    @redises.expire(@key, 9)
+    @redises.ttl(@key.to_sym).should > 0
+  end
+
+  context '[mock only]' do
     # These are mock-only since we can't actually manipulate time in
     # the real Redis.
 
@@ -24,7 +29,7 @@ describe "#ttl(key)" do
 
     before do
       @now = Time.now
-      Time.stub!(:now).and_return(@now)
+      Time.stub(:now).and_return(@now)
     end
 
     it "gives you the key's remaining lifespan in seconds" do

data/spec/commands/type_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe "#type(key)" do
+describe '#type(key)' do
   before do
     @key = 'mock-redis-test:type'
   end

data/spec/commands/watch_spec.rb

@@ -1,12 +1,17 @@
 require 'spec_helper'
 
-describe '#watch(key)' do
-  it "responds with nil" do
-    @redises.watch('mock-redis-test').should be_nil
+describe '#watch(key [, key, ...)' do
+  before do
+    @key1 = 'mock-redis-test-watch1'
+    @key2 = 'mock-redis-test-watch2'
+  end
+
+  it "returns 'OK'" do
+    @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,122 @@
+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
+    # TODO: Redis appears to be returning a timestamp a few seconds in the future
+    # so we're ignoring the last 5 digits (time in milliseconds)
+    @redises._gsub(/\d{5}-\d/, '....-.')
+  end
+
+  it 'returns an id based on the timestamp' do
+    t = Time.now.to_i
+    id = @redises.xadd(@key, { key: 'value' })
+    expect(id).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 must be greater than 0-0'
+      )
+  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
+
+  it 'supports a maxlen greater than the current size' do
+    @redises.xadd(@key, { key1: 'value1' }, id: '1234567891234-0')
+    @redises.xadd(@key, { key2: 'value2' }, id: '1234567891245-0', maxlen: 1000)
+    expect(@redises.xrange(@key, '-', '+')).to eq(
+      [
+        ['1234567891234-0', { 'key1' => 'value1' }],
+        ['1234567891245-0', { 'key2' => 'value2' }],
+      ]
+    )
+  end
+
+  it 'creates an empty stream with maxlen of 0' do
+    @redises.xadd(@key, { key: 'value' }, maxlen: 0)
+    expect(@redises.xlen(@key)).to eq 0
+    expect(@redises.xrange(@key, '-', '+')).to eq([])
+    expect(@redises.exists?(@key)).to eq true
+  end
+end

data/spec/commands/xlen_spec.rb

@@ -0,0 +1,22 @@
+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
+    # TODO: Redis appears to be returning a timestamp a few seconds in the future
+    # so we're ignoring the last 5 digits (time in milliseconds)
+    @redises._gsub(/\d{5}-\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