mock_redis 0.0.1

data/spec/commands/sdiff_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+
+describe '#sdiff(key [, key, ...])' do
+  before do
+    @numbers = 'mock-redis-test:sdiff:numbers'
+    @evens   = 'mock-redis-test:sdiff:odds'
+    @primes  = 'mock-redis-test:sdiff:primes'
+
+    (1..10).each {|i| @redises.sadd(@numbers, 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 first set minus the second set" do
+    @redises.sdiff(@numbers, @evens).should == %w[1 3 5 7 9]
+  end
+
+  it "returns the first set minus all the other sets" do
+    @redises.sdiff(@numbers, @evens, @primes).should == %w[1 9]
+  end
+
+  it "treats missing keys as empty sets" do
+    @redises.sdiff(@evens, 'mock-redis-test:nonesuch').should == %w[10 2 4 6 8]
+  end
+
+  it "returns the first set when called with a single argument" do
+    @redises.sdiff(@primes).should == %w[2 3 5 7]
+  end
+
+  it "raises an error if given 0 arguments" do
+    lambda do
+      @redises.sdiff()
+    end.should raise_error(RuntimeError)
+  end
+
+  it "raises an error if any argument is not a a set" do
+    @redises.set('foo', 1)
+
+    lambda do
+      @redises.sdiff(@numbers, 'foo')
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.sdiff('foo', @numbers)
+    end.should raise_error(RuntimeError)
+  end
+end

data/spec/commands/sdiffstore_spec.rb

@@ -0,0 +1,51 @@
+require 'spec_helper'
+
+describe '#sdiffstore(destination, key [, key, ...])' do
+  before do
+    @numbers     = 'mock-redis-test:sdiffstore:numbers'
+    @evens       = 'mock-redis-test:sdiffstore:odds'
+    @primes      = 'mock-redis-test:sdiffstore:primes'
+    @destination = 'mock-redis-test:sdiffstore:destination'
+
+    (1..10).each {|i| @redises.sadd(@numbers, 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
+    @redises.sdiffstore(@destination, @numbers, @evens).should == 5
+  end
+
+  it "stores the resulting set" do
+    @redises.sdiffstore(@destination, @numbers, @evens)
+    @redises.smembers(@destination).should == %w[1 3 5 7 9]
+  end
+
+  it "does not store empty sets" do
+    @redises.sdiffstore(@destination, @numbers, @numbers).should == 0
+    @redises.get(@destination).should be_nil
+  end
+
+  it "treats missing keys as empty sets" do
+    @redises.sdiffstore(@destination, @evens, 'mock-redis-test:nonesuch')
+    @redises.smembers(@destination).should == %w[10 2 4 6 8]
+  end
+
+  it "raises an error if given 0 sets" do
+    lambda do
+      @redises.sdiffstore(@destination)
+    end.should raise_error(RuntimeError)
+  end
+
+  it "raises an error if any argument is not a a set" do
+    @redises.set('mock-redis-test:notset', 1)
+
+    lambda do
+      @redises.sdiffstore(@destination, @numbers, 'mock-redis-test:notset')
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.sdiffstore(@destination, 'mock-redis-test:notset', @numbers)
+    end.should raise_error(RuntimeError)
+  end
+end

data/spec/commands/select_spec.rb

@@ -0,0 +1,53 @@
+require 'spec_helper'
+
+describe "#select(db)" do
+  before { @key = 'mock-redis-test:select' }
+
+  it "returns 'OK'" do
+    @redises.select(0).should == 'OK'
+  end
+
+  it "treats '0' and 0 the same" do
+    @redises.select('0')
+    @redises.set(@key, 'foo')
+    @redises.select(0)
+    @redises.get(@key).should == 'foo'
+  end
+
+  it "switches databases" do
+    @redises.select(0)
+    @redises.set(@key, 'foo')
+
+    @redises.select(1)
+    @redises.get(@key).should be_nil
+
+    @redises.select(0)
+    @redises.get(@key).should == 'foo'
+  end
+
+  context "[mock only]" do
+    # Time dependence introduces a bit of nondeterminism here
+    before do
+      @now = Time.now
+      Time.stub!(:now).and_return(@now)
+
+      @mock = @redises.mock
+
+      @mock.select(0)
+      @mock.set(@key, 1)
+      @mock.expire(@key, 101)
+
+      @mock.select(1)
+      @mock.set(@key, 2)
+      @mock.expire(@key, 201)
+    end
+
+    it "keeps expire times per-db" do
+      @mock.select(0)
+      @mock.ttl(@key).should == 100
+
+      @mock.select(1)
+      @mock.ttl(@key).should == 200
+    end
+  end
+end

data/spec/commands/set_spec.rb

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

data/spec/commands/setbit_spec.rb

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "#setbit(key, offset)" do
+  before do
+    @key = 'mock-redis-test:setbit'
+    @redises.set(@key, 'h')    # ASCII 0x68
+  end
+
+  it "returns the original stored bit's value" do
+    @redises.setbit(@key, 0, 1).should == 0
+    @redises.setbit(@key, 1, 1).should == 1
+  end
+
+  it "sets the bit within the string" do
+    $debug = true
+    @redises.setbit(@key, 7, 1)
+    $debug = false
+    @redises.get(@key).should == 'i'  # ASCII 0x69
+  end
+
+  it "does the right thing with multibyte characters" do
+    @redises.set(@key, "€99.94")   # the euro sign is 3 bytes wide in UTF-8
+    @redises.setbit(@key, 63, 1).should == 0
+    @redises.get(@key).should == "€99.95"
+  end
+
+  it "expands the string if necessary" do
+    @redises.setbit(@key, 9, 1)
+    @redises.get(@key).should == "h@"
+  end
+
+  it "sets added bits to 0" do
+    @redises.setbit(@key, 17, 1)
+    @redises.get(@key).should == "h\000@"
+  end
+
+  it "treats missing keys as empty strings" do
+    @redises.del(@key)
+    @redises.setbit(@key, 1, 1)
+    @redises.get(@key).should == "@"
+  end
+
+  it_should_behave_like "a string-only command"
+end
+ 

data/spec/commands/setex_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe '#setex(key, seconds, value)' do
+  before { @key = 'mock-redis-test:setex' }
+
+  it "responds with 'OK'" do
+    @redises.setex(@key, 10, 'value').should == 'OK'
+  end
+
+  it "sets the value" do
+    @redises.setex(@key, 10000, 'value')
+    @redises.get(@key).should == 'value'
+  end
+
+  it "sets the expiration time" do
+    @redises.setex(@key, 10000, 'value')
+
+    # no guarantee these are the same
+    @redises.real.ttl(@key).should > 0
+    @redises.mock.ttl(@key).should > 0
+  end
+end

data/spec/commands/setnx_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe '#setnx(key, value)' do
+  before { @key = 'mock-redis-test:setnx' }
+
+  it "returns true if the key was absent" do
+    @redises.setnx(@key, 1).should be_true
+  end
+
+  it "returns false if the key was present" do
+    @redises.set(@key, 2)
+    @redises.setnx(@key, 1).should be_false
+  end
+
+  it "sets the value if missing" do
+    @redises.setnx(@key, 'value')
+    @redises.get(@key).should == 'value'
+  end
+
+  it "does nothing if the value is present" do
+    @redises.set(@key, 'old')
+    @redises.setnx(@key, 'new')
+    @redises.get(@key).should == 'old'
+  end
+end

data/spec/commands/setrange_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+
+describe "#setrange(key, offset, value)" do
+  before do
+    @key = 'mock-redis-test:setrange'
+    @redises.set(@key, "This is a string")
+  end
+
+  it "returns the string's new length" do
+    @redises.setrange(@key, 0, "That").should == 16
+  end
+
+  it "updates part of the string" do
+    @redises.setrange(@key, 0, "That")
+    @redises.get(@key).should == "That is a string"
+  end
+
+  it "zero-pads the string if necessary" do
+    @redises.setrange(@key, 20, "X")
+    @redises.get(@key).should == "This is a string\000\000\000\000X"
+  end
+
+  it "stores things as strings" do
+    other_key = "mock-redis-test:setrange-other-key"
+    @redises.setrange(other_key, 0, 1)
+    @redises.get(other_key).should == "1"
+  end
+
+  it_should_behave_like "a string-only command"
+end

data/spec/commands/sinter_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+describe '#sinter(key [, key, ...])' do
+  before do
+    @numbers = 'mock-redis-test:sinter:numbers'
+    @evens   = 'mock-redis-test:sinter:evens'
+    @primes  = 'mock-redis-test:sinter:primes'
+
+    (1..10).each {|i| @redises.sadd(@numbers, 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
+    @redises.sinter(@evens, @primes).should == ["2"]
+  end
+
+  it "treats missing keys as empty sets" do
+    @redises.sinter(@destination, 'mock-redis-test:nonesuch').should == []
+  end
+
+  it "raises an error if given 0 sets" do
+    lambda do
+      @redises.sinter
+    end.should raise_error(RuntimeError)
+  end
+
+  it "raises an error if any argument is not a a set" do
+    @redises.set('mock-redis-test:notset', 1)
+
+    lambda do
+      @redises.sinter(@numbers, 'mock-redis-test:notset')
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.sinter('mock-redis-test:notset', @numbers)
+    end.should raise_error(RuntimeError)
+  end
+end
+
+    

data/spec/commands/sinterstore_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+describe '#sinterstore(destination, key [, key, ...])' do
+  before do
+    @numbers     = 'mock-redis-test:sinterstore:numbers'
+    @evens       = 'mock-redis-test:sinterstore:evens'
+    @primes      = 'mock-redis-test:sinterstore:primes'
+    @destination = 'mock-redis-test:sinterstore:destination'
+
+    (1..10).each {|i| @redises.sadd(@numbers, 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
+    @redises.sinterstore(@destination, @numbers, @evens).should == 5
+  end
+
+  it "stores the resulting set" do
+    @redises.sinterstore(@destination, @numbers, @evens)
+    @redises.smembers(@destination).should == %w[10 2 4 6 8]
+  end
+
+  it "does not store empty sets" do
+    @redises.sinterstore(@destination, 'mock-redis-test:nonesuch', @numbers).should == 0
+    @redises.get(@destination).should be_nil
+  end
+
+  it "raises an error if given 0 sets" do
+    lambda do
+      @redises.sinterstore(@destination)
+    end.should raise_error(RuntimeError)
+  end
+
+  it "raises an error if any argument is not a a set" do
+    @redises.set('mock-redis-test:notset', 1)
+
+    lambda do
+      @redises.sinterstore(@destination, @numbers, 'mock-redis-test:notset')
+    end.should raise_error(RuntimeError)
+
+    lambda do
+      @redises.sinterstore(@destination, 'mock-redis-test:notset', @numbers)
+    end.should raise_error(RuntimeError)
+  end
+end

data/spec/commands/sismember_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe '#sismember(key, member)' do
+  before do
+    @key = 'mock-redis-test:sismember'
+    @redises.sadd(@key, 'whiskey')
+    @redises.sadd(@key, 'beer')
+  end
+
+  it "returns true if member is in set" do
+    @redises.sismember(@key, 'whiskey').should be_true
+    @redises.sismember(@key, 'beer').should be_true
+  end
+
+  it "returns false if member is not in set" do
+    @redises.sismember(@key, 'cola').should be_false
+  end
+
+  it "stringifies member" do
+    @redises.sadd(@key, '1')
+    @redises.sismember(@key, 1).should be_true
+  end
+
+  it "treats a nonexistent value as an empty set" do
+    @redises.sismember('mock-redis-test:nonesuch', 'beer').should be_false
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/smembers_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+describe '#smembers(key)' do
+  before { @key = 'mock-redis-test:smembers' }
+
+  it "returns [] for an empty set" do
+    @redises.smembers(@key).should == []
+  end
+
+  it "returns the set's members" do
+    @redises.sadd(@key, 1)
+    @redises.sadd(@key, 2)
+    @redises.sadd(@key, 3)
+    @redises.smembers(@key).should == %w[1 2 3]
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/smove_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+describe '#smove(source, destination, member)' do
+  before do
+    @src  = 'mock-redis-test:smove-source'
+    @dest = 'mock-redis-test:smove-destination'
+
+    @redises.sadd(@src, 1)
+    @redises.sadd(@dest, 2)
+  end
+
+  it "returns true if the member exists in src" do
+    @redises.smove(@src, @dest, 1).should be_true
+  end
+
+  it "returns false if the member exists in src" do
+    @redises.smove(@src, @dest, 'nope').should be_false
+  end
+
+  it "returns true if the member exists in src and dest" do
+    @redises.sadd(@dest, 1)
+    @redises.smove(@src, @dest, 1).should be_true
+  end
+
+  it "moves member from source to destination" do
+    @redises.smove(@src, @dest, 1)
+    @redises.sismember(@dest, 1).should be_true
+    @redises.sismember(@src, 1).should be_false
+  end
+
+  it "cleans up empty sets" do
+    @redises.smove(@src, @dest, 1)
+    @redises.get(@src).should be_nil
+  end
+
+  it "treats a nonexistent value as an empty set" do
+    @redises.smove('mock-redis-test:nonesuch', @dest, 1).should be_false
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/spop_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe '#spop(key)' do
+  before do
+    @key = 'mock-redis-test:spop'
+
+    @redises.sadd(@key, 'value')
+  end
+
+  it "returns a member of the set" do
+    @redises.spop(@key).should == 'value'
+  end
+
+  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
+    @redises.spop(@key)
+    @redises.spop(@key).should be_nil
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/srandmember_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe '#srandmember(key)' do
+  before do
+    @key = 'mock-redis-test:srandmember'
+
+    @redises.sadd(@key, 'value')
+  end
+
+  it "returns a member of the set" do
+    @redises.srandmember(@key).should == 'value'
+  end
+
+  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
+    @redises.spop(@key)
+    @redises.srandmember(@key).should be_nil
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/srem_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+describe '#srem(key, member)' do
+  before do
+    @key = 'mock-redis-test:srem'
+
+    @redises.sadd(@key, 'bert')
+    @redises.sadd(@key, 'ernie')
+  end
+
+  it "returns true if member is in the set" do
+    @redises.srem(@key, 'bert').should be_true
+  end
+
+  it "returns false if member is not in the set" do
+    @redises.srem(@key, 'cookiemonster').should be_false
+  end
+
+  it "removes member from the set" do
+    @redises.srem(@key, 'ernie')
+    @redises.smembers(@key).should == ['bert']
+  end
+
+  it "stringifies member" do
+    @redises.sadd(@key, '1')
+    @redises.srem(@key, 1).should be_true
+  end
+
+  it "cleans up empty sets" do
+    @redises.smembers(@key).each {|m| @redises.srem(@key, m)}
+    @redises.get(@key).should be_nil
+  end
+
+  it_should_behave_like "a set-only command"
+end

data/spec/commands/strlen_spec.rb

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe "#strlen(key)" do
+  before do
+    @key = 'mock-redis-test:73288'
+    @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
+    @redises.strlen('mock-redis-test:does-not-exist').should == 0
+  end
+
+  it_should_behave_like "a string-only command"
+end

data/spec/commands/sunion_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+describe '#sunion(key [, key, ...])' do
+  before 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) }
+  end
+
+  it "returns the elements in the resulting set" do
+    @redises.sunion(@evens, @primes).should == %w[10 2 3 4 5 6 7 8]
+  end
+
+  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
+    lambda do
+      @redises.sunion
+    end.should raise_error(RuntimeError)
+  end
+
+  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)
+
+    lambda do
+      @redises.sunion('mock-redis-test:notset', @numbers)
+    end.should raise_error(RuntimeError)
+  end
+end
+
+    

data/spec/commands/sunionstore_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+describe '#sunionstore(destination, key [, key, ...])' do
+  before do
+    @evens       = 'mock-redis-test:sunionstore:evens'
+    @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) }
+  end
+
+  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
+    @redises.sunionstore(@destination, @primes, @evens)
+    @redises.smembers(@destination).should == %w[10 2 3 4 5 6 7 8]
+  end
+
+  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 "raises an error if given 0 sets" do
+    lambda do
+      @redises.sunionstore(@destination)
+    end.should raise_error(RuntimeError)
+  end
+
+  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)
+
+    lambda do
+      @redises.sunionstore(@destination, 'mock-redis-test:notset', @primes)
+    end.should raise_error(RuntimeError)
+  end
+end