ryansch-mock_redis 0.2.0.1

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,96 @@
+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 "removes existing elements in destination" do
+    @redises.zadd(@dest, 10, 'ten')
+
+    @redises.zinterstore(@dest, [@primes])
+    @redises.zrange(@dest, 0, -1, :with_scores => true).should ==
+      %w[two 2 three 3 five 5 seven 7]
+  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,68 @@
+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 "treats scores like floats, not strings" do
+    @redises.zadd(@key, "10", "Tyler")
+    @redises.zrangebyscore(@key, 1, 2).should == ['Washington', 'Adams']
+  end
+
+  it "treats -inf as negative infinity" do
+    @redises.zrangebyscore(@key, "-inf", 3).should ==
+      ["Washington", "Adams", "Jefferson"]
+  end
+
+  it "treats +inf as positive infinity" do
+    @redises.zrangebyscore(@key, 3, "+inf").should == ["Jefferson", "Madison"]
+  end
+
+  it "treats +inf as positive infinity" do
+    @redises.zrangebyscore(@key, 3, "+inf").should == ["Jefferson", "Madison"]
+  end
+
+  it "honors exclusive ranges on the left" do
+    @redises.zrangebyscore(@key, "(3", 4).should == ["Madison"]
+  end
+
+  it "honors exclusive ranges on the right" do
+    @redises.zrangebyscore(@key, "3", "(4").should == ["Jefferson"]
+  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,28 @@
+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
+
+  # As seen in http://redis.io/commands/zremrangebyscore
+  it "removes the elements for complex statements" do
+    @redises.zremrangebyscore(@key, '-inf', '(4')
+    @redises.zrange(@key, 0, -1).should == %w[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,47 @@
+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 "treats +inf as positive infinity" do
+    @redises.zrevrangebyscore(@key, "+inf", 3).
+      should == %w[Madison Jefferson]
+  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