require 'test_helper' describe "Redis::DistributedStore" do def setup @dmr = Redis::DistributedStore.new [ {:host => "localhost", :port => "6380", :db => 0}, {:host => "localhost", :port => "6381", :db => 0} ] @rabbit = OpenStruct.new :name => "bunny" @white_rabbit = OpenStruct.new :color => "white" @dmr.set "rabbit", @rabbit end def teardown @dmr.ring.nodes.each { |server| server.flushdb } end it "accepts connection params" do dmr = Redis::DistributedStore.new [ :host => "localhost", :port => "6380", :db => "1" ] dmr.ring.nodes.size == 1 mr = dmr.ring.nodes.first mr.to_s.must_equal("Redis Client connected to localhost:6380 against DB 1") end it "forces reconnection" do @dmr.nodes.each do |node| node.expects(:reconnect) end @dmr.reconnect end it "sets an object" do @dmr.set "rabbit", @white_rabbit @dmr.get("rabbit").must_equal(@white_rabbit) end it "gets an object" do @dmr.get("rabbit").must_equal(@rabbit) end it "mget" do @dmr.set "rabbit2", @white_rabbit begin @dmr.mget "rabbit", "rabbit2" do |rabbits| rabbit, rabbit2 = rabbits rabbits.length.must_equal(2) rabbit.must_equal(@rabbit) rabbit2.must_equal(@white_rabbit) end rescue Redis::Distributed::CannotDistribute # Not supported on redis-rb < 4, and hence Ruby < 2.2. end end it "mapped_mget" do @dmr.set "rabbit2", @white_rabbit begin result = @dmr.mapped_mget("rabbit", "rabbit2") result.keys.must_equal %w[ rabbit rabbit2 ] result["rabbit"].must_equal @rabbit result["rabbit2"].must_equal @white_rabbit rescue Redis::Distributed::CannotDistribute # Not supported on redis-rb < 4, and hence Ruby < 2.2. end end it "passes through ring replica options" do dmr = Redis::DistributedStore.new [ {:host => "localhost", :port => "6380", :db => 0}, {:host => "localhost", :port => "6381", :db => 0} ], replicas: 1024 dmr.ring.replicas.must_equal 1024 end it "uses a custom ring object" do my_ring = Redis::HashRing.new dmr = Redis::DistributedStore.new [ {:host => "localhost", :port => "6380", :db => 0}, {:host => "localhost", :port => "6381", :db => 0} ], ring: my_ring dmr.ring.must_equal my_ring dmr.ring.nodes.length.must_equal 2 end describe '#redis_version' do it 'returns redis version' do @dmr.nodes.first.expects(:redis_version) @dmr.redis_version end end describe '#supports_redis_version?' do it 'returns redis version' do @dmr.nodes.first.expects(:supports_redis_version?).with('2.8.0') @dmr.supports_redis_version?('2.8.0') end end describe "namespace" do it "uses namespaced key" do @dmr = Redis::DistributedStore.new [ {:host => "localhost", :port => "6380", :db => 0}, {:host => "localhost", :port => "6381", :db => 0} ], :namespace => "theplaylist" @dmr.expects(:node_for).with("theplaylist:rabbit").returns(@dmr.nodes.first) @dmr.get "rabbit" end end end