This commit is contained in:
Evan Weaver 2008-01-19 02:44:47 +00:00
Родитель dac403f892
Коммит f60463fa82
5 изменённых файлов: 708 добавлений и 1 удалений

9
test/setup.rb Normal file
Просмотреть файл

@ -0,0 +1,9 @@
# Start memcached
`ps awx`.split("\n").grep(/4304[1-3]/).map do |process|
system("kill -9 #{process.to_i}")
end
system "memcached -p 43042 &"
system "memcached -p 43043 &"

0
test/teardown.rb Normal file
Просмотреть файл

4
test/test_helper.rb Normal file
Просмотреть файл

@ -0,0 +1,4 @@
$LOAD_PATH << "#{File.dirname(__FILE__)}/../lib"
require 'memcached'
require 'test/unit'

Просмотреть файл

@ -1,2 +1,8 @@
require "#{File.dirname(__FILE__)}/../../lib/memcached"
require "#{File.dirname(__FILE__)}/../test_helper"
class BindingTest < Test::Unit::TestCase
def test_libmemcached_loaded
assert_nothing_raised { Libmemcached }
end
end

Просмотреть файл

@ -0,0 +1,688 @@
require "#{File.dirname(__FILE__)}/../test_helper"
class ClassTest < Test::Unit::TestCase
def setup
@cache = Memcached.new 'localhost:11211', :namespace => 'my_namespace'
end
def test_cache_get
server = util_setup_fake_server
assert_equal "\004\b\"\0170123456789",
@cache.cache_get(server, 'my_namespace:key')
assert_equal "get my_namespace:key\r\n",
server.socket.written.string
end
def test_cache_get_EOF
server = util_setup_fake_server
server.socket.data.string = ''
e = assert_raise Memcached::MemcachedError do
@cache.cache_get server, 'my_namespace:key'
end
assert_equal "lost connection to example.com:11211", e.message
end
def test_cache_get_bad_state
server = FakeServer.new
server.socket.data.write "bogus response\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
e = assert_raise Memcached::MemcachedError do
@cache.cache_get(server, 'my_namespace:key')
end
assert_equal "unexpected response \"bogus response\\r\\n\"", e.message
deny server.alive?
assert_equal "get my_namespace:key\r\n",
server.socket.written.string
end
def test_cache_get_miss
socket = FakeSocket.new
socket.data.write "END\r\n"
socket.data.rewind
server = FakeServer.new socket
assert_equal nil, @cache.cache_get(server, 'my_namespace:key')
assert_equal "get my_namespace:key\r\n",
socket.written.string
end
def test_cache_get_multi
server = util_setup_fake_server
server.socket.data.write "VALUE foo 0 7\r\n"
server.socket.data.write "\004\b\"\bfoo\r\n"
server.socket.data.write "VALUE bar 0 7\r\n"
server.socket.data.write "\004\b\"\bbar\r\n"
server.socket.data.write "END\r\n"
server.socket.data.rewind
result = @cache.cache_get_multi server, 'foo bar baz'
assert_equal 2, result.length
assert_equal "\004\b\"\bfoo", result['foo']
assert_equal "\004\b\"\bbar", result['bar']
end
def test_cache_get_multi_EOF
server = util_setup_fake_server
server.socket.data.string = ''
e = assert_raise Memcached::MemcachedError do
@cache.cache_get_multi server, 'my_namespace:key'
end
assert_equal "lost connection to example.com:11211", e.message
end
def test_cache_get_multi_bad_state
server = FakeServer.new
server.socket.data.write "bogus response\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
e = assert_raise Memcached::MemcachedError do
@cache.cache_get_multi server, 'my_namespace:key'
end
assert_equal "unexpected response \"bogus response\\r\\n\"", e.message
deny server.alive?
assert_equal "get my_namespace:key\r\n",
server.socket.written.string
end
def test_crc32_ITU_T
assert_equal 0, ''.crc32_ITU_T
assert_equal 1260851911, 'my_namespace:key'.crc32_ITU_T
end
def test_initialize
cache = Memcached.new :namespace => 'my_namespace', :readonly => true
assert_equal 'my_namespace', cache.namespace
assert_equal true, cache.readonly?
assert_equal true, cache.servers.empty?
end
def test_initialize_compatible
cache = Memcached.new ['localhost:11211', 'localhost:11212'],
:namespace => 'my_namespace', :readonly => true
assert_equal 'my_namespace', cache.namespace
assert_equal true, cache.readonly?
assert_equal false, cache.servers.empty?
end
def test_initialize_compatible_no_hash
cache = Memcached.new ['localhost:11211', 'localhost:11212']
assert_equal nil, cache.namespace
assert_equal false, cache.readonly?
assert_equal false, cache.servers.empty?
end
def test_initialize_compatible_one_server
cache = Memcached.new 'localhost:11211'
assert_equal nil, cache.namespace
assert_equal false, cache.readonly?
assert_equal false, cache.servers.empty?
end
def test_initialize_compatible_bad_arg
e = assert_raise ArgumentError do
cache = Memcached.new Object.new
end
assert_equal 'first argument must be Array, Hash or String', e.message
end
def test_initialize_multiple_servers
cache = Memcached.new %w[localhost:11211 localhost:11212],
:namespace => 'my_namespace', :readonly => true
assert_equal 'my_namespace', cache.namespace
assert_equal true, cache.readonly?
assert_equal false, cache.servers.empty?
deny_empty cache.instance_variable_get(:@buckets)
end
def test_initialize_too_many_args
assert_raises ArgumentError do
Memcached.new 1, 2, 3
end
end
def test_decr
server = FakeServer.new
server.socket.data.write "5\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.decr 'key'
assert_equal "decr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal 5, value
end
def test_decr_not_found
server = FakeServer.new
server.socket.data.write "NOT_FOUND\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.decr 'key'
assert_equal "decr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal nil, value
end
def test_decr_space_padding
server = FakeServer.new
server.socket.data.write "5 \r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.decr 'key'
assert_equal "decr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal 5, value
end
def test_get
util_setup_fake_server
value = @cache.get 'key'
assert_equal "get my_namespace:key\r\n",
@cache.servers.first.socket.written.string
assert_equal '0123456789', value
end
def test_get_bad_key
util_setup_fake_server
assert_raise ArgumentError do @cache.get 'k y' end
util_setup_fake_server
assert_raise ArgumentError do @cache.get 'k' * 250 end
end
def test_get_cache_get_IOError
socket = Object.new
def socket.write(arg) raise IOError, 'some io error'; end
server = FakeServer.new socket
@cache.servers = []
@cache.servers << server
e = assert_raise Memcached::MemcachedError do
@cache.get 'my_namespace:key'
end
assert_equal 'some io error', e.message
end
def test_get_cache_get_SystemCallError
socket = Object.new
def socket.write(arg) raise SystemCallError, 'some syscall error'; end
server = FakeServer.new socket
@cache.servers = []
@cache.servers << server
e = assert_raise Memcached::MemcachedError do
@cache.get 'my_namespace:key'
end
assert_equal 'unknown error - some syscall error', e.message
end
def test_get_no_connection
@cache.servers = 'localhost:1'
e = assert_raise Memcached::MemcachedError do
@cache.get 'key'
end
assert_equal 'No connection to server', e.message
end
def test_get_no_servers
@cache.servers = []
e = assert_raise Memcached::MemcachedError do
@cache.get 'key'
end
assert_equal 'No active servers', e.message
end
def test_get_multi
server = FakeServer.new
server.socket.data.write "VALUE my_namespace:key 0 14\r\n"
server.socket.data.write "\004\b\"\0170123456789\r\n"
server.socket.data.write "VALUE my_namespace:keyb 0 14\r\n"
server.socket.data.write "\004\b\"\0179876543210\r\n"
server.socket.data.write "END\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
values = @cache.get_multi 'key', 'keyb'
assert_equal "get my_namespace:key my_namespace:keyb\r\n",
server.socket.written.string
expected = { 'key' => '0123456789', 'keyb' => '9876543210' }
assert_equal expected.sort, values.sort
end
def test_get_raw
server = FakeServer.new
server.socket.data.write "VALUE my_namespace:key 0 10\r\n"
server.socket.data.write "0123456789\r\n"
server.socket.data.write "END\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.get 'key', true
assert_equal "get my_namespace:key\r\n",
@cache.servers.first.socket.written.string
assert_equal '0123456789', value
end
def test_get_server_for_key
server = @cache.get_server_for_key 'key'
assert_equal 'localhost', server.host
assert_equal 1, server.port
end
def test_get_server_for_key_multiple
s1 = util_setup_server @cache, 'one.example.com', ''
s2 = util_setup_server @cache, 'two.example.com', ''
@cache.instance_variable_set :@servers, [s1, s2]
@cache.instance_variable_set :@buckets, [s1, s2]
server = @cache.get_server_for_key 'keya'
assert_equal 'two.example.com', server.host
server = @cache.get_server_for_key 'keyb'
assert_equal 'one.example.com', server.host
end
def test_get_server_for_key_no_servers
@cache.servers = []
e = assert_raise Memcached::MemcachedError do
@cache.get_server_for_key 'key'
end
assert_equal 'No servers available', e.message
end
def test_get_server_for_key_spaces
e = assert_raise ArgumentError do
@cache.get_server_for_key 'space key'
end
assert_equal 'illegal character in key "space key"', e.message
end
def test_get_server_for_key_length
@cache.get_server_for_key 'x' * 250
long_key = 'x' * 251
e = assert_raise ArgumentError do
@cache.get_server_for_key long_key
end
assert_equal "key too long #{long_key.inspect}", e.message
end
def test_incr
server = FakeServer.new
server.socket.data.write "5\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.incr 'key'
assert_equal "incr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal 5, value
end
def test_incr_not_found
server = FakeServer.new
server.socket.data.write "NOT_FOUND\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.incr 'key'
assert_equal "incr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal nil, value
end
def test_incr_space_padding
server = FakeServer.new
server.socket.data.write "5 \r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
value = @cache.incr 'key'
assert_equal "incr my_namespace:key 1\r\n",
@cache.servers.first.socket.written.string
assert_equal 5, value
end
def test_make_cache_key
assert_equal 'my_namespace:key', @cache.make_cache_key('key')
@cache.namespace = nil
assert_equal 'key', @cache.make_cache_key('key')
end
def test_servers
server = FakeServer.new
@cache.servers = []
@cache.servers << server
assert_equal [server], @cache.servers
end
def test_servers_equals_type_error
e = assert_raise TypeError do
@cache.servers = [Object.new]
end
assert_equal 'cannot convert Object into Memcached::Server', e.message
end
def test_set
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.set 'key', 'value'
expected = "set my_namespace:key 0 0 9\r\n\004\b\"\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_set_expiry
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.set 'key', 'value', 5
expected = "set my_namespace:key 0 5 9\r\n\004\b\"\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_set_raw
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.set 'key', 'value', 0, true
expected = "set my_namespace:key 0 0 5\r\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_set_readonly
cache = Memcached.new :readonly => true
e = assert_raise Memcached::MemcachedError do
cache.set 'key', 'value'
end
assert_equal 'Update of readonly cache', e.message
end
def test_set_too_big
server = FakeServer.new
server.socket.data.write "SERVER_ERROR object too large for cache\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
e = assert_raise Memcached::MemcachedError do
@cache.set 'key', 'v'
end
assert_equal 'object too large for cache', e.message
end
def test_add
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.add 'key', 'value'
expected = "add my_namespace:key 0 0 9\r\n\004\b\"\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_add_exists
server = FakeServer.new
server.socket.data.write "NOT_STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.add 'key', 'value'
expected = "add my_namespace:key 0 0 9\r\n\004\b\"\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_add_expiry
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.add 'key', 'value', 5
expected = "add my_namespace:key 0 5 9\r\n\004\b\"\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_add_raw
server = FakeServer.new
server.socket.data.write "STORED\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
@cache.add 'key', 'value', 0, true
expected = "add my_namespace:key 0 0 5\r\nvalue\r\n"
assert_equal expected, server.socket.written.string
end
def test_add_readonly
cache = Memcached.new :readonly => true
e = assert_raise Memcached::MemcachedError do
cache.add 'key', 'value'
end
assert_equal 'Update of readonly cache', e.message
end
def test_delete
server = FakeServer.new
@cache.servers = []
@cache.servers << server
@cache.delete 'key'
expected = "delete my_namespace:key 0\r\n"
assert_equal expected, server.socket.written.string
end
def test_delete_with_expiry
server = FakeServer.new
@cache.servers = []
@cache.servers << server
@cache.delete 'key', 300
expected = "delete my_namespace:key 300\r\n"
assert_equal expected, server.socket.written.string
end
def test_flush_all
@cache.servers = []
3.times { @cache.servers << FakeServer.new }
@cache.flush_all
expected = "flush_all\r\n"
@cache.servers.each do |server|
assert_equal expected, server.socket.written.string
end
end
def test_flush_all_failure
socket = FakeSocket.new
socket.data.write "ERROR\r\n"
socket.data.rewind
server = FakeServer.new socket
def server.host() "localhost"; end
def server.port() 11211; end
@cache.servers = []
@cache.servers << server
assert_raise Memcached::MemcachedError do
@cache.flush_all
end
assert_equal "flush_all\r\n", socket.written.string
end
def test_stats
socket = FakeSocket.new
socket.data.write "STAT pid 20188\r\nSTAT total_items 32\r\nSTAT version 1.2.3\r\nSTAT rusage_user 1:300\r\nSTAT dummy ok\r\nEND\r\n"
socket.data.rewind
server = FakeServer.new socket
def server.host() 'localhost'; end
def server.port() 11211; end
@cache.servers = []
@cache.servers << server
expected = {
'localhost:11211' => {
'pid' => 20188, 'total_items' => 32, 'version' => '1.2.3',
'rusage_user' => 1.0003, 'dummy' => 'ok'
}
}
assert_equal expected, @cache.stats
assert_equal "stats\r\n", socket.written.string
end
def test_basic_threaded_operations_should_work
cache = Memcached.new :multithread => true,
:namespace => 'my_namespace',
:readonly => false
server = util_setup_server cache, 'example.com', "OK\r\n"
cache.instance_variable_set :@servers, [server]
assert_nothing_raised do
cache.set "test", "test value"
end
end
def test_basic_unthreaded_operations_should_work
cache = Memcached.new :multithread => false,
:namespace => 'my_namespace',
:readonly => false
server = util_setup_server cache, 'example.com', "OK\r\n"
cache.instance_variable_set :@servers, [server]
assert_nothing_raised do
cache.set "test", "test value"
end
end
def util_setup_fake_server
server = FakeServer.new
server.socket.data.write "VALUE my_namespace:key 0 14\r\n"
server.socket.data.write "\004\b\"\0170123456789\r\n"
server.socket.data.write "END\r\n"
server.socket.data.rewind
@cache.servers = []
@cache.servers << server
return server
end
def util_setup_server(memcache, host, responses)
server = Memcached::Server.new memcache, host
server.instance_variable_set :@sock, StringIO.new(responses)
@cache.servers = []
@cache.servers << server
return server
end
end