|
|
|
@ -33,11 +33,28 @@ |
|
|
|
|
|
|
|
|
|
#include <string.h> |
|
|
|
|
|
|
|
|
|
#include <grpc/support/alloc.h> |
|
|
|
|
#include <grpc/support/log.h> |
|
|
|
|
#include <grpc/support/string_util.h> |
|
|
|
|
|
|
|
|
|
#include "src/core/ext/client_config/resolver_registry.h" |
|
|
|
|
#include "src/core/ext/client_config/resolver_result.h" |
|
|
|
|
|
|
|
|
|
#include "test/core/util/test_config.h" |
|
|
|
|
|
|
|
|
|
typedef struct on_resolution_arg { |
|
|
|
|
char *expected_server_name; |
|
|
|
|
grpc_resolver_result *resolver_result; |
|
|
|
|
} on_resolution_arg; |
|
|
|
|
|
|
|
|
|
void on_resolution_cb(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
|
|
|
|
on_resolution_arg *res = arg; |
|
|
|
|
const char *server_name = |
|
|
|
|
grpc_resolver_result_get_server_name(res->resolver_result); |
|
|
|
|
GPR_ASSERT(strcmp(res->expected_server_name, server_name) == 0); |
|
|
|
|
grpc_resolver_result_unref(exec_ctx, res->resolver_result); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static void test_succeeds(grpc_resolver_factory *factory, const char *string) { |
|
|
|
|
grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
|
|
|
|
grpc_uri *uri = grpc_uri_parse(string, 0); |
|
|
|
@ -50,9 +67,19 @@ static void test_succeeds(grpc_resolver_factory *factory, const char *string) { |
|
|
|
|
args.uri = uri; |
|
|
|
|
resolver = grpc_resolver_factory_create_resolver(factory, &args); |
|
|
|
|
GPR_ASSERT(resolver != NULL); |
|
|
|
|
|
|
|
|
|
on_resolution_arg on_res_arg; |
|
|
|
|
memset(&on_res_arg, 0, sizeof(on_res_arg)); |
|
|
|
|
on_res_arg.expected_server_name = gpr_strdup(uri->path); |
|
|
|
|
grpc_closure *on_resolution = |
|
|
|
|
grpc_closure_create(on_resolution_cb, &on_res_arg); |
|
|
|
|
|
|
|
|
|
grpc_resolver_next(&exec_ctx, resolver, &on_res_arg.resolver_result, |
|
|
|
|
on_resolution); |
|
|
|
|
GRPC_RESOLVER_UNREF(&exec_ctx, resolver, "test_succeeds"); |
|
|
|
|
grpc_uri_destroy(uri); |
|
|
|
|
grpc_exec_ctx_finish(&exec_ctx); |
|
|
|
|
gpr_free(on_res_arg.expected_server_name); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static void test_fails(grpc_resolver_factory *factory, const char *string) { |
|
|
|
|