From 7ab91f09b127b67e726c03a7f4384f0970096977 Mon Sep 17 00:00:00 2001 From: Brian Aker Date: Tue, 5 Feb 2008 08:10:52 -0800 Subject: [PATCH] Committing new test (framekwork). --- tests/Makefile.am | 9 ++- tests/atomsmasher.c | 191 ++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 197 insertions(+), 3 deletions(-) create mode 100644 tests/atomsmasher.c diff --git a/tests/Makefile.am b/tests/Makefile.am index 5e50f097..04c1c8f8 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -16,16 +16,19 @@ EXTRA_DIST = output.res output2.res\ LIBS = noinst_HEADERS = test.h server.h -noinst_PROGRAMS = testapp testplus udptest +noinst_PROGRAMS = testapp testplus udptest atomsmasher testapp_SOURCES = test.c function.c ../src/generator.c ../src/execute.c server.c testapp_LDADD = $(LDADDS) testplus_SOURCES = test.c plus.cpp server.c -testplus_LDADD = $(LDADDS) ../lib/libmemcached.la +testplus_LDADD = $(LDADDS) udptest_SOURCES = test.c udp.c server.c -udptest_LDADD = $(LDADDS) ../lib/libmemcached.la +udptest_LDADD = $(LDADDS) + +atomsmasher_SOURCES = test.c atomsmasher.c server.c ../src/generator.c ../src/execute.c +atomsmasher_LDADD = $(LDADDS) record: ./testapp > output.res diff --git a/tests/atomsmasher.c b/tests/atomsmasher.c new file mode 100644 index 00000000..c267fcee --- /dev/null +++ b/tests/atomsmasher.c @@ -0,0 +1,191 @@ +/* + Sample test application. +*/ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "server.h" +#include "../lib/common.h" +#include "../src/generator.h" +#include "../src/execute.h" + +#ifndef INT64_MAX +#define INT64_MAX LONG_MAX +#endif +#ifndef INT32_MAX +#define INT32_MAX INT_MAX +#endif + + +#include "test.h" + +#define GLOBAL_COUNT 100000 +#define TEST_COUNTER 100000 +static uint32_t global_count; + +static pairs_st *global_pairs; +static char *global_keys[GLOBAL_COUNT]; +static size_t global_keys_length[GLOBAL_COUNT]; + +uint8_t cleanup_pairs(memcached_st *memc) +{ + pairs_free(global_pairs); + + return 0; +} + +uint8_t generate_pairs(memcached_st *memc) +{ + unsigned long long x; + global_pairs= pairs_generate(GLOBAL_COUNT, 400); + global_count= GLOBAL_COUNT; + + for (x= 0; x < global_count; x++) + { + global_keys[x]= global_pairs[x].key; + global_keys_length[x]= global_pairs[x].key_length; + } + + return 0; +} + +uint8_t drizzle(memcached_st *memc) +{ + unsigned int x; + memcached_return rc; + + { + char *return_value; + size_t return_value_length; + uint32_t flags; + + for (x= 0; x < TEST_COUNTER; x++) + { + uint32_t test_bit; + uint8_t which; + + test_bit= random() % GLOBAL_COUNT; + which= random() % 2; + + if (which == 0) + { + return_value= memcached_get(memc, global_keys[test_bit], global_keys_length[test_bit], + &return_value_length, &flags, &rc); + if (rc == MEMCACHED_SUCCESS && return_value) + free(return_value); + else + WATCHPOINT_ERROR(rc); + } + else + { + rc= memcached_set(memc, global_pairs[test_bit].key, + global_pairs[test_bit].key_length, + global_pairs[test_bit].value, + global_pairs[test_bit].value_length, + 0, 0); + if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_BUFFERED) + { + WATCHPOINT_ERROR(rc); + WATCHPOINT_ASSERT(0); + } + } + } + } + + return 0; +} + +memcached_return pre_nonblock(memcached_st *memc) +{ + memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL); + + return MEMCACHED_SUCCESS; +} + +memcached_return pre_md5(memcached_st *memc) +{ + memcached_hash value= MEMCACHED_HASH_MD5; + memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value); + + return MEMCACHED_SUCCESS; +} + +memcached_return pre_hsieh(memcached_st *memc) +{ + memcached_hash value= MEMCACHED_HASH_HSIEH; + memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value); + + return MEMCACHED_SUCCESS; +} + +memcached_return enable_consistent(memcached_st *memc) +{ + memcached_server_distribution value= MEMCACHED_DISTRIBUTION_CONSISTENT; + memcached_hash hash; + memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, &value); + pre_hsieh(memc); + + value= (memcached_server_distribution)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION); + assert(value == MEMCACHED_DISTRIBUTION_CONSISTENT); + + hash= (memcached_hash)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH); + assert(hash == MEMCACHED_HASH_HSIEH); + + + return MEMCACHED_SUCCESS; +} + +test_st generate_tests[] ={ + {"generate_pairs", 1, generate_pairs }, + {"cleanup", 1, cleanup_pairs }, + {0, 0, 0} +}; + + +collection_st collection[] ={ + {"generate", 0, 0, generate_tests}, + {"generate_hsieh", pre_hsieh, 0, generate_tests}, + {"generate_hsieh_consistent", enable_consistent, 0, generate_tests}, + {"generate_md5", pre_md5, 0, generate_tests}, + {"generate_nonblock", pre_nonblock, 0, generate_tests}, + {0, 0, 0, 0} +}; + +#define SERVERS_TO_CREATE 5 + +void *world_create(void) +{ + server_startup_st *construct; + + construct= (server_startup_st *)malloc(sizeof(server_startup_st)); + memset(construct, 0, sizeof(server_startup_st)); + construct->count= SERVERS_TO_CREATE; + construct->udp= 0; + server_startup(construct); + + return construct; +} + +void world_destroy(void *p) +{ + server_startup_st *construct= (server_startup_st *)p; + memcached_server_st *servers= (memcached_server_st *)construct->servers; + memcached_server_list_free(servers); + + server_shutdown(construct); + free(construct); +} + +void get_world(world_st *world) +{ + world->collections= collection; + world->create= world_create; + world->destroy= world_destroy; +} -- 2.30.2