-#include <vector>
-#include <iostream>
-#include <string>
-#include <cerrno>
-#include <cassert>
-
-#define BUILDING_LIBMEMCACHED
-// !NEVER use common.h, always use memcached.h in your own apps
-#include <libmemcached/common.h>
-#include <libmemcached/util.h>
-
-#include <tests/parser.h>
-#include <tests/print.h>
-
-enum scanner_type_t
-{
- NIL,
- UNSIGNED,
- SIGNED,
- ARRAY
-};
-
-
-struct scanner_string_st {
- const char *c_str;
- size_t size;
-};
-
-static inline scanner_string_st scanner_string(const char *arg, size_t arg_size)
-{
- scanner_string_st local= { arg, arg_size };
- return local;
-}
-
-#define make_scanner_string(X) scanner_string((X), static_cast<size_t>(sizeof(X) - 1))
-
-static struct scanner_string_st scanner_string_null= { 0, 0};
-
-struct scanner_variable_t {
- enum scanner_type_t type;
- struct scanner_string_st option;
- struct scanner_string_st result;
- test_return_t (*check_func)(memcached_st *memc, const scanner_string_st &hostname);
-};
-
-// Check and make sure the first host is what we expect it to be
-static test_return_t __check_host(memcached_st *memc, const scanner_string_st &hostname)
-{
- memcached_server_instance_st instance=
- memcached_server_instance_by_position(memc, 0);
-
- test_true(instance);
-
- const char *first_hostname = memcached_server_name(instance);
- test_true(first_hostname);
- test_strcmp(first_hostname, hostname.c_str);
-
- return TEST_SUCCESS;
-}
-
-// Check and make sure the prefix_key is what we expect it to be
-static test_return_t __check_prefix_key(memcached_st *memc, const scanner_string_st &hostname)
-{
- memcached_server_instance_st instance=
- memcached_server_instance_by_position(memc, 0);
-
- test_true(instance);
-
- const char *first_hostname = memcached_server_name(instance);
- test_true(first_hostname);
- test_strcmp(first_hostname, hostname.c_str);
-
- return TEST_SUCCESS;
-}
-
-static test_return_t __check_IO_MSG_WATERMARK(memcached_st *memc, const scanner_string_st &value)
-{
- uint64_t value_number;
-
- value_number= atoll(value.c_str);
-
- test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK) == value_number);
- return TEST_SUCCESS;
-}
-
-static test_return_t __check_REMOVE_FAILED_SERVERS(memcached_st *memc, const scanner_string_st &)
-{
- test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS));
- return TEST_SUCCESS;
-}
-
-static test_return_t __check_NOREPLY(memcached_st *memc, const scanner_string_st &)
-{
- test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NOREPLY));
- return TEST_SUCCESS;
-}
-
-static test_return_t __check_VERIFY_KEY(memcached_st *memc, const scanner_string_st &)
-{
- test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_VERIFY_KEY));
- return TEST_SUCCESS;
-}
-
-static test_return_t __check_distribution_RANDOM(memcached_st *memc, const scanner_string_st &)
-{
- test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION) == MEMCACHED_DISTRIBUTION_RANDOM);
- return TEST_SUCCESS;
-}
-
-scanner_variable_t test_server_strings[]= {
- { ARRAY, make_scanner_string("--server=localhost"), make_scanner_string("localhost"), __check_host },
- { ARRAY, make_scanner_string("--server=10.0.2.1"), make_scanner_string("10.0.2.1"), __check_host },
- { ARRAY, make_scanner_string("--server=example.com"), make_scanner_string("example.com"), __check_host },
- { ARRAY, make_scanner_string("--server=localhost:30"), make_scanner_string("localhost"), __check_host },
- { ARRAY, make_scanner_string("--server=10.0.2.1:20"), make_scanner_string("10.0.2.1"), __check_host },
- { ARRAY, make_scanner_string("--server=example.com:1024"), make_scanner_string("example.com"), __check_host },
- { NIL, scanner_string_null, scanner_string_null, NULL }
-};
-
-scanner_variable_t test_server_strings_with_weights[]= {
- { ARRAY, make_scanner_string("--server=10.0.2.1:30/?40"), make_scanner_string("10.0.2.1"), __check_host },
- { ARRAY, make_scanner_string("--server=example.com:1024/?30"), make_scanner_string("example.com"), __check_host },
- { ARRAY, make_scanner_string("--server=10.0.2.1/?20"), make_scanner_string("10.0.2.1"), __check_host },
- { ARRAY, make_scanner_string("--server=example.com/?10"), make_scanner_string("example.com"), __check_host },
- { NIL, scanner_string_null, scanner_string_null, NULL }
-};
-
-scanner_variable_t bad_test_strings[]= {
- { ARRAY, make_scanner_string("-servers=localhost:11221,localhost:11222,localhost:11223,localhost:11224,localhost:11225"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("-- servers=a.example.com:81,localhost:82,b.example.com"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--servers=localhost:+80"), scanner_string_null, NULL},
- { ARRAY, make_scanner_string("--servers=localhost.com."), scanner_string_null, NULL},
- { ARRAY, make_scanner_string("--server=localhost.com."), scanner_string_null, NULL},
- { ARRAY, make_scanner_string("--server=localhost.com.:80"), scanner_string_null, NULL},
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-scanner_variable_t test_number_options[]= {
- { ARRAY, make_scanner_string("--CONNECT-TIMEOUT=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--IO-BYTES-WATERMARK=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--IO-KEY-PREFETCH=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--IO-MSG-WATERMARK=456"), make_scanner_string("456"), __check_IO_MSG_WATERMARK },
- { ARRAY, make_scanner_string("--NUMBER-OF-REPLICAS=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--POLL-TIMEOUT=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--RCV-TIMEOUT=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--REMOVE-FAILED-SERVERS=3"), scanner_string_null, __check_REMOVE_FAILED_SERVERS },
- { ARRAY, make_scanner_string("--RETRY-TIMEOUT=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--SND-TIMEOUT=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--SOCKET-RECV-SIZE=456"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--SOCKET-SEND-SIZE=456"), scanner_string_null, NULL },
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-scanner_variable_t test_boolean_options[]= {
- { ARRAY, make_scanner_string("--BINARY-PROTOCOL"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--BUFFER-REQUESTS"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH-WITH-NAMESPACE"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--NOREPLY"), scanner_string_null, __check_NOREPLY },
- { ARRAY, make_scanner_string("--RANDOMIZE-REPLICA-READ"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--SORT-HOSTS"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--SUPPORT-CAS"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--TCP-NODELAY"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--TCP-KEEPALIVE"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--TCP-KEEPIDLE"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--USE-UDP"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--VERIFY-KEY"), scanner_string_null, __check_VERIFY_KEY },
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-scanner_variable_t prefix_key_strings[]= {
- { ARRAY, make_scanner_string("--NAMESPACE=foo"), make_scanner_string("foo"), __check_prefix_key },
- { ARRAY, make_scanner_string("--NAMESPACE=\"foo\""), make_scanner_string("foo"), __check_prefix_key },
- { ARRAY, make_scanner_string("--NAMESPACE=\"This_is_a_very_long_key\""), make_scanner_string("This_is_a_very_long_key"), __check_prefix_key },
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-scanner_variable_t distribution_strings[]= {
- { ARRAY, make_scanner_string("--DISTRIBUTION=consistent"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--DISTRIBUTION=consistent,CRC"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--DISTRIBUTION=consistent,MD5"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--DISTRIBUTION=random"), scanner_string_null, __check_distribution_RANDOM },
- { ARRAY, make_scanner_string("--DISTRIBUTION=modula"), scanner_string_null, NULL },
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-scanner_variable_t hash_strings[]= {
- { ARRAY, make_scanner_string("--HASH=CRC"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=FNV1A_32"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=FNV1A_64"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=FNV1_32"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=FNV1_64"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=JENKINS"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=MD5"), scanner_string_null, NULL },
- { ARRAY, make_scanner_string("--HASH=MURMUR"), scanner_string_null, NULL },
- { NIL, scanner_string_null, scanner_string_null, NULL}
-};
-
-
-static test_return_t _test_option(scanner_variable_t *scanner, bool test_true_opt= true)
-{
- for (scanner_variable_t *ptr= scanner; ptr->type != NIL; ptr++)
- {
- memcached_st *memc= memcached(ptr->option.c_str, ptr->option.size);
- if (test_true_opt)
- {
- char buffer[2048];
- if (not memc)
- {
- libmemcached_check_configuration(ptr->option.c_str, ptr->option.size, buffer, sizeof(buffer));
- }
-
- test_true_got(memc, buffer);
-
- if (ptr->check_func)
- {
- test_return_t test_rc= (*ptr->check_func)(memc, ptr->result);
- if (test_rc != TEST_SUCCESS)
- {
- memcached_free(memc);
- return test_rc;
- }
- }
-
- memcached_free(memc);
- }
- else
- {
- test_false_with(memc, ptr->option.c_str);
- }
- }
-
- return TEST_SUCCESS;
-}
-
-test_return_t server_test(memcached_st *)
-{
- return _test_option(test_server_strings);
-}
-
-test_return_t server_with_weight_test(memcached_st *)
-{
- return _test_option(test_server_strings_with_weights);
-}
-
-test_return_t servers_bad_test(memcached_st *)
-{
- test_return_t rc;
- if ((rc= _test_option(bad_test_strings, false)) != TEST_SUCCESS)
- {
- return rc;
- }
-
- return TEST_SUCCESS;
-}
-
-test_return_t parser_number_options_test(memcached_st*)
-{
- return _test_option(test_number_options);
-}
-
-test_return_t parser_boolean_options_test(memcached_st*)
-{
- return _test_option(test_boolean_options);
-}
-
-test_return_t behavior_parser_test(memcached_st*)
-{
- return TEST_SUCCESS;
-}
-
-test_return_t parser_hash_test(memcached_st*)
-{
- return _test_option(hash_strings);
-}
-
-test_return_t parser_distribution_test(memcached_st*)
-{
- return _test_option(distribution_strings);
-}
-
-test_return_t parser_key_prefix_test(memcached_st*)
-{
- return _test_option(distribution_strings);
-}
-
-#define SUPPORT_EXAMPLE_CNF "support/example.cnf"
-
-test_return_t memcached_create_with_options_with_filename(memcached_st*)
-{
- if (access(SUPPORT_EXAMPLE_CNF, R_OK))
- return TEST_SKIPPED;
-
- memcached_st *memc_ptr;
- memc_ptr= memcached(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""));
- test_true_got(memc_ptr, "memcached() failed");
- test_strcmp(SUPPORT_EXAMPLE_CNF, memcached_array_string(memc_ptr->configure.filename));
- memcached_free(memc_ptr);
-
- return TEST_SUCCESS;
-}
-
-test_return_t libmemcached_check_configuration_with_filename_test(memcached_st*)
-{
- if (access(SUPPORT_EXAMPLE_CNF, R_OK))
- return TEST_SKIPPED;
-
- memcached_return_t rc;
- char buffer[BUFSIZ];
-
- rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""), buffer, sizeof(buffer));
- test_true_got(rc == MEMCACHED_SUCCESS, (rc == MEMCACHED_ERRNO) ? strerror(errno) : memcached_strerror(NULL, rc));
-
- rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=support/example.cnf"), buffer, sizeof(buffer));
- test_false_with(rc == MEMCACHED_SUCCESS, memcached_strerror(NULL, rc));
-
- rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"bad-path/example.cnf\""), buffer, sizeof(buffer));
- test_true_got(rc == MEMCACHED_ERRNO, memcached_strerror(NULL, rc));
-
- return TEST_SUCCESS;
-}
-
-test_return_t libmemcached_check_configuration_test(memcached_st*)
-{
- memcached_return_t rc;
- char buffer[BUFSIZ];
-
- rc= libmemcached_check_configuration(test_literal_param("--server=localhost"), buffer, sizeof(buffer));
- test_true_got(rc == MEMCACHED_SUCCESS, buffer);
-
- rc= libmemcached_check_configuration(test_literal_param("--dude=localhost"), buffer, sizeof(buffer));
- test_false_with(rc == MEMCACHED_SUCCESS, buffer);
- test_true(rc == MEMCACHED_PARSE_ERROR);
-
- return TEST_SUCCESS;
-}
-
-test_return_t memcached_create_with_options_test(memcached_st*)
-{
- memcached_st *memc_ptr;
- memc_ptr= memcached(test_literal_param("--server=localhost"));
- test_true_got(memc_ptr, memcached_last_error_message(memc_ptr));
- memcached_free(memc_ptr);
-
- memc_ptr= memcached(test_literal_param("--dude=localhost"));
- test_false_with(memc_ptr, memcached_last_error_message(memc_ptr));
-
- return TEST_SUCCESS;
-}
-
-test_return_t test_include_keyword(memcached_st*)
-{
- if (access(SUPPORT_EXAMPLE_CNF, R_OK))
- return TEST_SKIPPED;
-
- char buffer[BUFSIZ];
- memcached_return_t rc;
- rc= libmemcached_check_configuration(test_literal_param("INCLUDE \"support/example.cnf\""), buffer, sizeof(buffer));
- test_true_got(rc == MEMCACHED_SUCCESS, buffer);