1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3 * Libmemcached Client and Server
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
15 * * Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following disclaimer
17 * in the documentation and/or other materials provided with the
20 * * The names of its contributors may not be used to endorse or
21 * promote products derived from this software without specific prior
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
40 #include <tests/virtual_buckets.h>
42 #include <libmemcached/memcached.h>
44 struct libtest_string_t
{
49 static inline libtest_string_t
libtest_string(const char *arg
, size_t arg_size
)
51 libtest_string_t local
= { arg
, arg_size
};
55 #define make_libtest_string(X) libtest_string((X), static_cast<size_t>(sizeof(X) - 1))
57 static libtest_string_t libtest_string_t_null
= { 0, 0};
59 static bool libtest_string_is_null(const libtest_string_t
&string
)
61 if (string
.c_str
== 0 and string
.size
== 0)
73 expect_t basic_keys
[]= {
74 { make_libtest_string("hello"), 0, 0 },
75 { make_libtest_string("doctor"), 0, 0 },
76 { make_libtest_string("name"), 1, 3 },
77 { make_libtest_string("continue"), 1, 3 },
78 { make_libtest_string("yesterday"), 0, 0 },
79 { make_libtest_string("tomorrow"), 1, 1 },
80 { make_libtest_string("another key"), 2, 2 },
81 { libtest_string_t_null
, 0, 0 }
84 test_return_t
virtual_back_map(memcached_st
*)
86 memcached_return_t rc
;
87 memcached_server_st
*server_pool
;
90 memc
= memcached_create(NULL
);
93 uint32_t server_map
[] = { 0, 1, 2, 1 };
94 rc
= memcached_bucket_set(memc
, server_map
, NULL
, 4, 2);
95 test_true(rc
== MEMCACHED_SUCCESS
);
97 memcached_server_distribution_t dt
;
98 dt
= memcached_behavior_get_distribution(memc
);
99 test_true(dt
== MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
);
101 memcached_behavior_set_key_hash(memc
, MEMCACHED_HASH_CRC
);
102 test_true(rc
== MEMCACHED_SUCCESS
);
104 memcached_hash_t hash_type
= memcached_behavior_get_key_hash(memc
);
105 test_true(hash_type
== MEMCACHED_HASH_CRC
);
107 server_pool
= memcached_servers_parse("localhost:11211, localhost1:11210, localhost2:11211");
108 test_true(server_pool
);
109 memcached_server_push(memc
, server_pool
);
111 /* verify that the server list was parsed okay. */
112 test_true(memcached_server_count(memc
) == 3);
113 test_true(strcmp(server_pool
[0].hostname
, "localhost") == 0);
114 test_true(server_pool
[0].port
== 11211);
116 test_true(strcmp(server_pool
[1].hostname
, "localhost1") == 0);
117 test_true(server_pool
[1].port
== 11210);
119 test_true(strcmp(server_pool
[2].hostname
, "localhost2") == 0);
120 test_true(server_pool
[2].port
== 11211);
122 dt
= memcached_behavior_get_distribution(memc
);
123 hash_type
= memcached_behavior_get_key_hash(memc
);
124 test_true(dt
== MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
);
125 test_true(hash_type
== MEMCACHED_HASH_CRC
);
127 /* verify the standard ketama set. */
128 for (expect_t
*ptr
= basic_keys
; not libtest_string_is_null(ptr
->key
); ptr
++)
130 uint32_t server_idx
= memcached_generate_hash(memc
, ptr
->key
.c_str
, ptr
->key
.size
);
133 snprintf(buffer
, sizeof(buffer
), "%.*s:%lu Got/Expected %u == %u", (int)ptr
->key
.size
, ptr
->key
.c_str
, (unsigned long)ptr
->key
.size
, server_idx
, ptr
->server_id
);
134 test_true_got(server_idx
== ptr
->server_id
, buffer
);
137 memcached_server_list_free(server_pool
);
138 memcached_free(memc
);