mock_redis 0.0.1

data/spec/commands/ttl_spec.rb

@@ -0,0 +1,36 @@
+require 'spec_helper'
+
+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
+    @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
+  end
+
+  context "[mock only]" do
+    # These are mock-only since we can't actually manipulate time in
+    # the real Redis.
+
+    before(:all) do
+      @mock = @redises.mock
+    end
+
+    before do
+      @now = Time.now
+      Time.stub!(:now).and_return(@now)
+    end
+
+    it "gives you the key's remaining lifespan in seconds" do
+      @mock.expire(@key, 5)
+      @mock.ttl(@key).should == 4
+    end
+
+  end
+end

data/spec/commands/type_spec.rb

@@ -0,0 +1,36 @@
+require 'spec_helper'
+
+describe "#type(key)" do
+  before do
+    @key = 'mock-redis-test:type'
+  end
+
+  it "returns 'none' for no key" do
+    @redises.type(@key).should == 'none'
+  end
+
+  it "returns 'string' for a string" do
+    @redises.set(@key, 'stringlish')
+    @redises.type(@key).should == 'string'
+  end
+
+  it "returns 'list' for a list" do
+    @redises.lpush(@key, 100)
+    @redises.type(@key).should == 'list'
+  end
+
+  it "returns 'hash' for a hash" do
+    @redises.hset(@key, 100, 200)
+    @redises.type(@key).should == 'hash'
+  end
+
+  it "returns 'set' for a set" do
+    @redises.sadd(@key, 100)
+    @redises.type(@key).should == 'set'
+  end
+
+  it "returns 'zset' for a zset" do
+    @redises.zadd(@key, 1, 2)
+    @redises.type(@key).should == 'zset'
+  end
+end

data/spec/commands/unwatch_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe '#unwatch' do
+  it "responds with 'OK'" do
+    @redises.unwatch.should == 'OK'
+  end
+end

data/spec/commands/watch_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe '#watch(key)' do
+  it "responds with 'OK'" do
+    @redises.watch('mock-redis-test').should == 'OK'
+  end
+end

data/spec/commands/zadd_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe "#zadd(key, score, member)" do
+  before { @key = 'mock-redis-test:zadd' }
+
+  it "returns true if member wasn't present in the set" do
+    @redises.zadd(@key, 1, 'foo').should be_true
+  end
+
+  it "returns false if member was present in the set" do
+    @redises.zadd(@key, 1, 'foo')
+    @redises.zadd(@key, 1, 'foo').should be_false
+  end
+
+  it "adds member to the set" do
+    @redises.zadd(@key, 1, 'foo')
+    @redises.zrange(@key, 0, -1).should == ['foo']
+  end
+
+  it "updates the score" do
+    @redises.zadd(@key, 1, 'foo')
+    @redises.zadd(@key, 2, 'foo')
+
+    @redises.zscore(@key, 'foo').should == "2"
+  end
+
+  it_should_behave_like "arg 1 is a score"
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zcard_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+
+describe "#zcard(key)" do
+  before do
+    @key = 'mock-redis-test:zcard'
+  end
+
+  it "returns the number of elements in the zset" do
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zcard(@key).should == 2
+  end
+
+  it "returns 0 for nonexistent sets" do
+    @redises.zcard(@key).should == 0
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zcount_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+
+describe "#zcount(key, min, max)" do
+  before do
+    @key = 'mock-redis-test:zcount'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the number of members in the zset with scores in (min..max)" do
+    @redises.zcount(@key, 3, 10).should == 2
+  end
+
+  it "returns 0 if there are no such members" do
+    @redises.zcount(@key, 100, 200).should == 0
+  end
+
+  it_should_behave_like "arg 1 is a score"
+  it_should_behave_like "arg 2 is a score"
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zincrby_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe "#zincrby(key, increment, member)" do
+  before do
+    @key = 'mock-redis-test:zincrby'
+    @redises.zadd(@key, 1, 'bert')
+  end
+
+  it "returns the new score as a string" do
+    @redises.zincrby(@key, 10, 'bert').should == "11"
+  end
+
+  it "updates the item's score" do
+    @redises.zincrby(@key, 10, 'bert')
+    @redises.zscore(@key, 'bert').should == "11"
+  end
+
+  it "adds missing members with score increment" do
+    @redises.zincrby(@key, 5.5, 'bigbird').should == "5.5"
+  end
+
+  it_should_behave_like "arg 1 is a score"
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zinterstore_spec.rb

