X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=tests%2Ftest.c;h=fb501fadc2edcc6d57270c2cd47ce52cdc21a5d8;hb=51d1a5f37c90faa416a5fbd9238aafbde718b074;hp=37ed800bce55415cefe62927c74daad19c80c88a;hpb=44152c2062812b74edbcad6b7bf15e78d93ed12c;p=awesomized%2Flibmemcached diff --git a/tests/test.c b/tests/test.c index 37ed800b..fb501fad 100644 --- a/tests/test.c +++ b/tests/test.c @@ -1,649 +1,258 @@ +/* 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 +#include +#include +#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)); - } - - 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); -} +#include "test.h" -void delete_test(void) +static void world_stats_print(world_stats_st *stats) { - 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); + 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); } -void flush_test(void) +static long int timedif(struct timeval a, struct timeval b) { - 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); + long us, s; - memcached_free(memc); + us = (int)(a.tv_usec - b.tv_usec); + us /= 1000; + s = (int)(a.tv_sec - b.tv_sec); + s *= 1000; + return s + us; } -void get_test(void) +const char *test_strerror(test_return_t code) { - 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 get_test3(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); - - 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); + 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(); + } } -void get_test4(void) +void create_core(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 (getenv("LIBMEMCACHED_NO_COREDUMP") == NULL) { - 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); + pid_t pid= fork(); + + if (pid == 0) + { + abort(); + } + else + { + while (waitpid(pid, NULL, 0) != pid) + { + ; + } + } } - - 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) +static test_return_t _runner_default(test_callback_fn func, void *p) { - 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) + if (func) { - assert(return_value); + return func(p); } - 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))) + else { - 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++; + return TEST_SUCCESS; } - - 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); -} +static world_runner_st defualt_runners= { + _runner_default, + _runner_default, + _runner_default +}; -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); -} -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); -} - -void add_host_test1(void) -{ - unsigned int x; - memcached_st *memc; - memcached_server_st *servers; - memcached_return rc; - char servername[]= "0.example.com"; + if (world.create) + { + test_return_t error; + world_ptr= world.create(&error); + if (error != TEST_SUCCESS) + exit(1); + } + else + { + world_ptr= NULL; + } - 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 && world.runner->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 && world.runner->post) + { + (void) world.runner->post(next->post, 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"); -int main(int argc, char *argv[]) -{ - /* Clean the server before beginning testing */ - flush_test(); - init_test(); - allocation_test(); - connection_test(); - error_test(); - set_test(); - 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) + if (world.destroy) { - get_stats_multiple(); + test_return_t error; + error= world.destroy(world_ptr); + + if (error != TEST_SUCCESS) + { + fprintf(stderr, "Failure during shutdown.\n"); + stats.failed++; // We do this to make our exit code return 1 + } } - get_stats_keys(); - /* Clean up whatever we might have left */ - flush_test(); - return 0; + world_stats_print(&stats); + + return stats.failed == 0 ? 0 : 1; }