redis 2.1.1 → 2.2.0

data/test/connection_handling_test.rb

@@ -0,0 +1,87 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require File.expand_path("./redis_mock", File.dirname(__FILE__))
+
+include RedisMock::Helper
+
+setup do
+  init Redis.new(OPTIONS)
+end
+
+test "AUTH" do
+  replies = {
+    :auth => lambda { |password| $auth = password; "+OK" },
+    :get  => lambda { |key| $auth == "secret" ? "$3\r\nbar" : "$-1" },
+  }
+
+  redis_mock(replies) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380, :password => "secret"))
+
+    assert "bar" == redis.get("foo")
+  end
+end
+
+test "PING" do |r|
+  assert "PONG" == r.ping
+end
+
+test "SELECT" do |r|
+  r.set "foo", "bar"
+
+  r.select 14
+  assert nil == r.get("foo")
+
+  r.client.disconnect
+
+  assert nil == r.get("foo")
+end
+
+test "QUIT" do |r|
+  r.quit
+
+  assert !r.client.connected?
+end
+
+test "SHUTDOWN" do
+  redis_mock(:shutdown => lambda { "+SHUTDOWN" }) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    assert "SHUTDOWN" == redis.shutdown
+  end
+end
+
+test "SLAVEOF" do
+  redis_mock(:slaveof => lambda { |host, port| "+SLAVEOF #{host} #{port}" }) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    assert "SLAVEOF localhost 6381" == redis.slaveof("localhost", 6381)
+  end
+end
+
+test "BGREWRITEAOF" do
+  redis_mock(:bgrewriteaof => lambda { "+BGREWRITEAOF" }) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    assert "BGREWRITEAOF" == redis.bgrewriteaof
+  end
+end
+
+test "CONFIG GET" do |r|
+  assert "300" == r.config(:get, "*")["timeout"]
+
+  assert r.config(:get, "timeout") == { "timeout" => "300" }
+end
+
+test "CONFIG SET" do |r|
+  begin
+    assert "OK" == r.config(:set, "timeout", 200)
+    assert "200" == r.config(:get, "*")["timeout"]
+
+    assert "OK" == r.config(:set, "timeout", 100)
+    assert "100" == r.config(:get, "*")["timeout"]
+  ensure
+    r.config :set, "timeout", 300
+  end
+end
+

data/test/db/.gitignore

@@ -0,0 +1 @@
+redis.pid

data/test/distributed_blocking_commands_test.rb

@@ -0,0 +1,53 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+test "BLPOP" do |r|
+  r.lpush("foo", "s1")
+  r.lpush("foo", "s2")
+
+  wire = Wire.new do
+    redis = Redis::Distributed.new(NODES)
+    Wire.sleep 0.3
+    redis.lpush("foo", "s3")
+  end
+
+  assert ["foo", "s2"] == r.blpop("foo", 1)
+  assert ["foo", "s1"] == r.blpop("foo", 1)
+  assert ["foo", "s3"] == r.blpop("foo", 1)
+
+  wire.join
+end
+
+test "BRPOP" do |r|
+  r.rpush("foo", "s1")
+  r.rpush("foo", "s2")
+
+  wire = Wire.new do
+    redis = Redis::Distributed.new(NODES)
+    Wire.sleep 0.3
+    redis.rpush("foo", "s3")
+  end
+
+  assert ["foo", "s2"] == r.brpop("foo", 1)
+  assert ["foo", "s1"] == r.brpop("foo", 1)
+  assert ["foo", "s3"] == r.brpop("foo", 1)
+
+  wire.join
+end
+
+test "BRPOP should unset a configured socket timeout" do |r|
+  r = Redis::Distributed.new(NODES, :timeout => 1)
+
+  assert_nothing_raised do
+    r.brpop("foo", 2)
+  end # Errno::EAGAIN raised if socket times out before redis command times out
+
+  assert r.nodes.all? { |node| node.client.timeout == 1 }
+end

data/test/distributed_commands_on_hashes_test.rb

@@ -0,0 +1,12 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+load './test/lint/hashes.rb'
+

data/test/distributed_commands_on_lists_test.rb

@@ -0,0 +1,24 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+load './test/lint/lists.rb'
+
+test "RPOPLPUSH" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.rpoplpush("foo", "bar")
+  end
+end
+
+test "BRPOPLPUSH" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.brpoplpush("foo", "bar", 1)
+  end
+end
+