@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe "#zinterstore(destination, keys, [:weights => [w,w,], [:aggregate => :sum|:min|:max])" do
+  before do
+    @odds   = 'mock-redis-test:zinterstore:odds'
+    @primes = 'mock-redis-test:zinterstore:primes'
+    @dest   = 'mock-redis-test:zinterstore:dest'
+
+    @redises.zadd(@odds, 1, 'one')
+    @redises.zadd(@odds, 3, 'three')
+    @redises.zadd(@odds, 5, 'five')
+    @redises.zadd(@odds, 7, 'seven')
+    @redises.zadd(@odds, 9, 'nine')
+
+    @redises.zadd(@primes, 2, 'two')
+    @redises.zadd(@primes, 3, 'three')
+    @redises.zadd(@primes, 5, 'five')
+    @redises.zadd(@primes, 7, 'seven')
+  end
+
+  it "returns the number of elements in the new set" do
+    @redises.zinterstore(@dest, [@odds, @primes]).should == 3
+  end
+
+  it "sums the members' scores by default" do
+    @redises.zinterstore(@dest, [@odds, @primes])
+    @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+      %w[three 6 five 10 seven 14]
+  end
+
+  it "raises an error if keys is empty" do
+    lambda do
+      @redises.zinterstore(@dest, [])
+    end.should raise_error(RuntimeError)
+  end
+
+  context "the :weights argument" do
+    it "multiplies the scores by the weights while aggregating" do
+      @redises.zinterstore(@dest, [@odds, @primes], :weights => [2, 3])
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[three 15 five 25 seven 35]
+    end
+
+    it "raises an error if the number of weights != the number of keys" do
+      lambda do
+        @redises.zinterstore(@dest, [@odds, @primes], :weights => [1,2,3])
+      end.should raise_error(RuntimeError)
+    end
+  end
+
+  context "the :aggregate argument" do
+    before do
+      @smalls = 'mock-redis-test:zinterstore:smalls'
+      @bigs   = 'mock-redis-test:zinterstore:bigs'
+
+      @redises.zadd(@smalls, 1, 'bert')
+      @redises.zadd(@smalls, 2, 'ernie')
+      @redises.zadd(@bigs, 100, 'bert')
+      @redises.zadd(@bigs, 200, 'ernie')
+    end
+
+    it "aggregates scores with min when :aggregate => :min is specified" do
+      @redises.zinterstore(@dest, [@bigs, @smalls], :aggregate => :min)
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[bert 1 ernie 2]
+    end
+
+    it "aggregates scores with max when :aggregate => :max is specified" do
+      @redises.zinterstore(@dest, [@bigs, @smalls], :aggregate => :max)
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[bert 100 ernie 200]
+    end
+
+    it "allows 'min', 'MIN', etc. as aliases for :min" do
+      @redises.zinterstore(@dest, [@bigs, @smalls], :aggregate => 'min')
+      @redises.zscore(@dest, 'bert').should == '1'
+
+      @redises.zinterstore(@dest, [@bigs, @smalls], :aggregate => 'MIN')
+      @redises.zscore(@dest, 'bert').should == '1'
+    end
+
+    it "raises an error for unknown aggregation function" do
+      lambda do
+        @redises.zinterstore(@dest, [@bigs, @smalls], :aggregate => :mix)
+      end.should raise_error(RuntimeError)
+    end
+  end
+end

