X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=tests%2Ftest.c;h=47db82c733a2e43aec483f3e5fe28d281a570e28;hb=7f50da6ed2f30cc137b51415edc64810ef84c69a;hp=85adaf1cb3006dc28edffaf0597b558a33958800;hpb=82b9ad4652c38c7fc0a31505fd38f95246e379f9;p=awesomized%2Flibmemcached diff --git a/tests/test.c b/tests/test.c index 85adaf1c..47db82c7 100644 --- a/tests/test.c +++ b/tests/test.c @@ -2,13 +2,17 @@ Sample test application. */ #include -#include -#include #include #include #include +#include +#include +#include #include -#include "../lib/common.h" +#include +#include "server.h" + +#include "test.h" long int timedif(struct timeval a, struct timeval b) { @@ -21,786 +25,30 @@ long int timedif(struct timeval a, struct timeval b) return s + us; } -void init_test(memcached_st *not_used) -{ - memcached_st memc; - - (void)memcached_create(&memc); - memcached_free(&memc); -} - -void allocation_test(memcached_st *not_used) -{ - memcached_st *memc; - memc= memcached_create(NULL); - assert(memc); - memcached_free(memc); -} - -void connection_test(memcached_st *memc) -{ - memcached_return rc; - - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); -} - -void error_test(memcached_st *memc) -{ - memcached_return rc; - - for (rc= MEMCACHED_SUCCESS; rc < MEMCACHED_MAXIMUM_RETURN; rc++) - { - printf("Error %d -> %s\n", rc, memcached_strerror(memc, rc)); - } -} - -void set_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); -} - -void add_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - rc= memcached_add(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_NOTSTORED); -} - -void replace_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - rc= memcached_replace(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); -} - -void delete_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - rc= memcached_delete(memc, key, strlen(key), (time_t)0); - assert(rc == MEMCACHED_SUCCESS); -} - -void flush_test(memcached_st *memc) -{ - memcached_return rc; - - rc= memcached_flush(memc, 0); - assert(rc == MEMCACHED_SUCCESS); -} - -void get_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *string; - size_t string_length; - uint16_t flags; - - rc= memcached_delete(memc, key, strlen(key), (time_t)0); - assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_NOTFOUND); - - string= memcached_get(memc, key, strlen(key), - &string_length, &flags, &rc); - - assert(rc == MEMCACHED_NOTFOUND); - assert(string_length == 0); - assert(!string); -} - -void get_test2(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - char *string; - size_t string_length; - uint16_t flags; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - string= memcached_get(memc, key, strlen(key), - &string_length, &flags, &rc); - - assert(string); - assert(rc == MEMCACHED_SUCCESS); - assert(string_length == strlen(value)); - assert(!memcmp(string, value, string_length)); - - free(string); -} - -void set_test2(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value= "train in the brain"; - size_t value_length= strlen(value); - unsigned int x; - - for (x= 0; x < 10; x++) - { - rc= memcached_set(memc, key, strlen(key), - value, value_length, - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - } -} - -void set_test3(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value; - size_t value_length= 8191; - unsigned int x; - - value = (char*)malloc(value_length); - assert(value); - - for (x= 0; x < value_length; x++) - value[x] = (char) (x % 127); - - for (x= 0; x < 1; x++) - { - rc= memcached_set(memc, key, strlen(key), - value, value_length, - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - } - - free(value); -} - -void get_test3(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value; - size_t value_length= 8191; - char *string; - size_t string_length; - uint16_t flags; - int x; - - value = (char*)malloc(value_length); - assert(value); - - for (x= 0; x < value_length; x++) - value[x] = (char) (x % 127); - - rc= memcached_set(memc, key, strlen(key), - value, value_length, - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - string= memcached_get(memc, key, strlen(key), - &string_length, &flags, &rc); - - assert(rc == MEMCACHED_SUCCESS); - assert(string); - assert(string_length == value_length); - assert(!memcmp(string, value, string_length)); - - free(string); - free(value); -} - -void get_test4(memcached_st *memc) -{ - memcached_return rc; - char *key= "foo"; - char *value; - size_t value_length= 8191; - char *string; - size_t string_length; - uint16_t flags; - int x; - - value = (char*)malloc(value_length); - assert(value); - - for (x= 0; x < value_length; x++) - value[x] = (char) (x % 127); - - rc= memcached_set(memc, key, strlen(key), - value, value_length, - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - for (x= 0; x < 10; x++) - { - string= memcached_get(memc, key, strlen(key), - &string_length, &flags, &rc); - - assert(rc == MEMCACHED_SUCCESS); - assert(string); - assert(string_length == value_length); - assert(!memcmp(string, value, string_length)); - free(string); - } - - free(value); -} - -void stats_servername_test(memcached_st *memc) -{ - memcached_return rc; - memcached_stat_st stat; - rc= memcached_stat_servername(&stat, NULL, - "localhost", - MEMCACHED_DEFAULT_PORT); -} - -void increment_test(memcached_st *memc) -{ - unsigned int new_number; - memcached_return rc; - char *key= "number"; - char *value= "0"; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - rc= memcached_increment(memc, key, strlen(key), - 1, &new_number); - assert(rc == MEMCACHED_SUCCESS); - assert(new_number == 1); - - rc= memcached_increment(memc, key, strlen(key), - 1, &new_number); - assert(rc == MEMCACHED_SUCCESS); - assert(new_number == 2); -} - -void decrement_test(memcached_st *memc) -{ - unsigned int new_number; - memcached_return rc; - char *key= "number"; - char *value= "3"; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - rc= memcached_decrement(memc, key, strlen(key), - 1, &new_number); - assert(rc == MEMCACHED_SUCCESS); - assert(new_number == 2); - - rc= memcached_decrement(memc, key, strlen(key), - 1, &new_number); - assert(rc == MEMCACHED_SUCCESS); - assert(new_number == 1); -} - -void quit_test(memcached_st *memc) -{ - memcached_return rc; - char *key= "fudge"; - char *value= "sanford and sun"; - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)10, (uint16_t)3); - assert(rc == MEMCACHED_SUCCESS); - memcached_quit(memc); - - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)50, (uint16_t)9); - assert(rc == MEMCACHED_SUCCESS); -} - -void mget_test(memcached_st *memc) -{ - memcached_return rc; - char *keys[]= {"fudge", "son", "food"}; - size_t key_length[]= {5, 3, 4}; - unsigned int x; - uint16_t flags; - - char return_key[MEMCACHED_MAX_KEY]; - size_t return_key_length; - char *return_value; - size_t return_value_length; - - /* We need to empty the server before continueing test */ - rc= memcached_flush(memc, 0); - assert(rc == MEMCACHED_SUCCESS); - - rc= memcached_mget(memc, keys, key_length, 3); - assert(rc == MEMCACHED_SUCCESS); - - while ((return_value= memcached_fetch(memc, return_key, &return_key_length, - &return_value_length, &flags, &rc)) != NULL) - { - assert(return_value); - } - assert(!return_value); - assert(return_value_length == 0); - assert(rc == MEMCACHED_NOTFOUND); - - for (x= 0; x < 3; x++) - { - rc= memcached_set(memc, keys[x], key_length[x], - keys[x], key_length[x], - (time_t)50, (uint16_t)9); - assert(rc == MEMCACHED_SUCCESS); - } - - rc= memcached_mget(memc, keys, key_length, 3); - assert(rc == MEMCACHED_SUCCESS); - - x= 0; - while ((return_value= memcached_fetch(memc, return_key, &return_key_length, - &return_value_length, &flags, &rc))) - { - assert(return_value); - assert(rc == MEMCACHED_SUCCESS); - assert(return_key_length == return_value_length); - assert(!memcmp(return_value, return_key, return_value_length)); - free(return_value); - x++; - } -} - -void get_stats_keys(memcached_st *memc) -{ - char **list; - char **ptr; - memcached_stat_st stat; - memcached_return rc; - - list= memcached_stat_get_keys(memc, &stat, &rc); - assert(rc == MEMCACHED_SUCCESS); - for (ptr= list; *ptr; ptr++) - printf("Found key %s\n", *ptr); - fflush(stdout); - - free(list); -} - -void get_stats(memcached_st *memc) -{ - unsigned int x; - char **list; - char **ptr; - memcached_return rc; - memcached_stat_st *stat; - - stat= memcached_stat(memc, NULL, &rc); - assert(rc == MEMCACHED_SUCCESS); - - assert(rc == MEMCACHED_SUCCESS); - assert(stat); - - for (x= 0; x < memcached_server_count(memc); x++) - { - list= memcached_stat_get_keys(memc, &stat[x], &rc); - assert(rc == MEMCACHED_SUCCESS); - for (ptr= list; *ptr; ptr++) - printf("Found key %s\n", *ptr); - - free(list); - } - - free(stat); -} - -void add_host_test(memcached_st *memc) -{ - unsigned int x; - memcached_server_st *servers; - memcached_return rc; - char servername[]= "0.example.com"; - - servers= memcached_server_list_append(NULL, servername, 400, &rc); - assert(servers); - assert(1 == memcached_server_list_count(servers)); - - for (x= 2; x < 20; x++) - { - char buffer[SMALL_STRING_LEN]; - - snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x); - servers= memcached_server_list_append(servers, buffer, 401, - &rc); - assert(rc == MEMCACHED_SUCCESS); - assert(x == memcached_server_list_count(servers)); - } - - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); - - memcached_server_list_free(servers); -} - -/* We don't test the behavior itself, we test the switches */ -void behavior_test(memcached_st *memc) -{ - unsigned long long value; - unsigned int set= 1; - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK); - assert(value == 1); - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY); - assert(value == 1); - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_MD5_HASHING); - assert(value == 1); - - set= 0; - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK); - assert(value == 0); - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY); - assert(value == 0); - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, &set); - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_MD5_HASHING); - assert(value == 0); - - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE); - assert(value > 0); - - value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE); - assert(value > 0); -} - -/* Test case provided by Cal Haldenbrand */ -void user_supplied_bug1(memcached_st *memc) -{ - unsigned int setter= 1; - unsigned int x; - - unsigned long long total= 0; - int size= 0; - char key[10]; - char randomstuff[6 * 1024]; - memcached_return rc; - - memset(randomstuff, 0, 6 * 1024); - - /* We just keep looking at the same values over and over */ - srandom(10); - - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter); - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter); - - - /* add key */ - for (x= 0 ; total < 20 * 1024576 ; x++ ) - { - unsigned int j= 0; - - size= (rand() % ( 5 * 1024 ) ) + 400; - memset(randomstuff, 0, 6 * 1024); - assert(size < 6 * 1024); /* Being safe here */ - - for (j= 0 ; j < size ;j++) - randomstuff[j] = (char) (rand() % 26) + 97; - - total += size; - sprintf(key, "%d", x); - rc = memcached_set(memc, key, strlen(key), - randomstuff, strlen(randomstuff), 10, 0); - /* If we fail, lets try again */ - if (rc != MEMCACHED_SUCCESS) - rc = memcached_set(memc, key, strlen(key), - randomstuff, strlen(randomstuff), 10, 0); - assert(rc == MEMCACHED_SUCCESS); - } -} - -/* Test case provided by Cal Haldenbrand */ -void user_supplied_bug2(memcached_st *memc) -{ - int errors; - unsigned int setter; - unsigned int x; - unsigned long long total; - - setter= 1; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter); - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter); -#ifdef NOT_YET - setter = 20 * 1024576; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter); - setter = 20 * 1024576; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter); - getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE); - getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE); -#endif - - for (x= 0, errors= 0, total= 0 ; total < 20 * 1024576 ; x++) - { - memcached_return rc= MEMCACHED_SUCCESS; - char buffer[SMALL_STRING_LEN]; - uint16_t flags= 0; - size_t val_len= 0; - char *getval; - - memset(buffer, 0, SMALL_STRING_LEN); - - snprintf(buffer, SMALL_STRING_LEN, "%u", x); - getval= memcached_get(memc, buffer, strlen(buffer), - &val_len, &flags, &rc); - if (rc != MEMCACHED_SUCCESS) - { - if (rc == MEMCACHED_NOTFOUND) - errors++; - else - assert(0); - - continue; - } - total+= val_len; - errors= 0; - free(getval); - } -} - -/* Do a large mget() over all the keys we think exist */ -#define KEY_COUNT 3000 // * 1024576 -void user_supplied_bug3(memcached_st *memc) -{ - memcached_return rc; - unsigned int setter; - unsigned int x; - char **keys; - size_t key_lengths[KEY_COUNT]; - - setter= 1; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter); - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter); -#ifdef NOT_YET - setter = 20 * 1024576; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter); - setter = 20 * 1024576; - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter); - getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE); - getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE); -#endif - - keys= (char **)malloc(sizeof(char *) * KEY_COUNT); - assert(keys); - memset(keys, 0, (sizeof(char *) * KEY_COUNT)); - for (x= 0; x < KEY_COUNT; x++) - { - char buffer[30]; - - snprintf(buffer, 30, "%u", x); - keys[x]= strdup(buffer); - key_lengths[x]= strlen(keys[x]); - } - - rc= memcached_mget(memc, keys, key_lengths, KEY_COUNT); - assert(rc == MEMCACHED_SUCCESS); - - /* Turn this into a help function */ - { - char return_key[MEMCACHED_MAX_KEY]; - size_t return_key_length; - char *return_value; - size_t return_value_length; - uint16_t flags; - - while ((return_value= memcached_fetch(memc, return_key, &return_key_length, - &return_value_length, &flags, &rc))) - { - assert(return_value); - assert(rc == MEMCACHED_SUCCESS); - free(return_value); - } - } - - for (x= 0; x < KEY_COUNT; x++) - free(keys[x]); - free(keys); -} - -void string_alloc_null(memcached_st *memc) -{ - memcached_string_st *string; - - string= memcached_string_create(memc, 0); - assert(string); - memcached_string_free(memc, string); -} - -void string_alloc_with_size(memcached_st *memc) -{ - memcached_string_st *string; - - string= memcached_string_create(memc, 1024); - assert(string); - memcached_string_free(memc, string); -} - -void string_alloc_with_size_toobig(memcached_st *memc) -{ - memcached_string_st *string; - - string= memcached_string_create(memc, INT64_MAX); - assert(string == NULL); -} - -void string_alloc_append(memcached_st *memc) -{ - unsigned int x; - char buffer[SMALL_STRING_LEN]; - memcached_string_st *string; - - /* Ring the bell! */ - memset(buffer, 6, SMALL_STRING_LEN); - - string= memcached_string_create(memc, 100); - assert(string); - - for (x= 0; x < 1024; x++) - { - memcached_return rc; - rc= memcached_string_append(memc, string, buffer, SMALL_STRING_LEN); - assert(rc == MEMCACHED_SUCCESS); - } - memcached_string_free(memc, string); -} - -void string_alloc_append_toobig(memcached_st *memc) -{ - memcached_return rc; - unsigned int x; - char buffer[SMALL_STRING_LEN]; - memcached_string_st *string; - - /* Ring the bell! */ - memset(buffer, 6, SMALL_STRING_LEN); - - string= memcached_string_create(memc, 100); - assert(string); - - for (x= 0; x < 1024; x++) - { - rc= memcached_string_append(memc, string, buffer, SMALL_STRING_LEN); - assert(rc == MEMCACHED_SUCCESS); - } - rc= memcached_string_append(memc, string, buffer, INT64_MAX); - assert(rc == MEMCACHED_MEMORY_ALLOCATION_FAILURE); - memcached_string_free(memc, string); -} - -void add_host_test1(memcached_st *memc) -{ - unsigned int x; - memcached_return rc; - char servername[]= "0.example.com"; - memcached_server_st *servers; - - servers= memcached_server_list_append(NULL, servername, 400, &rc); - assert(servers); - assert(1 == memcached_server_list_count(servers)); - - for (x= 2; x < 20; x++) - { - char buffer[SMALL_STRING_LEN]; - - snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x); - servers= memcached_server_list_append(servers, buffer, 401, - &rc); - assert(rc == MEMCACHED_SUCCESS); - assert(x == memcached_server_list_count(servers)); - } - - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); - - memcached_server_list_free(servers); -} - -void pre_nonblock(memcached_st *memc) -{ - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL); -} - -void pre_md5(memcached_st *memc) -{ - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, NULL); -} - -void pre_nodelay(memcached_st *memc) -{ - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL); - memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, NULL); -} - -typedef struct collection_st collection_st; -typedef struct test_st test_st; - -struct test_st { - char *name; - unsigned int requires_flush; - void (*function)(memcached_st *memc); -}; - -struct collection_st { - char *name; - void (*pre)(memcached_st *memc); - void (*post)(memcached_st *memc); - test_st *tests; -}; - int main(int argc, char *argv[]) { unsigned int x; - char *server_list; char *collection_to_run= NULL; char *wildcard= NULL; + server_startup_st *startup_ptr; memcached_server_st *servers; + world_st world; + collection_st *collection; + collection_st *next; + uint8_t failed; + void *world_ptr; + + memset(&world, 0, sizeof(world_st)); + get_world(&world); + collection= world.collections; + + if (world.create) + world_ptr= world.create(); + else + world_ptr= NULL; + startup_ptr= (server_startup_st *)world_ptr; + servers= (memcached_server_st *)startup_ptr->servers; if (argc > 1) collection_to_run= argv[1]; @@ -808,150 +56,83 @@ int main(int argc, char *argv[]) if (argc == 3) wildcard= argv[2]; - if (!(server_list= getenv("MEMCACHED_SERVERS"))) - server_list= "localhost"; - - printf("servers %s\n", server_list); - srandom(time(NULL)); - - servers= memcached_servers_parse(server_list); - assert(servers); - - for (x= 0; x < memcached_server_list_count(servers); x++) - { - printf("\t%s : %u\n", servers[x].hostname, servers[x].port); - assert(servers[x].stack_responses == 0); - assert(servers[x].fd == -1); - assert(servers[x].cursor_active == 0); - } - - printf("\n"); - - /* Clean the server before beginning testing */ - test_st tests[] ={ - {"flush", 0, flush_test }, - {"init", 0, init_test }, - {"allocation", 0, allocation_test }, - {"error", 0, error_test }, - {"set", 0, set_test }, - {"set2", 0, set_test2 }, - {"set3", 0, set_test3 }, - {"add", 0, add_test }, - {"replace", 0, replace_test }, - {"delete", 1, delete_test }, - {"get", 1, get_test }, - {"get2", 0, get_test2 }, - {"get3", 0, get_test3 }, - {"get4", 0, get_test4 }, - {"stats_servername", 0, stats_servername_test }, - {"increment", 0, increment_test }, - {"decrement", 0, decrement_test }, - {"quit", 0, quit_test }, - {"mget", 0, mget_test }, - {"get_stats", 0, get_stats }, - {"add_host_test", 0, add_host_test }, - {"get_stats_keys", 0, get_stats_keys }, - {"behavior_test", 0, get_stats_keys }, - {0, 0, 0} - }; - - test_st string_tests[] ={ - {"string alloc with null", 0, string_alloc_null }, - {"string alloc with 1K", 0, string_alloc_with_size }, - {"string alloc with malloc failure", 0, string_alloc_with_size_toobig }, - {"string append", 0, string_alloc_append }, - {"string append failure (too big)", 0, string_alloc_append_toobig }, - {0, 0, 0} - }; - - test_st user_tests[] ={ - {"user_supplied_bug1", 0, user_supplied_bug1 }, - {"user_supplied_bug2", 0, user_supplied_bug2 }, - {"user_supplied_bug3", 0, user_supplied_bug3 }, - {0, 0, 0} - }; - - - collection_st collection[] ={ - {"block", 0, 0, tests}, - {"nonblock", pre_nonblock, 0, tests}, - {"nodelay", pre_nodelay, 0, tests}, - {"md5", pre_md5, 0, tests}, - {"string", 0, 0, string_tests}, - {"user", 0, 0, user_tests}, - {0, 0, 0, 0} - }; - - collection_st *next; for (next= collection; next->name; next++) { test_st *run; run= next->tests; - - if (collection_to_run && strcmp(collection_to_run, next->name)) + if (collection_to_run && fnmatch(collection_to_run, next->name, 0)) continue; fprintf(stderr, "\n%s\n\n", next->name); for (x= 0; run->name; run++) { - if (wildcard && strcmp(wildcard, run->name)) - continue; - - fprintf(stderr, "Testing %s", run->name); - + unsigned int loop; memcached_st *memc; memcached_return rc; struct timeval start_time, end_time; + long int load_time; + + if (wildcard && fnmatch(wildcard, run->name, 0)) + continue; + + fprintf(stderr, "Testing %s", run->name); memc= memcached_create(NULL); assert(memc); - if (run->requires_flush) - memcached_flush(memc, 0); - rc= memcached_server_push(memc, servers); assert(rc == MEMCACHED_SUCCESS); - unsigned int loop; + if (run->requires_flush) + { + memcached_flush(memc, 0); + memcached_quit(memc); + } + for (loop= 0; loop < memcached_server_list_count(servers); loop++) { - assert(memc->hosts[loop].stack_responses == 0); assert(memc->hosts[loop].fd == -1); assert(memc->hosts[loop].cursor_active == 0); } if (next->pre) - next->pre(memc); + { + memcached_return rc; + rc= next->pre(memc); + + if (rc != MEMCACHED_SUCCESS) + { + fprintf(stderr, "\t\t\t\t\t [ skipping ]\n"); + goto error; + } + } gettimeofday(&start_time, NULL); - run->function(memc); + failed= run->function(memc); gettimeofday(&end_time, NULL); - long int load_time= timedif(end_time, start_time); - fprintf(stderr, "\t\t\t\t\t %ld.%03ld [ ok ]\n", load_time / 1000, - load_time % 1000); + load_time= timedif(end_time, start_time); + if (failed) + fprintf(stderr, "\t\t\t\t\t %ld.%03ld [ failed ]\n", load_time / 1000, + load_time % 1000); + else + fprintf(stderr, "\t\t\t\t\t %ld.%03ld [ ok ]\n", load_time / 1000, + load_time % 1000); if (next->post) - next->post(memc); + (void)next->post(memc); assert(memc); +error: memcached_free(memc); } } - /* Clean up whatever we might have left */ - { - memcached_st *memc; - memc= memcached_create(NULL); - assert(memc); - flush_test(memc); - memcached_free(memc); - } - fprintf(stderr, "All tests completed successfully\n\n"); - memcached_server_list_free(servers); + if (world.destroy) + world.destroy(world_ptr); return 0; }