data/test/distributed_commands_on_sets_test.rb

@@ -0,0 +1,85 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+load './test/lint/sets.rb'
+
+test "SMOVE" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "bar", "s2"
+
+    r.smove("foo", "bar", "s1")
+  end
+end
+
+test "SINTER" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+
+    r.sinter("foo", "bar")
+  end
+end
+
+test "SINTERSTORE" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+
+    r.sinterstore("baz", "foo", "bar")
+  end
+end
+
+test "SUNION" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+    r.sadd "bar", "s3"
+
+    r.sunion("foo", "bar")
+  end
+end
+
+test "SUNIONSTORE" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+    r.sadd "bar", "s3"
+
+    r.sunionstore("baz", "foo", "bar")
+  end
+end
+
+test "SDIFF" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+    r.sadd "bar", "s3"
+
+    r.sdiff("foo", "bar")
+  end
+end
+
+test "SDIFFSTORE" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.sadd "foo", "s1"
+    r.sadd "foo", "s2"
+    r.sadd "bar", "s2"
+    r.sadd "bar", "s3"
+
+    r.sdiffstore("baz", "foo", "bar")
+  end
+end
+

data/test/distributed_commands_on_strings_test.rb

@@ -0,0 +1,50 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+load './test/lint/strings.rb'
+
+test "MGET" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.mget("foo", "bar")
+  end
+end
+
+test "MGET mapped" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.mapped_mget("foo", "bar")
+  end
+end
+
+test "MSET" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.mset(:foo, "s1", :bar, "s2")
+  end
+end
+
+test "MSET mapped" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.mapped_mset(:foo => "s1", :bar => "s2")
+  end
+end
+
+test "MSETNX" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.set("foo", "s1")
+    r.msetnx(:foo, "s2", :bar, "s3")
+  end
+end
+
+test "MSETNX mapped" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.set("foo", "s1")
+    r.mapped_msetnx(:foo => "s2", :bar => "s3")
+  end
+end
+

data/test/distributed_commands_on_value_types_test.rb

@@ -0,0 +1,73 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init(Redis::Distributed.new(NODES, :logger => ::Logger.new(log)))
+end
+
+load "./test/lint/value_types.rb"
+
+test "DEL" do |r|
+  r.set "foo", "s1"
+  r.set "bar", "s2"
+  r.set "baz", "s3"
+
+  assert ["bar", "baz", "foo"] == r.keys("*").sort
+
+  assert 1 == r.del("foo")
+
+  assert ["bar", "baz"] == r.keys("*").sort
+
+  assert 2 == r.del("bar", "baz")
+
+  assert [] == r.keys("*").sort
+end
+
+test "RANDOMKEY" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.randomkey
+  end
+end
+
+test "RENAME" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.set("foo", "s1")
+    r.rename "foo", "bar"
+  end
+
+  assert "s1" == r.get("foo")
+  assert nil == r.get("bar")
+end
+
+test "RENAMENX" do |r|
+  assert_raise Redis::Distributed::CannotDistribute do
+    r.set("foo", "s1")
+    r.rename "foo", "bar"
+  end
+
+  assert "s1" == r.get("foo")
+  assert nil  == r.get("bar")
+end
+
+test "DBSIZE" do |r|
+  assert [0] == r.dbsize
+
+  r.set("foo", "s1")
+
+  assert [1] == r.dbsize
+end
+
+test "FLUSHDB" do |r|
+  r.set("foo", "s1")
+  r.set("bar", "s2")
+
+  assert [2] == r.dbsize
+
+  r.flushdb
+
+  assert [0] == r.dbsize
+end
+

data/test/distributed_commands_requiring_clustering_test.rb