data/spec/commands/zrange_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+
+describe "#zrange(key, start, stop [, :with_scores => true])" do
+  before do
+    @key = 'mock-redis-test:zrange'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the elements in order by score" do
+    @redises.zrange(@key, 0, 1).should == ['Washington', 'Adams']
+  end
+
+  it "returns the elements in order by score (negative indices)" do
+    @redises.zrange(@key, -2, -1).should == ['Jefferson', 'Madison']
+  end
+
+  it "returns the scores when :with_scores is specified" do
+    @redises.zrange(@key, 0, 1, :with_scores => true).
+      should == ["Washington", "1", "Adams", "2"]
+  end
+
+  it "returns the scores when :withscores is specified" do
+    @redises.zrange(@key, 0, 1, :withscores => true).
+      should == ["Washington", "1", "Adams", "2"]
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrangebyscore_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe "#zrangebyscore(key, start, stop [:with_scores => true] [:limit => [offset count]])" do
+  before do
+    @key = 'mock-redis-test:zrangebyscore'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the elements in order by score" do
+    @redises.zrangebyscore(@key, 1, 2).should == ['Washington', 'Adams']
+  end
+
+  it "returns the scores when :with_scores is specified" do
+    @redises.zrangebyscore(@key, 1, 2, :with_scores => true).
+      should == ["Washington", "1", "Adams", "2"]
+  end
+
+  it "returns the scores when :withscores is specified" do
+    @redises.zrangebyscore(@key, 1, 2, :withscores => true).
+      should == ["Washington", "1", "Adams", "2"]
+  end
+
+  it "honors the :limit => [offset count] argument" do
+    @redises.zrangebyscore(@key, -100, 100, :limit => [1, 2]).
+      should == ["Adams", "Jefferson"]
+  end
+
+  it "raises an error if :limit isn't a 2-tuple" do
+    lambda do
+      @redises.zrangebyscore(@key, -100, 100, :limit => [1, 2, 3])
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.zrangebyscore(@key, -100, 100, :limit => "1, 2")
+    end.should raise_error(RuntimeError)
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrank_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+
+describe "#zrank(key, member)" do
+  before do
+    @key = 'mock-redis-test:zrank'
+
+    @redises.zadd(@key, 1, 'one')
+    @redises.zadd(@key, 2, 'two')
+    @redises.zadd(@key, 3, 'three')
+  end
+
+  it "returns nil if member wasn't present in the set" do
+    @redises.zrank(@key, 'foo').should be_nil
+  end
+
+  it "returns the index of the member in the set" do
+    @redises.zrank(@key, 'one').should == 0
+    @redises.zrank(@key, 'two').should == 1
+    @redises.zrank(@key, 'three').should == 2
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrem_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe "#zrem(key, member)" do
+  before do
+    @key = 'mock-redis-test:zrem'
+
+    @redises.zadd(@key, 1, 'one')
+    @redises.zadd(@key, 2, 'two')
+  end
+
+  it "returns true if member is present in the set" do
+    @redises.zrem(@key, 'one').should be_true
+  end
+
+  it "returns false if member is not present in the set" do
+    @redises.zrem(@key, 'nobody home').should be_false
+  end
+
+  it "removes member from the set" do
+    @redises.zrem(@key, 'one')
+    @redises.zrange(@key, 0, -1).should == ['two']
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zremrangebyrank_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe "#zremrangebyrank(key, start, stop)" do
+  before do
+    @key = 'mock-redis-test:zremrangebyrank'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the number of elements in range" do
+    @redises.zremrangebyrank(@key, 2, 3).should == 2
+  end
+
+  it "removes the elements" do
+    @redises.zremrangebyrank(@key, 2, 3)
+    @redises.zrange(@key, 0, -1).should == %w[Washington Adams]
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zremrangebyscore_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe "#zremrangebyscore(key, min, max)" do
+  before do
+    @key = 'mock-redis-test:zremrangebyscore'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the number of elements in range" do
+    @redises.zremrangebyscore(@key, 2, 3).should == 2
+  end
+
+  it "removes the elements" do
+    @redises.zremrangebyscore(@key, 2, 3)
+    @redises.zrange(@key, 0, -1).should == %w[Washington Madison]
+  end
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrevrange_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+
+describe "#zrevrange(key, start, stop [, :with_scores => true])" do
+  before do
+    @key = 'mock-redis-test:zrevrange'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the elements in order by score" do
+    @redises.zrevrange(@key, 0, 1).should == ['Madison', 'Jefferson']
+  end
+
+  it "returns the elements in order by score (negative indices)" do
+    @redises.zrevrange(@key, -2, -1).should == ['Adams', 'Washington']
+  end
+
+  it "returns the scores when :with_scores is specified" do
+    @redises.zrevrange(@key, 2, 3, :with_scores => true).
+      should == ["Adams", "2", "Washington", "1"]
+  end
+
+  it "returns the scores when :withscores is specified" do
+    @redises.zrevrange(@key, 2, 3, :withscores => true).
+      should == ["Adams", "2", "Washington", "1"]
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrevrangebyscore_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe "#zrevrangebyscore(key, start, stop [:with_scores => true] [:limit => [offset count]])" do
+  before do
+    @key = 'mock-redis-test:zrevrangebyscore'
+    @redises.zadd(@key, 1, 'Washington')
+    @redises.zadd(@key, 2, 'Adams')
+    @redises.zadd(@key, 3, 'Jefferson')
+    @redises.zadd(@key, 4, 'Madison')
+  end
+
+  it "returns the elements in order by score" do
+    @redises.zrevrangebyscore(@key, 4, 3).should == ['Madison', 'Jefferson']
+  end
+
+  it "returns the scores when :with_scores is specified" do
+    @redises.zrevrangebyscore(@key, 4, 3, :with_scores => true).
+      should == %w[Madison 4 Jefferson 3]
+  end
+
+  it "returns the scores when :withscores is specified" do
+    @redises.zrevrangebyscore(@key, 4, 3, :withscores => true).
+      should == %w[Madison 4 Jefferson 3]
+  end
+
+  it "honors the :limit => [offset count] argument" do
+    @redises.zrevrangebyscore(@key, 100, -100, :limit => [1, 2]).
+      should == ["Jefferson", "Adams"]
+  end
+
+  it "raises an error if :limit isn't a 2-tuple" do
+    lambda do
+      @redises.zrevrangebyscore(@key, 100, -100, :limit => [1, 2, 3])
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.zrevrangebyscore(@key, 100, -100, :limit => "1, 2")
+    end.should raise_error(RuntimeError)
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zrevrank_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+
+describe "#zrevrank(key, member)" do
+  before do
+    @key = 'mock-redis-test:zrevrank'
+
+    @redises.zadd(@key, 1, 'one')
+    @redises.zadd(@key, 2, 'two')
+    @redises.zadd(@key, 3, 'three')
+  end
+
+  it "returns nil if member wasn't present in the set" do
+    @redises.zrevrank(@key, 'foo').should be_nil
+  end
+
+  it "returns the index of the member in the set (ordered by -score)" do
+    @redises.zrevrank(@key, 'one').should == 2
+    @redises.zrevrank(@key, 'two').should == 1
+    @redises.zrevrank(@key, 'three').should == 0
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zscore_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+describe "#zscore(key, member)" do
+  before { @key = 'mock-redis-test:zscore' }
+
+  it "returns the score as a string" do
+    @redises.zadd(@key, 0.25, 'foo').should be_true
+    @redises.zscore(@key, 'foo').should == "0.25"
+  end
+
+  it "returns nil if member is not present in the set" do
+    @redises.zscore(@key, 'foo').should be_nil
+  end
+
+  it_should_behave_like "a zset-only command"
+end

