boourns-memcached 1.4.6

data/test/unit/memcached_test.rb

@@ -0,0 +1,1340 @@
+
+require File.expand_path("#{File.dirname(__FILE__)}/../test_helper")
+
+class MemcachedTest < Test::Unit::TestCase
+
+  def setup
+    @servers = ['localhost:43042', 'localhost:43043', "#{UNIX_SOCKET_NAME}0"]
+    @udp_servers = ['localhost:43052', 'localhost:43053']
+
+    # Maximum allowed prefix key size for :hash_with_prefix_key_key => false
+    @prefix_key = 'prefix_key_'
+
+    @options = {
+      :prefix_key => @prefix_key,
+      :hash => :default,
+      :distribution => :modula,
+      :show_backtraces => true}
+    @cache = Memcached.new(@servers, @options)
+
+    @binary_protocol_options = {
+      :prefix_key => @prefix_key,
+      :hash => :default,
+      :distribution => :modula,
+      :binary_protocol => true,
+      :show_backtraces => true}
+    @binary_protocol_cache = Memcached.new(@servers, @binary_protocol_options)
+
+    @udp_options = {
+      :prefix_key => @prefix_key,
+      :hash => :default,
+      :use_udp => true,
+      :distribution => :modula,
+      :show_backtraces => true}
+    @udp_cache = Memcached.new(@udp_servers, @udp_options)
+
+    @noblock_options = {
+      :prefix_key => @prefix_key,
+      :no_block => true,
+      :noreply => true,
+      :buffer_requests => true,
+      :hash => :default,
+      :distribution => :modula,
+      :show_backtraces => true}
+    @noblock_cache = Memcached.new(@servers, @noblock_options)
+
+    @value = OpenStruct.new(:a => 1, :b => 2, :c => GenericClass)
+    @marshalled_value = Marshal.dump(@value)
+  end
+
+  # Initialize
+
+  def test_initialize
+    cache = Memcached.new @servers, :prefix_key => 'test'
+    assert_equal 'test', cache.prefix_key
+    assert_equal 3, cache.send(:server_structs).size
+    assert_equal 'localhost', cache.send(:server_structs).first.hostname
+    assert_equal 43042, cache.send(:server_structs).first.port
+  end
+
+  def test_initialize_with_ip_addresses
+    cache = Memcached.new ['127.0.0.1:43042', '127.0.0.1:43043']
+    assert_equal '127.0.0.1', cache.send(:server_structs).first.hostname
+    assert_equal '127.0.0.1', cache.send(:server_structs).last.hostname
+  end
+
+  def test_initialize_without_port
+    cache = Memcached.new ['localhost']
+    assert_equal 'localhost', cache.send(:server_structs).first.hostname
+    assert_equal 11211, cache.send(:server_structs).first.port
+  end
+
+  def test_initialize_with_ports_and_weights
+    cache = Memcached.new ['localhost:43042:2', 'localhost:43043:10']
+    assert_equal 2, cache.send(:server_structs).first.weight
+    assert_equal 43043, cache.send(:server_structs).last.port
+    assert_equal 10, cache.send(:server_structs).last.weight
+  end
+
+  def test_initialize_with_hostname_only
+    addresses = (1..8).map { |i| "app-cache-%02d" % i }
+    cache = Memcached.new(addresses)
+    addresses.each_with_index do |address, index|
+      assert_equal address, cache.send(:server_structs)[index].hostname
+      assert_equal 11211, cache.send(:server_structs)[index].port
+    end
+  end
+
+  def test_initialize_with_ip_address_and_options
+    cache = Memcached.new '127.0.0.1:43042', :ketama_weighted => false
+    assert_equal '127.0.0.1', cache.send(:server_structs).first.hostname
+    assert_equal false, cache.options[:ketama_weighted]
+  end
+
+  def test_options_are_set
+    Memcached::DEFAULTS.merge(@noblock_options).each do |key, expected|
+      value = @noblock_cache.options[key]
+      unless key == :rcv_timeout or key == :poll_timeout or key == :prefix_key or key == :ketama_weighted
+        assert(expected == value, "#{key} should be #{expected} but was #{value}")
+      end
+    end
+  end
+
+  def test_options_are_frozen
+    assert_raise(TypeError, RuntimeError) do
+      @cache.options[:no_block] = true
+    end
+  end
+
+  def test_behaviors_are_set
+    Memcached::BEHAVIORS.keys.each do |key, value|
+      assert_not_nil @cache.send(:get_behavior, key)
+    end
+  end
+
+  def test_initialize_with_invalid_server_strings
+    assert_raise(ArgumentError) { Memcached.new ":43042" }
+    assert_raise(ArgumentError) { Memcached.new "localhost:memcached" }
+    assert_raise(ArgumentError) { Memcached.new "local host:43043:1" }
+  end
+
+  def test_initialize_with_invalid_options
+    assert_raise(ArgumentError) do
+      Memcached.new @servers, :sort_hosts => true, :distribution => :consistent
+    end
+  end
+
+  def test_initialize_with_invalid_prefix_key
+    assert_raise(ArgumentError) do
+      Memcached.new @servers, :prefix_key => "x" * 128
+    end
+  end
+
+  def test_initialize_without_prefix_key
+    cache = Memcached.new @servers
+    assert_equal 3, cache.send(:server_structs).size
+  end
+
+  def test_set_prefix_key
+    cache = Memcached.new @servers, :prefix_key => "foo"
+    cache.set_prefix_key("bar")
+    assert_equal "bar", cache.prefix_key
+  end
+
+  def test_set_prefix_key_to_empty_string
+    cache = Memcached.new @servers, :prefix_key => "foo"
+    cache.set_prefix_key("")
+    assert_equal "", cache.prefix_key
+  end
+
+  def test_memcached_callback_set_with_empty_string_should_not_raise_exception
+    cache = Memcached.new @servers, :prefix_key => "foo"
+    assert_nothing_raised do
+      cache.set_prefix_key("")
+    end
+  end
+
+  def test_initialize_negative_behavior
+    cache = Memcached.new @servers,
+      :buffer_requests => false
+    assert_nothing_raised do
+      cache.set key, @value
+    end
+  end
+
+  def test_initialize_without_backtraces
+    cache = Memcached.new @servers,
+      :show_backtraces => false
+    cache.delete key rescue
+    begin
+      cache.get key
+    rescue Memcached::NotFound => e
+      assert e.backtrace.empty?
+    end
+    begin
+      cache.append key, @value
+    rescue Memcached::NotStored => e
+      assert e.backtrace.empty?
+    end
+  end
+
+  def test_initialize_with_backtraces
+    cache = Memcached.new @servers, :show_backtraces => true
+    cache.delete key rescue nil
+    begin
+      cache.get key
+    rescue Memcached::NotFound => e
+      assert !e.backtrace.empty?
+    end
+  end
+
+  def test_initialize_sort_hosts
+    # Original
+    cache = Memcached.new(@servers.sort,
+      :sort_hosts => false,
+      :distribution => :modula
+    )
+    assert_equal @servers.sort,
+      cache.servers
+
+    # Original with sort_hosts
+    cache = Memcached.new(@servers.sort,
+      :sort_hosts => true,
+      :distribution => :modula
+    )
+    assert_equal @servers.sort,
+      cache.servers
+
+    # Reversed
+    cache = Memcached.new(@servers.sort.reverse,
+      :sort_hosts => false,
+      :distribution => :modula
+    )
+      assert_equal @servers.sort.reverse,
+    cache.servers
+
+    # Reversed with sort_hosts
+    cache = Memcached.new(@servers.sort.reverse,
+      :sort_hosts => true,
+      :distribution => :modula
+    )
+    assert_equal @servers.sort,
+      cache.servers
+  end
+
+  def test_initialize_single_server
+    cache = Memcached.new 'localhost:43042'
+    assert_equal 1, cache.send(:server_structs).size
+  end
+
+  def test_initialize_strange_argument
+    assert_raise(ArgumentError) { Memcached.new 1 }
+  end
+
+  # Get
+
+  def test_get
+    @cache.set key, @value
+    assert_equal @value, @cache.get(key)
+    assert_equal @value, @binary_protocol_cache.get(key)
+
+    @udp_cache.set(key, @value)
+    assert_raises(Memcached::ActionNotSupported) do
+      @udp_cache.get(key)
+    end
+  end
+
+  def test_get_nil
+    @cache.set key, nil, 0
+    result = @cache.get key
+    assert_equal nil, result
+  end
+
+  def test_get_from_last
+    cache = Memcached.new(@servers, :distribution => :random)
+    10.times { |n| cache.set key, n }
+    10.times do
+      assert_equal cache.get(key), cache.get_from_last(key)
+    end
+  end
+
+  def test_get_missing
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotFound) do
+      result = @cache.get key
+    end
+  end
+
+  def test_get_with_server_timeout
+    socket = stub_server 43047
+    cache = Memcached.new("localhost:43047:1", :timeout => 0.5, :exception_retry_limit => 0)
+    assert 0.49 < (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    cache = Memcached.new("localhost:43047:1", :poll_timeout => 0.001, :rcv_timeout => 0.5, :exception_retry_limit => 0)
+    assert 0.49 < (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    cache = Memcached.new("localhost:43047:1", :poll_timeout => 0.25, :rcv_timeout => 0.25, :exception_retry_limit => 0)
+    assert 0.51 > (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    socket.close
+  end
+
+  def test_get_with_no_block_server_timeout
+    socket = stub_server 43048
+    cache = Memcached.new("localhost:43048:1", :no_block => true, :timeout => 0.25, :exception_retry_limit => 0)
+    assert 0.24 < (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    cache = Memcached.new("localhost:43048:1", :no_block => true, :poll_timeout => 0.25, :rcv_timeout => 0.001, :exception_retry_limit => 0)
+    assert 0.24 < (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    cache = Memcached.new("localhost:43048:1", :no_block => true,
+      :poll_timeout => 0.001,
+      :rcv_timeout => 0.25, # No affect in no-block mode
+      :exception_retry_limit => 0
+    )
+    assert 0.24 > (Benchmark.measure do
+      assert_raise(Memcached::ATimeoutOccurred) do
+        result = cache.get key
+      end
+    end).real
+
+    socket.close
+  end
+
+  def test_get_with_prefix_key
+    # Prefix_key
+    cache = Memcached.new(
+      # We can only use one server because the key is hashed separately from the prefix key
+      @servers.first,
+      :prefix_key => @prefix_key,
+      :hash => :default,
+      :distribution => :modula
+    )
+    cache.set key, @value
+    assert_equal @value, cache.get(key)
+
+    # No prefix_key specified
+    cache = Memcached.new(
+      @servers.first,
+      :hash => :default,
+      :distribution => :modula
+    )
+    assert_nothing_raised do
+      assert_equal @value, cache.get("#{@prefix_key}#{key}")
+    end
+  end
+
+  def test_values_with_null_characters_are_not_truncated
+    value = OpenStruct.new(:a => Object.new) # Marshals with a null \000
+    @cache.set key, value
+    result = @cache.get key, false
+    non_wrapped_result = Rlibmemcached.memcached_get(
+      @cache.instance_variable_get("@struct"),
+      key
+    ).first
+    assert result.size > non_wrapped_result.size
+  end
+
+  def test_get_multi
+    @cache.set "#{key}_1", 1
+    @cache.set "#{key}_2", 2
+    assert_equal({"#{key}_1" => 1, "#{key}_2" => 2},
+      @cache.get(["#{key}_1", "#{key}_2"]))
+  end
+
+  def test_get_multi_missing
+    @cache.set "#{key}_1", 1
+    @cache.delete "#{key}_2" rescue nil
+    @cache.set "#{key}_3", 3
+    @cache.delete "#{key}_4" rescue nil
+    assert_equal(
+      {"test_get_multi_missing_3"=>3, "test_get_multi_missing_1"=>1},
+      @cache.get(["#{key}_1", "#{key}_2",  "#{key}_3",  "#{key}_4"])
+     )
+  end
+
+  def test_get_multi_binary
+    @binary_protocol_cache.set "#{key}_1", 1
+    @binary_protocol_cache.delete "#{key}_2" rescue nil
+    @binary_protocol_cache.set "#{key}_3", 3
+    assert_equal(
+      {"test_get_multi_binary_3"=>3, "test_get_multi_binary_1"=>1},
+      @binary_protocol_cache.get(["#{key}_1", "#{key}_2",  "#{key}_3"])
+     )
+  end
+
+  def test_get_multi_binary_one_record_missing
+    @binary_protocol_cache.delete("magic_key") rescue nil
+    assert_equal({}, @binary_protocol_cache.get(["magic_key"]))
+  end
+
+  def test_get_multi_binary_one_record
+    @binary_protocol_cache.set("magic_key", 1)
+    assert_equal({"magic_key" => 1}, @binary_protocol_cache.get(["magic_key"]))
+  end
+
+  def test_get_multi_completely_missing
+    @cache.delete "#{key}_1" rescue nil
+    @cache.delete "#{key}_2" rescue nil
+    assert_equal(
+      {},
+      @cache.get(["#{key}_1", "#{key}_2"])
+     )
+  end
+
+  def test_get_multi_empty_string
+    @cache.set "#{key}_1", "", 0, false
+    assert_equal({"#{key}_1" => ""},
+      @cache.get(["#{key}_1"], false))
+  end
+
+  def test_get_multi_checks_types
+    assert_raises(TypeError, ArgumentError) do
+      @cache.get([nil])
+    end
+  end
+
+  def test_set_and_get_unmarshalled
+    @cache.set key, @value
+    result = @cache.get key, false
+    assert_equal @marshalled_value, result
+  end
+
+  def test_set_unmarshalled_and_get_unmarshalled
+    @cache.set key, @marshalled_value, 0, false
+    result = @cache.get key, false
+    assert_equal @marshalled_value, result
+  end
+
+  def test_set_unmarshalled_error
+    assert_raises(TypeError) do
+      @cache.set key, @value, 0, false
+    end
+  end
+
+  def test_get_multi_unmarshalled
+    @cache.set "#{key}_1", "1", 0, false
+    @cache.set "#{key}_2", "2", 0, false
+    assert_equal(
+      {"#{key}_1" => "1", "#{key}_2" => "2"},
+      @cache.get(["#{key}_1", "#{key}_2"], false)
+    )
+  end
+
+  def test_get_multi_mixed_marshalling
+    @cache.set "#{key}_1", 1
+    @cache.set "#{key}_2", "2", 0, false
+    assert_nothing_raised do
+      @cache.get(["#{key}_1", "#{key}_2"], false)
+    end
+    assert_raise(ArgumentError, TypeError) do
+      @cache.get(["#{key}_1", "#{key}_2"])
+    end
+  end
+
+  def test_random_distribution_is_statistically_random
+    cache = Memcached.new(@servers, :distribution => :random)
+    read_cache = Memcached.new(@servers.first)
+    hits = 4
+
+    while hits == 4 do
+      cache.flush
+      20.times do |i|
+        cache.set "#{key}#{i}", @value
+      end
+
+      hits = 0
+      20.times do |i|
+        begin
+          read_cache.get "#{key}#{i}"
+          hits += 1
+        rescue Memcached::NotFound
+        end
+      end
+    end
+
+    assert_not_equal 4, hits
+  end
+
+  # Set
+
+  def test_set
+    assert_nothing_raised do
+      @cache.set(key, @value)
+    end
+
+    assert_nothing_raised do
+      @binary_protocol_cache.set(key, @value)
+    end
+
+    assert_nothing_raised do
+      @udp_cache.set(key, @value)
+    end
+  end
+
+  def test_set_expiry
+    @cache.set key, @value, 1
+    assert_nothing_raised do
+      @cache.get key
+    end
+    sleep(2)
+    assert_raise(Memcached::NotFound) do
+      @cache.get key
+    end
+
+    assert_raise(TypeError) do
+      @cache.set key, @value, Time.now
+    end
+  end
+
+  def test_set_with_default_ttl
+    cache = Memcached.new(
+      @servers,
+      :default_ttl => 1
+    )
+    cache.set key, @value
+    assert_nothing_raised do
+      cache.get key
+    end
+    sleep(2)
+    assert_raise(Memcached::NotFound) do
+      cache.get key
+    end
+  end
+
+  def disabled_test_set_retry_on_client_error
+    # FIXME Test passes, but Mocha doesn't restore the original method properly
+    Rlibmemcached.stubs(:memcached_set).raises(Memcached::ClientError)
+    Rlibmemcached.stubs(:memcached_set).returns(0)
+
+    assert_nothing_raised do
+      @cache.set(key, @value)
+    end
+  end
+
+  # Delete
+
+  def test_delete
+    @cache.set key, @value
+    @cache.delete key
+    assert_raise(Memcached::NotFound) do
+      @cache.get key
+    end
+
+    @binary_protocol_cache.set key, @value
+    @binary_protocol_cache.delete key
+    assert_raise(Memcached::NotFound) do
+      @binary_protocol_cache.get key
+    end
+  end
+
+  def test_missing_delete
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotFound) do
+      @cache.delete key
+    end
+  end
+
+  # Flush
+
+  def test_flush
+    @cache.set key, @value
+    assert_equal @value,
+      @cache.get(key)
+    @cache.flush
+    assert_raise(Memcached::NotFound) do
+      @cache.get key
+    end
+  end
+
+  # Add
+
+  def test_add
+    @cache.delete key rescue nil
+    @cache.add key, @value
+    assert_equal @value, @cache.get(key)
+  end
+
+  def test_existing_add
+    @cache.set key, @value
+    assert_raise(Memcached::NotStored) do
+      @cache.add key, @value
+    end
+  end
+
+  def test_add_expiry
+    @cache.delete key rescue nil
+    @cache.set key, @value, 1
+    assert_nothing_raised do
+      @cache.get key
+    end
+    sleep(1)
+    assert_raise(Memcached::NotFound) do
+      @cache.get key
+    end
+  end
+
+  def test_unmarshalled_add
+    @cache.delete key rescue nil
+    @cache.add key, @marshalled_value, 0, false
+    assert_equal @marshalled_value, @cache.get(key, false)
+    assert_equal @value, @cache.get(key)
+  end
+
+  # Increment and decrement
+
+  def test_increment
+    @cache.set key, "10", 0, false
+    assert_equal 11, @cache.increment(key)
+  end
+
+  def test_increment_binary
+    @binary_protocol_cache.set key, "10", 0, false
+    assert_equal 11, @binary_protocol_cache.increment(key)
+  end
+
+  def test_increment_offset
+    @cache.set key, "10", 0, false
+    assert_equal 15, @cache.increment(key, 5)
+  end
+
+  def test_missing_increment
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotFound) do
+      @cache.increment key
+    end
+  end
+
+  def test_decrement
+    @cache.set key, "10", 0, false
+    assert_equal 9, @cache.decrement(key)
+  end
+
+  def test_decrement_binary
+    @binary_protocol_cache.set key, "10", 0, false
+    assert_equal 9, @binary_protocol_cache.decrement(key)
+  end
+
+  def test_decrement_offset
+    @cache.set key, "10", 0, false
+    assert_equal 5, @cache.decrement(key, 5)
+  end
+
+  def test_missing_decrement
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotFound) do
+      @cache.decrement key
+    end
+  end
+
+  # Exist
+
+  def test_missing_exist
+    assert_raise(Memcached::NotFound) do
+      @cache.exist key
+    end
+  end
+
+  def test_exist
+    @cache.set key, @value
+    assert_nil @cache.exist(key)
+  end
+
+  def test_missing_exist_binary
+    assert_raise(Memcached::NotFound) do
+      @binary_protocol_cache.exist key
+    end
+  end
+
+  def test_exist_binary
+    @binary_protocol_cache.set key, @value
+    assert_nil @binary_protocol_cache.exist(key)
+  end
+
+  # Replace
+
+  def test_replace
+    @cache.set key, nil
+    assert_nothing_raised do
+      @cache.replace key, @value
+    end
+    assert_equal @value, @cache.get(key)
+  end
+
+  def test_missing_replace
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotStored) do
+      @cache.replace key, @value
+    end
+    assert_raise(Memcached::NotFound) do
+      assert_equal @value, @cache.get(key)
+    end
+  end
+
+  # Append and prepend
+
+  def test_append
+    @cache.set key, "start", 0, false
+    assert_nothing_raised do
+      @cache.append key, "end"
+    end
+    assert_equal "startend", @cache.get(key, false)
+
+    @binary_protocol_cache.set key, "start", 0, false
+    assert_nothing_raised do
+      @binary_protocol_cache.append key, "end"
+    end
+    assert_equal "startend", @binary_protocol_cache.get(key, false)
+  end
+
+  def test_missing_append
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotStored) do
+      @cache.append key, "end"
+    end
+    assert_raise(Memcached::NotFound) do
+      assert_equal @value, @cache.get(key)
+    end
+
+    @binary_protocol_cache.delete key rescue nil
+    assert_raise(Memcached::NotStored) do
+      @binary_protocol_cache.append key, "end"
+    end
+    assert_raise(Memcached::NotFound) do
+      assert_equal @value, @binary_protocol_cache.get(key)
+    end
+  end
+
+  def test_prepend
+    @cache.set key, "end", 0, false
+    assert_nothing_raised do
+      @cache.prepend key, "start"
+    end
+    assert_equal "startend", @cache.get(key, false)
+  end
+
+  def test_missing_prepend
+    @cache.delete key rescue nil
+    assert_raise(Memcached::NotStored) do
+      @cache.prepend key, "end"
+    end
+    assert_raise(Memcached::NotFound) do
+      assert_equal @value, @cache.get(key)
+    end
+  end
+
+  def test_cas
+    cache = Memcached.new(
+      @servers,
+      :prefix_key => @prefix_key,
+      :support_cas => true
+    )
+    value2 = OpenStruct.new(:d => 3, :e => 4, :f => GenericClass)
+
+    # Existing set
+    cache.set key, @value
+    cache.cas(key) do |current|
+      assert_equal @value, current
+      value2
+    end
+    assert_equal value2, cache.get(key)
+
+    # Existing test without marshalling
+    cache.set(key, "foo", 0, false)
+    cache.cas(key, 0, false) do |current|
+      "#{current}bar"
+    end
+    assert_equal "foobar", cache.get(key, false)
+
+    # Missing set
+    cache.delete key
+    assert_raises(Memcached::NotFound) do
+      cache.cas(key) {}
+    end
+
+    # Conflicting set
+    cache.set key, @value
+    assert_raises(Memcached::ConnectionDataExists) do
+      cache.cas(key) do |current|
+        cache.set key, value2
+        current
+      end
+    end
+  end
+
+  # Error states
+
+  def test_key_with_spaces
+    key = "i have a space"
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.set key, @value
+    end
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.get(key)
+    end
+  end
+
+  def test_key_with_null
+    key = "with\000null"
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.set key, @value
+    end
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.get(key)
+    end
+
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      response = @cache.get([key])
+    end
+  end
+
+  def test_key_with_invalid_control_characters
+    key = "ch\303\242teau"
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.set key, @value
+    end
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.get(key)
+    end
+
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      response = @cache.get([key])
+    end
+  end
+
+  def test_key_too_long
+    key = "x"*251
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.set key, @value
+    end
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.get(key)
+    end
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @cache.get([key])
+    end
+  end
+
+  def test_verify_key_disabled
+    cache = Memcached.new @servers, :verify_key => false
+    key = "i have a space"
+    assert_raises(Memcached::ProtocolError) do
+      cache.set key, @value
+    end
+    assert_raises(Memcached::NotFound) do
+      cache.get key, @value
+    end
+  end
+
+  def test_key_error_message
+    key = "i have a space"
+    @cache.get key, @value
+    assert false # Never reached
+  rescue Memcached::ABadKeyWasProvidedOrCharactersOutOfRange => e
+    assert_match /#{key}/, e.message
+  end
+
+  def test_server_error_message
+    assert_raises(Memcached::NotStored) do
+      @cache.set key, "I'm big" * 1000000
+    end
+  end
+
+  def test_errno_message
+    Rlibmemcached::MemcachedServerSt.any_instance.stubs("cached_errno").returns(1)
+    @cache.send(:check_return_code, Rlibmemcached::MEMCACHED_ERRNO, key)
+  rescue Memcached::SystemError => e
+    assert_match /^Errno 1: "Operation not permitted". Key/, e.message
+  end
+
+  # Stats
+
+  def disable_test_stats
+    stats = @cache.stats
+    assert_equal 3, stats[:pid].size
+    assert_instance_of Fixnum, stats[:pid].first
+    assert_instance_of String, stats[:version].first
+
+    stats = @binary_protocol_cache.stats
+    assert_equal 3, stats[:pid].size
+    assert_instance_of Fixnum, stats[:pid].first
+    assert_instance_of String, stats[:version].first
+
+    assert_nothing_raised do
+      @noblock_cache.stats
+    end
+    assert_raises(TypeError) do
+      @udp_cache.stats
+    end
+  end
+
+  def test_missing_stats
+    cache = Memcached.new('localhost:43041')
+    assert_raises(Memcached::SomeErrorsWereReported) { cache.stats }
+  end
+
+  # Clone
+
+  def test_clone
+    cache = @cache.clone
+    assert_equal cache.servers, @cache.servers
+    assert_not_equal cache, @cache
+
+    # Definitely check that the structs are unlinked
+    assert_not_equal @cache.instance_variable_get('@struct').object_id,
+      cache.instance_variable_get('@struct').object_id
+
+    assert_nothing_raised do
+      @cache.set key, @value
+    end
+  end
+
+  # Pipelined, non-blocking IO
+
+  def test_buffered_requests_return_value
+    cache = Memcached.new @servers,
+      :buffer_requests => true
+    assert_nothing_raised do
+      cache.set key, @value
+    end
+    ret = Rlibmemcached.memcached_set(
+      cache.instance_variable_get("@struct"),
+      key,
+      @marshalled_value,
+      0,
+      Memcached::FLAGS
+    )
+    assert_equal Rlibmemcached::MEMCACHED_BUFFERED, ret
+  end
+
+  def test_no_block_return_value
+    assert_nothing_raised do
+      @noblock_cache.set key, @value
+    end
+    ret = Rlibmemcached.memcached_set(
+      @noblock_cache.instance_variable_get("@struct"),
+      key,
+      @marshalled_value,
+      0,
+      Memcached::FLAGS
+    )
+    assert_equal Rlibmemcached::MEMCACHED_BUFFERED, ret
+  end
+
+  def test_no_block_prepend
+    @cache.set key, "help", 0, false
+    @noblock_cache.prepend key, "help"
+    assert_equal "help",
+      @cache.get(key, false)
+    @noblock_cache.get "no_exist", false rescue nil
+    assert_equal "helphelp",
+      @cache.get(key, false)
+  end
+
+  def test_no_block_get
+    @noblock_cache.set key, @value
+    assert_equal @value,
+      @noblock_cache.get(key)
+  end
+
+  def test_no_block_missing_delete
+   @noblock_cache.delete key rescue nil
+    assert_nothing_raised do
+      @noblock_cache.delete key
+      @noblock_cache.delete key
+    end
+  end
+
+  def test_no_block_set_invalid_key
+    assert_raises(Memcached::ABadKeyWasProvidedOrCharactersOutOfRange) do
+      @noblock_cache.set "I'm so bad", @value
+    end
+  end
+
+  def test_no_block_set_object_too_large
+    assert_raise( Memcached::NotStored) do
+      @noblock_cache.set key, "I'm big" * 1000000
+    end
+  end
+
+  def test_no_block_existing_add
+    # Should still raise
+    noblock_cache = Memcached.new(@servers, @noblock_options.merge(:noreply => false))
+    noblock_cache.set key, @value
+    assert_raise(Memcached::NotStored) do
+      noblock_cache.add key, @value
+    end
+  end
+
+  # Server removal and consistent hashing
+
+  def test_unresponsive_server
+    socket = stub_server 43041
+
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 2,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 0
+    )
+
+    # Hit second server up to the server_failure_limit
+    key2 = 'test_missing_server'
+    assert_raise(Memcached::ATimeoutOccurred) { cache.set(key2, @value) }
+    assert_raise(Memcached::ATimeoutOccurred) { cache.get(key2, @value) }
+
+    # Hit second server and pass the limit
+    key2 = 'test_missing_server'
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    # Hit first server on retry
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    sleep(2)
+
+    # Hit second server again after restore, expect same failure
+    key2 = 'test_missing_server'
+    assert_raise(Memcached::ATimeoutOccurred) do
+      cache.set(key2, @value)
+    end
+
+    socket.close
+  end
+
+  def test_unresponsive_server_retries_greater_than_server_failure_limit
+    socket = stub_server 43041
+
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 2,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 3
+    )
+
+    key2 = 'test_missing_server'
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    sleep(2)
+
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    socket.close
+  end
+
+  def test_unresponsive_server_retries_equals_server_failure_limit
+    socket = stub_server 43041
+
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 2,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 3
+    )
+
+    key2 = 'test_missing_server'
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    sleep(2)
+
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    socket.close
+  end
+
+  def test_unresponsive_server_retries_less_than_server_failure_limit
+    socket = stub_server 43041
+
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 2,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 1
+    )
+
+    key2 = 'test_missing_server'
+    assert_raise(Memcached::ATimeoutOccurred) { cache.set(key2, @value) }
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    sleep(2)
+
+    assert_raise(Memcached::ATimeoutOccurred) { cache.set(key2, @value) }
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    socket.close
+  end
+
+  def test_wrong_failure_counter
+    cache = Memcached.new(
+      [@servers.last],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 1,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 0
+                          )
+
+    # This is an abuse of knowledge, but it's necessary to verify that
+    # the library is handling the counter properly.
+    struct = cache.instance_variable_get(:@struct)
+    server = Memcached::Lib.memcached_server_by_key(struct, "marmotte").first
+
+    # set to ensure connectivity
+    cache.set("marmotte", "milk")
+    server.server_failure_counter = 0
+    cache.quit
+    assert_equal 0, server.server_failure_counter
+  end
+
+  def test_missing_server
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :prefix_key => @prefix_key,
+      :auto_eject_hosts => true,
+      :server_failure_limit => 2,
+      :retry_timeout => 1,
+      :hash_with_prefix_key => false,
+      :hash => :md5,
+      :exception_retry_limit => 0
+    )
+
+    # Hit second server up to the server_failure_limit
+    key2 = 'test_missing_server'
+    assert_raise(Memcached::SystemError) { cache.set(key2, @value) }
+    assert_raise(Memcached::SystemError) { cache.get(key2, @value) }
+
+    # Hit second server and pass the limit
+    key2 = 'test_missing_server'
+    begin
+      cache.get(key2)
+    rescue => e
+      assert_equal Memcached::ServerIsMarkedDead, e.class
+      assert_match /localhost:43041/, e.message
+    end
+
+    # Hit first server on retry
+    assert_nothing_raised do
+      cache.set(key2, @value)
+      assert_equal cache.get(key2), @value
+    end
+
+    sleep(2)
+
+    # Hit second server again after restore, expect same failure
+    key2 = 'test_missing_server'
+    assert_raise(Memcached::SystemError) do
+      cache.set(key2, @value)
+    end
+  end
+
+  def test_unresponsive_with_random_distribution
+    socket = stub_server 43041
+    failures = [Memcached::ATimeoutOccurred, Memcached::ServerIsMarkedDead]
+
+    cache = Memcached.new(
+      [@servers.last, 'localhost:43041'],
+      :auto_eject_hosts => true,
+      :distribution => :random,
+      :server_failure_limit => 1,
+      :retry_timeout => 1,
+      :exception_retry_limit => 0
+    )
+
+    # Provoke the errors in 'failures'
+    exceptions = []
+    100.times { begin; cache.set key, @value; rescue => e; exceptions << e; end }
+    assert_equal failures, exceptions.map { |x| x.class }
+
+    # Hit first server on retry
+    assert_nothing_raised { cache.set(key, @value) }
+
+    # Hit second server again after restore, expect same failures
+    sleep(2)
+    exceptions = []
+    100.times { begin; cache.set key, @value; rescue => e; exceptions << e; end }
+    assert_equal failures, exceptions.map { |x| x.class }
+
+    socket.close
+  end
+
+  def test_consistent_hashing
+    keys = %w(EN6qtgMW n6Oz2W4I ss4A8Brr QShqFLZt Y3hgP9bs CokDD4OD Nd3iTSE1 24vBV4AU H9XBUQs5 E5j8vUq1 AzSh8fva PYBlK2Pi Ke3TgZ4I AyAIYanO oxj8Xhyd eBFnE6Bt yZyTikWQ pwGoU7Pw 2UNDkKRN qMJzkgo2 keFXbQXq pBl2QnIg ApRl3mWY wmalTJW1 TLueug8M wPQL4Qfg uACwus23 nmOk9R6w lwgZJrzJ v1UJtKdG RK629Cra U2UXFRqr d9OQLNl8 KAm1K3m5 Z13gKZ1v tNVai1nT LhpVXuVx pRib1Itj I1oLUob7 Z1nUsd5Q ZOwHehUa aXpFX29U ZsnqxlGz ivQRjOdb mB3iBEAj)
+
+    # Five servers
+    cache = Memcached.new(
+      @servers + ['localhost:43044', 'localhost:43045', 'localhost:43046'],
+      :prefix_key => @prefix_key
+    )
+
+    cache.flush
+    keys.each do |key|
+      cache.set(key, @value)
+    end
+
+    # Pull a server
+    cache = Memcached.new(
+      @servers + ['localhost:43044', 'localhost:43046'],
+      :prefix_key => @prefix_key
+    )
+
+    failed = 0
+    keys.each_with_index do |key, i|
+      begin
+        cache.get(key)
+      rescue Memcached::NotFound
+        failed += 1
+      end
+    end
+
+    assert(failed < keys.size / 3, "#{failed} failed out of #{keys.size}")
+  end
+
+  # Concurrency
+
+  def test_thread_contention
+    threads = []
+    4.times do |index|
+      threads << Thread.new do
+        cache = @cache.clone
+        assert_nothing_raised do
+          cache.set("test_thread_contention_#{index}", index)
+        end
+        assert_equal index, cache.get("test_thread_contention_#{index}")
+      end
+    end
+    threads.each {|thread| thread.join}
+  end
+
+  # Hash
+
+  def test_hash
+    assert_equal 3157003241,
+      Rlibmemcached.memcached_generate_hash_rvalue("test", Rlibmemcached::MEMCACHED_HASH_FNV1_32)
+  end
+
+  def test_noop_hash
+    %w[foo bar baz qux quux].each do |word|
+      assert_equal 1,
+        Rlibmemcached.memcached_generate_hash_rvalue(word, Rlibmemcached::MEMCACHED_HASH_NONE)
+    end
+  end
+
+  # Memory cleanup
+
+  def test_reset
+    original_struct = @cache.instance_variable_get("@struct")
+    assert_nothing_raised do
+      @cache.reset
+    end
+    assert_not_equal original_struct,
+      @cache.instance_variable_get("@struct")
+  end
+
+  # NOTE: This breaks encapsulation, but there's no other easy way to test this without
+  # mocking out Rlibmemcached calls
+  def test_reraise_invalid_return_code
+    assert_raise Memcached::Error do
+      @cache.send(:check_return_code, 5000)
+    end
+  end
+
+  private
+
+  def key
+    caller.first[/.*[` ](.*)'/, 1] # '
+  end
+
+  def stub_server(port)
+    socket = TCPServer.new('127.0.0.1', port)
+    Thread.new { socket.accept }
+    socket
+  end
+
+end
+