@@ -0,0 +1,148 @@
+# encoding: UTF-8
+
+require File.expand_path("./helper", File.dirname(__FILE__))
+require "redis/distributed"
+
+setup do
+  log = StringIO.new
+  init Redis::Distributed.new(NODES, :logger => ::Logger.new(log))
+end
+
+test "RENAME" do |r|
+  r.set("{qux}foo", "s1")
+  r.rename "{qux}foo", "{qux}bar"
+
+  assert "s1" == r.get("{qux}bar")
+  assert nil == r.get("{qux}foo")
+end
+
+test "RENAMENX" do |r|
+  r.set("{qux}foo", "s1")
+  r.set("{qux}bar", "s2")
+
+  assert false == r.renamenx("{qux}foo", "{qux}bar")
+
+  assert "s1" == r.get("{qux}foo")
+  assert "s2" == r.get("{qux}bar")
+end
+
+test "BRPOPLPUSH" do |r|
+  r.rpush "{qux}foo", "s1"
+  r.rpush "{qux}foo", "s2"
+
+  assert_equal "s2", r.brpoplpush("{qux}foo", "{qux}bar", 1)
+  assert_equal ["s2"], r.lrange("{qux}bar", 0, -1)
+end
+
+test "RPOPLPUSH" do |r|
+  r.rpush "{qux}foo", "s1"
+  r.rpush "{qux}foo", "s2"
+
+  assert "s2" == r.rpoplpush("{qux}foo", "{qux}bar")
+  assert ["s2"] == r.lrange("{qux}bar", 0, -1)
+  assert "s1" == r.rpoplpush("{qux}foo", "{qux}bar")
+  assert ["s1", "s2"] == r.lrange("{qux}bar", 0, -1)
+end
+
+test "SMOVE" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}bar", "s2"
+
+  assert r.smove("{qux}foo", "{qux}bar", "s1")
+  assert r.sismember("{qux}bar", "s1")
+end
+
+test "SINTER" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+
+  assert ["s2"] == r.sinter("{qux}foo", "{qux}bar")
+end
+
+test "SINTERSTORE" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+
+  r.sinterstore("{qux}baz", "{qux}foo", "{qux}bar")
+
+  assert ["s2"] == r.smembers("{qux}baz")
+end
+
+test "SUNION" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+  r.sadd "{qux}bar", "s3"
+
+  assert ["s1", "s2", "s3"] == r.sunion("{qux}foo", "{qux}bar").sort
+end
+
+test "SUNIONSTORE" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+  r.sadd "{qux}bar", "s3"
+
+  r.sunionstore("{qux}baz", "{qux}foo", "{qux}bar")
+
+  assert ["s1", "s2", "s3"] == r.smembers("{qux}baz").sort
+end
+
+test "SDIFF" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+  r.sadd "{qux}bar", "s3"
+
+  assert ["s1"] == r.sdiff("{qux}foo", "{qux}bar")
+  assert ["s3"] == r.sdiff("{qux}bar", "{qux}foo")
+end
+
+test "SDIFFSTORE" do |r|
+  r.sadd "{qux}foo", "s1"
+  r.sadd "{qux}foo", "s2"
+  r.sadd "{qux}bar", "s2"
+  r.sadd "{qux}bar", "s3"
+
+  r.sdiffstore("{qux}baz", "{qux}foo", "{qux}bar")
+
+  assert ["s1"] == r.smembers("{qux}baz")
+end
+
+test "SORT" do |r|
+  r.set("{qux}foo:1", "s1")
+  r.set("{qux}foo:2", "s2")
+
+  r.rpush("{qux}bar", "1")
+  r.rpush("{qux}bar", "2")
+
+  assert ["s1"] == r.sort("{qux}bar", :get => "{qux}foo:*", :limit => [0, 1])
+  assert ["s2"] == r.sort("{qux}bar", :get => "{qux}foo:*", :limit => [0, 1], :order => "desc alpha")
+end
+
+test "SORT with an array of GETs" do |r|
+  r.set("{qux}foo:1:a", "s1a")
+  r.set("{qux}foo:1:b", "s1b")
+
+  r.set("{qux}foo:2:a", "s2a")
+  r.set("{qux}foo:2:b", "s2b")
+
+  r.rpush("{qux}bar", "1")
+  r.rpush("{qux}bar", "2")
+
+  assert ["s1a", "s1b"] == r.sort("{qux}bar", :get => ["{qux}foo:*:a", "{qux}foo:*:b"], :limit => [0, 1])
+  assert ["s2a", "s2b"] == r.sort("{qux}bar", :get => ["{qux}foo:*:a", "{qux}foo:*:b"], :limit => [0, 1], :order => "desc alpha")
+end
+
+test "SORT with STORE" do |r|
+  r.set("{qux}foo:1", "s1")
+  r.set("{qux}foo:2", "s2")
+
+  r.rpush("{qux}bar", "1")
+  r.rpush("{qux}bar", "2")
+
+  r.sort("{qux}bar", :get => "{qux}foo:*", :store => "{qux}baz")
+  assert ["s1", "s2"] == r.lrange("{qux}baz", 0, -1)
+end