X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=tests%2Ftest.c;h=793bcbd7264d08af68112aefa033e4e931af0980;hb=7c986323bd4eece0d805f4df17eb03ea094f84f6;hp=7a3b86291e354bdeb8f13c4fe3cfdf94736404b8;hpb=cf34e0d38a58e81ec3993af0db6c221585d712e5;p=m6w6%2Flibmemcached diff --git a/tests/test.c b/tests/test.c index 7a3b8629..793bcbd7 100644 --- a/tests/test.c +++ b/tests/test.c @@ -1,709 +1,244 @@ +/* uTest + * Copyright (C) 2006-2009 Brian Aker + * All rights reserved. + * + * Use and distribution licensed under the BSD license. See + * the COPYING file in the parent directory for full text. + */ + /* Sample test application. */ #include -#include -#include #include #include - -void init_test(void) -{ - memcached_st memc; - - (void)memcached_create(&memc); - memcached_free(&memc); -} - -void allocation_test(void) -{ - memcached_st *memc; - memc= memcached_create(NULL); - assert(memc); - memcached_free(memc); -} - -void connection_test(void) -{ - memcached_return rc; - memcached_st *memc; - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - assert(memc); - memcached_free(memc); -} - -void error_test(void) -{ - memcached_st *memc; - memc= memcached_create(NULL); - memcached_return rc; - - for (rc= MEMCACHED_SUCCESS; rc < MEMCACHED_MAXIMUM_RETURN; rc++) - { - printf("Error %d -> %s\n", rc, memcached_strerror(memc, rc)); +#include +#include +#include +#include +#include +#include +#include +#include "server.h" + +#include "test.h" + +static void world_stats_print(world_stats_st *stats) +{ + fprintf(stderr, "Total\t\t\t\t%u\n", stats->total); + fprintf(stderr, "\tFailed\t\t\t%u\n", stats->failed); + fprintf(stderr, "\tSkipped\t\t\t%u\n", stats->skipped); + fprintf(stderr, "\tSucceeded\t\t%u\n", stats->success); +} + +static long int timedif(struct timeval a, struct timeval b) +{ + register int us, s; + + us = (int)(a.tv_usec - b.tv_usec); + us /= 1000; + s = (int)(a.tv_sec - b.tv_sec); + s *= 1000; + return s + us; +} + +static const char *test_strerror(test_return_t code) +{ + switch (code) { + case TEST_SUCCESS: + return "ok"; + case TEST_FAILURE: + return "failed"; + case TEST_MEMORY_ALLOCATION_FAILURE: + return "memory allocation"; + case TEST_SKIPPED: + return "skipped"; + case TEST_MAXIMUM_RETURN: + default: + fprintf(stderr, "Unknown return value\n"); + abort(); } - assert(memc); - memcached_free(memc); -} - -void set_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_set(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - memcached_free(memc); -} - -void add_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_add(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_NOTSTORED); - - memcached_free(memc); -} - -void replace_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_replace(memc, key, strlen(key), - value, strlen(value), - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); - - memcached_free(memc); -} - -void delete_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - 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); - - memcached_free(memc); -} - -void flush_test(void) -{ - memcached_st *memc; - memcached_return rc; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_flush(memc, 0); - assert(rc == MEMCACHED_SUCCESS); - - memcached_free(memc); -} - -void get_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *string; - size_t string_length; - uint16_t flags; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - string= memcached_get(memc, key, strlen(key), - &string_length, &flags, &rc); - - assert(rc == MEMCACHED_NOTFOUND); - assert(string_length == 0); - assert(!string); - - memcached_free(memc); -} - -void get_test2(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "when we sanitize"; - char *string; - size_t string_length; - uint16_t flags; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); } -void set_test2(void) +void create_core(void) { - memcached_st *memc; - memcached_return rc; - char *key= "foo"; - char *value= "train in the brain"; - size_t value_length= strlen(value); - unsigned int x; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - for (x= 0; x < 10; x++) + if (getenv("LIBMEMCACHED_NO_COREDUMP") == NULL) { - rc= memcached_set(memc, key, strlen(key), - value, value_length, - (time_t)0, (uint16_t)0); - assert(rc == MEMCACHED_SUCCESS); + pid_t pid= fork(); + + if (pid == 0) + { + abort(); + } + else + { + while (waitpid(pid, NULL, 0) != pid) + { + ; + } + } } - - memcached_free(memc); } -void set_test3(void) -{ - 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); - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); -} - -void get_test3(void) +static test_return_t _runner_default(test_callback_fn func, void *p) { - 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); - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); -} - -void get_test4(void) -{ - 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); - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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++) + if (func) { - 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); + return func(p); } - - free(value); - - memcached_free(memc); -} - -void stats_servername_test(void) -{ - memcached_return rc; - memcached_stat_st stat; - rc= memcached_stat_servername(&stat, NULL, - "localhost", - MEMCACHED_DEFAULT_PORT); -} - -void increment_test(void) -{ - memcached_st *memc; - unsigned int new_number; - memcached_return rc; - char *key= "number"; - char *value= "0"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); -} - -void decrement_test(void) -{ - memcached_st *memc; - unsigned int new_number; - memcached_return rc; - char *key= "number"; - char *value= "3"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); -} - -void quit_test(void) -{ - memcached_st *memc; - memcached_return rc; - char *key= "fudge"; - char *value= "sanford and sun"; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - - memcached_free(memc); -} - -void mget_test(void) -{ - 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; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - /* 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) + else { - assert(return_value); + return TEST_SUCCESS; } - 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(key_length[x] == return_value_length); - assert(!memcmp(return_value, keys[x], return_value_length)); - free(return_value); - x++; - } - - memcached_free(memc); } -void get_stats_keys(void) -{ - char **list; - char **ptr; - memcached_st *memc; - memcached_stat_st stat; - memcached_return rc; - - memc= memcached_create(NULL); - assert(memc); - - list= memcached_stat_get_keys(memc, &stat, &rc); - assert(rc == MEMCACHED_SUCCESS); - for (ptr= list; *ptr; ptr++) - printf("Found key %s\n", *ptr); - - free(list); - memcached_free(memc); -} - -void get_stats(void) -{ - unsigned int x; - char **list; - char **ptr; - memcached_return rc; - memcached_st *memc; - memcached_stat_st *stat; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - - 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); - memcached_free(memc); -} +static world_runner_st defualt_runners= { + _runner_default, + _runner_default, + _runner_default +}; -void get_stats_multiple(void) -{ - unsigned int x; - char **list; - char **ptr; - memcached_return rc; - memcached_st *memc; - memcached_stat_st *stat; - - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_server_add(memc, "localhost", 5555); - assert(rc == MEMCACHED_SUCCESS); - - 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); - memcached_free(memc); -} -void add_host_test(void) +int main(int argc, char *argv[]) { + test_return_t return_code; unsigned int x; - memcached_st *memc; - memcached_server_st *servers; - memcached_return rc; - char servername[]= "0.example.com"; + char *collection_to_run= NULL; + char *wildcard= NULL; + world_st world; + collection_st *collection; + collection_st *next; + void *world_ptr; - memc= memcached_create(NULL); - assert(memc); - rc= memcached_server_add(memc, "localhost", 0); - assert(rc == MEMCACHED_SUCCESS); + world_stats_st stats; - servers= memcached_server_list_append(NULL, servername, 400, &rc); - assert(servers); - assert(1 == memcached_server_list_count(servers)); + memset(&stats, 0, sizeof(stats)); + memset(&world, 0, sizeof(world)); + get_world(&world); - for (x= 2; x < 20; x++) + if (! world.runner) { - 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)); + world.runner= &defualt_runners; } - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); - rc= memcached_server_push(memc, servers); - assert(rc == MEMCACHED_SUCCESS); + collection= world.collections; - memcached_server_list_free(servers); - memcached_free(memc); -} + if (world.create) + world_ptr= world.create(); + else + world_ptr= NULL; -void add_host_test1(void) -{ - unsigned int x; - memcached_st *memc; - memcached_server_st *servers; - memcached_return rc; - char servername[]= "0.example.com"; - - memc= memcached_create(NULL); - assert(memc); + if (argc > 1) + collection_to_run= argv[1]; - servers= memcached_server_list_append(NULL, servername, 400, &rc); - assert(servers); - assert(1 == memcached_server_list_count(servers)); + if (argc == 3) + wildcard= argv[2]; - for (x= 2; x < 20; x++) + for (next= collection; next->name; next++) { - 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)); + test_st *run; + + run= next->tests; + 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++) + { + struct timeval start_time, end_time; + long int load_time= 0; + + if (wildcard && fnmatch(wildcard, run->name, 0)) + continue; + + fprintf(stderr, "Testing %s", run->name); + + if (world.collection_startup) + { + world.collection_startup(world_ptr); + } + + if (run->requires_flush && world.flush) + { + world.flush(world_ptr); + } + + if (world.pre_run) + { + world.pre_run(world_ptr); + } + + + if (next->pre) + { + return_code= world.runner->pre(next->pre, world_ptr); + + if (return_code != TEST_SUCCESS) + { + goto error; + } + } + + gettimeofday(&start_time, NULL); + return_code= world.runner->run(run->test_fn, world_ptr); + gettimeofday(&end_time, NULL); + load_time= timedif(end_time, start_time); + + if (next->post) + { + (void) world.runner->pre(next->pre, world_ptr); + } + + if (world.post_run) + { + world.post_run(world_ptr); + } + +error: + stats.total++; + + fprintf(stderr, "\t\t\t\t\t"); + + switch (return_code) + { + case TEST_SUCCESS: + fprintf(stderr, "%ld.%03ld ", load_time / 1000, load_time % 1000); + stats.success++; + break; + case TEST_FAILURE: + stats.failed++; + break; + case TEST_SKIPPED: + stats.skipped++; + break; + case TEST_MEMORY_ALLOCATION_FAILURE: + case TEST_MAXIMUM_RETURN: + default: + break; + } + + fprintf(stderr, "[ %s ]\n", test_strerror(return_code)); + + if (world.on_error) + { + test_return_t rc; + rc= world.on_error(return_code, world_ptr); + + if (rc != TEST_SUCCESS) + break; + } + } } - 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); - memcached_free(memc); -} + fprintf(stderr, "All tests completed successfully\n\n"); + if (world.destroy) + world.destroy(world_ptr); -int main(int argc, char *argv[]) -{ - /* Clean the server before beginning testing */ - flush_test(); - init_test(); - allocation_test(); - connection_test(); - error_test(); - set_test(); - set_test2(); - set_test3(); - add_test(); - replace_test(); - flush_test(); - delete_test(); - flush_test(); - get_test(); - get_test2(); - get_test3(); - get_test4(); - stats_servername_test(); - - increment_test(); - decrement_test(); - quit_test(); - mget_test(); - get_stats(); - add_host_test(); - - /* The multiple tests */ - if (argc == 2) - { - get_stats_multiple(); - } - get_stats_keys(); + world_stats_print(&stats); - /* Clean up whatever we might have left */ - flush_test(); - return 0; + return stats.failed == 0 ? 0 : 1; }