data/spec/commands/zunionstore_spec.rb

@@ -0,0 +1,96 @@
+require 'spec_helper'
+
+describe "#zunionstore(destination, keys, [:weights => [w,w,], [:aggregate => :sum|:min|:max])" do
+  before do
+    @set1 = 'mock-redis-test:zunionstore1'
+    @set2 = 'mock-redis-test:zunionstore2'
+    @set3 = 'mock-redis-test:zunionstore3'
+    @dest = 'mock-redis-test:zunionstoredest'
+
+    @redises.zadd(@set1, 1, 'one')
+
+    @redises.zadd(@set2, 1, 'one')
+    @redises.zadd(@set2, 2, 'two')
+
+    @redises.zadd(@set3, 1, 'one')
+    @redises.zadd(@set3, 2, 'two')
+    @redises.zadd(@set3, 3, 'three')
+  end
+
+  it "returns the number of elements in the new set" do
+    @redises.zunionstore(@dest, [@set1, @set2, @set3]).should == 3
+  end
+
+  it "sums the members' scores by default" do
+    @redises.zunionstore(@dest, [@set1, @set2, @set3])
+    @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+      %w[one 3 three 3 two 4]
+  end
+
+  it "raises an error if keys is empty" do
+    lambda do
+      @redises.zunionstore(@dest, [])
+    end.should raise_error(RuntimeError)
+  end
+
+  context "the :weights argument" do
+    it "multiplies the scores by the weights while aggregating" do
+      @redises.zunionstore(@dest, [@set1, @set2, @set3], :weights => [2, 3, 5])
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[one 10 three 15 two 16]
+    end
+
+    it "raises an error if the number of weights != the number of keys" do
+      lambda do
+        @redises.zunionstore(@dest, [@set1, @set2, @set3], :weights => [1,2])
+      end.should raise_error(RuntimeError)
+    end
+  end
+
+  context "the :aggregate argument" do
+    before do
+      @smalls = 'mock-redis-test:zunionstore:smalls'
+      @bigs   = 'mock-redis-test:zunionstore:bigs'
+
+      @redises.zadd(@smalls, 1, 'bert')
+      @redises.zadd(@smalls, 2, 'ernie')
+      @redises.zadd(@bigs, 100, 'bert')
+      @redises.zadd(@bigs, 200, 'ernie')
+    end
+
+    it "aggregates scores with min when :aggregate => :min is specified" do
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => :min)
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[bert 1 ernie 2]
+    end
+
+    it "aggregates scores with max when :aggregate => :max is specified" do
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => :max)
+      @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+        %w[bert 100 ernie 200]
+    end
+
+    it "ignores scores for missing members" do
+      @redises.zadd(@smalls, 3, 'grover')
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => :min)
+      @redises.zscore(@dest, 'grover').should == '3'
+
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => :max)
+      @redises.zscore(@dest, 'grover').should == '3'
+    end
+
+    it "allows 'min', 'MIN', etc. as aliases for :min" do
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => 'min')
+      @redises.zscore(@dest, 'bert').should == '1'
+
+      @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => 'MIN')
+      @redises.zscore(@dest, 'bert').should == '1'
+    end
+
+    it "raises an error for unknown aggregation function" do
+      lambda do
+        @redises.zunionstore(@dest, [@bigs, @smalls], :aggregate => :mix)
+      end.should raise_error(RuntimeError)
+    end
+  end
+end