gorsuch-redis 3.0.0.rc1

data/test/connection_handling_test.rb

@@ -0,0 +1,204 @@
+# 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
+  commands = {
+    :shutdown => lambda { :exit }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    # SHUTDOWN does not reply: test that it does not raise here.
+    assert nil == redis.shutdown
+  end
+end
+
+test "SHUTDOWN with error" do
+  connections = 0
+  commands = {
+    :select => lambda { |*_| connections += 1; "+OK\r\n" },
+    :connections => lambda { ":#{connections}\r\n" },
+    :shutdown => lambda { "-ERR could not shutdown\r\n" }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    connections = redis.connections
+
+    # SHUTDOWN replies with an error: test that it gets raised
+    assert_raise Redis::CommandError do
+      redis.shutdown
+    end
+
+    # The connection should remain in tact
+    assert connections == redis.connections
+  end
+end
+
+test "SHUTDOWN from pipeline" do
+  commands = {
+    :shutdown => lambda { :exit }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    result = redis.pipelined do
+      redis.shutdown
+    end
+
+    assert nil == result
+    assert !redis.client.connected?
+  end
+end
+
+test "SHUTDOWN with error from pipeline" do
+  connections = 0
+  commands = {
+    :select => lambda { |*_| connections += 1; "+OK\r\n" },
+    :connections => lambda { ":#{connections}\r\n" },
+    :shutdown => lambda { "-ERR could not shutdown\r\n" }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    connections = redis.connections
+
+    # SHUTDOWN replies with an error: test that it gets raised
+    assert_raise Redis::CommandError do
+      redis.pipelined do
+        redis.shutdown
+      end
+    end
+
+    # The connection should remain in tact
+    assert connections == redis.connections
+  end
+end
+
+test "SHUTDOWN from MULTI/EXEC" do
+  commands = {
+    :multi => lambda { "+OK\r\n" },
+    :shutdown => lambda { "+QUEUED\r\n" },
+    :exec => lambda { :exit }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    result = redis.multi do
+      redis.shutdown
+    end
+
+    assert nil == result
+    assert !redis.client.connected?
+  end
+end
+
+test "SHUTDOWN with error from MULTI/EXEC" do
+  connections = 0
+  commands = {
+    :select => lambda { |*_| connections += 1; "+OK\r\n" },
+    :connections => lambda { ":#{connections}\r\n" },
+    :multi => lambda { "+OK\r\n" },
+    :shutdown => lambda { "+QUEUED\r\n" },
+    :exec => lambda { "*1\r\n-ERR could not shutdown\r\n" }
+  }
+
+  redis_mock(commands) do
+    redis = Redis.new(OPTIONS.merge(:port => 6380))
+
+    connections = redis.connections
+
+    # SHUTDOWN replies with an error: test that it gets returned
+    # We should test for Redis::CommandError here, but hiredis doesn't yet do
+    # custom error classes.
+    assert_raise(StandardError) do
+      redis.multi { redis.shutdown }
+    end
+
+    # The connection should remain intact
+    assert connections == redis.connections
+  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 r.config(:get, "*")["timeout"] != nil
+
+  config = r.config(:get, "timeout")
+  assert ["timeout"] == config.keys
+  assert config.values.compact.size > 0
+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,11 @@
+# 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,23 @@
+# 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,84 @@
+# 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_sorted_sets_test.rb

@@ -0,0 +1,19 @@
+# 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/sorted_sets.rb'
+
+test "ZCOUNT" do |r|
+  r.zadd "foo", 1, "s1"
+  r.zadd "foo", 2, "s2"
+  r.zadd "foo", 3, "s3"
+
+  assert 2 == r.zcount("foo", 2, 3)
+end

data/test/distributed_commands_on_strings_test.rb

@@ -0,0 +1,49 @@
+# 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