/* Tests census noop stubs in a simulated rpc flow */
void test_census_stubs(void) {
census_op_id op_id;
census_rpc_stats* stats = census_rpc_stats_create_empty();
census_aggregated_rpc_stats data_map = {0, NULL};
/* Initializes census library at server start up time. */
census_init();
/* Starts tracing at the beginning of a rpc. */
op_id = census_tracing_start_op();
/* Appends custom annotations on a trace object. */
census_tracing_print(op_id, "annotation foo");
census_tracing_print(op_id, "annotation bar");
/* Appends method tag on the trace object. */
census_add_method_tag(op_id, "service_foo/method.bar");
/* Either record client side stats or server side stats associated with the
op_id. Here for testing purpose, we record both. */
census_record_rpc_client_stats(op_id, stats);
census_record_rpc_server_stats(op_id, stats);
/* Ends a tracing. */
census_tracing_end_op(op_id);
/* In process stats queries. */
census_get_server_stats(&data_map);
census_aggregated_rpc_stats_set_empty(&data_map);
census_get_client_stats(&data_map);
census_aggregated_rpc_stats_set_empty(&data_map);
gpr_free(stats);
census_shutdown();
}
static void test_create_and_destroy(void) {
census_rpc_stats *stats = NULL;
census_aggregated_rpc_stats agg_stats = {0, NULL};
stats = census_rpc_stats_create_empty();
GPR_ASSERT(stats != NULL);
GPR_ASSERT(stats->cnt == 0 && stats->rpc_error_cnt == 0 &&
stats->app_error_cnt == 0 && stats->elapsed_time_ms == 0.0 &&
stats->api_request_bytes == 0 && stats->wire_request_bytes == 0 &&
stats->api_response_bytes == 0 && stats->wire_response_bytes == 0);
gpr_free(stats);
census_aggregated_rpc_stats_set_empty(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 0);
GPR_ASSERT(agg_stats.stats == NULL);
agg_stats.num_entries = 1;
agg_stats.stats = (census_per_method_rpc_stats *)gpr_malloc(
sizeof(census_per_method_rpc_stats));
agg_stats.stats[0].method = gpr_strdup("foo");
census_aggregated_rpc_stats_set_empty(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 0);
GPR_ASSERT(agg_stats.stats == NULL);
}
static void test_record_stats_on_unknown_op_id(void) {
census_op_id unknown_id = {0xDEAD, 0xBEEF};
census_rpc_stats stats = {1, 2, 3, 4, 5.1, 6.2, 7.3, 8.4};
census_aggregated_rpc_stats agg_stats = {0, NULL};
census_init();
/* Tests that recording stats against unknown id is noop. */
census_record_rpc_client_stats(unknown_id, &stats);
census_record_rpc_server_stats(unknown_id, &stats);
census_get_server_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 0);
GPR_ASSERT(agg_stats.stats == NULL);
census_get_client_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 0);
GPR_ASSERT(agg_stats.stats == NULL);
census_aggregated_rpc_stats_set_empty(&agg_stats);
census_shutdown();
}
/* Get stats from input stats store */
static void get_stats(census_ht *store, census_aggregated_rpc_stats *data) {
GPR_ASSERT(data != NULL);
if (data->num_entries != 0) {
census_aggregated_rpc_stats_set_empty(data);
}
gpr_mu_lock(&g_mu);
if (store != NULL) {
size_t n;
unsigned i, j;
gpr_timespec now = gpr_now(GPR_CLOCK_REALTIME);
census_ht_kv *kv = census_ht_get_all_elements(store, &n);
if (kv != NULL) {
data->num_entries = n;
data->stats =
(per_method_stats *)gpr_malloc(sizeof(per_method_stats) * n);
for (i = 0; i < n; i++) {
census_window_stats_sums sums[NUM_INTERVALS];
for (j = 0; j < NUM_INTERVALS; j++) {
sums[j].statistic = (void *)census_rpc_stats_create_empty();
}
data->stats[i].method = gpr_strdup(kv[i].k.ptr);
census_window_stats_get_sums(kv[i].v, now, sums);
data->stats[i].minute_stats =
*(census_rpc_stats *)sums[MINUTE_INTERVAL].statistic;
data->stats[i].hour_stats =
*(census_rpc_stats *)sums[HOUR_INTERVAL].statistic;
data->stats[i].total_stats =
*(census_rpc_stats *)sums[TOTAL_INTERVAL].statistic;
for (j = 0; j < NUM_INTERVALS; j++) {
gpr_free(sums[j].statistic);
}
}
gpr_free(kv);
}
}
gpr_mu_unlock(&g_mu);
}
static void test_record_and_get_stats(void) {
census_rpc_stats stats = {1, 2, 3, 4, 5.1, 6.2, 7.3, 8.4};
census_op_id id;
census_aggregated_rpc_stats agg_stats = {0, NULL};
/* Record client stats twice with the same op_id. */
census_init();
id = census_tracing_start_op();
census_add_method_tag(id, "m1");
census_record_rpc_client_stats(id, &stats);
census_record_rpc_client_stats(id, &stats);
census_tracing_end_op(id);
/* Server stats expect to be empty */
census_get_server_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 0);
GPR_ASSERT(agg_stats.stats == NULL);
/* Client stats expect to have one entry */
census_get_client_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 1);
GPR_ASSERT(agg_stats.stats != NULL);
GPR_ASSERT(strcmp(agg_stats.stats[0].method, "m1") == 0);
GPR_ASSERT(agg_stats.stats[0].minute_stats.cnt == 2 &&
agg_stats.stats[0].hour_stats.cnt == 2 &&
agg_stats.stats[0].total_stats.cnt == 2);
ASSERT_NEAR(agg_stats.stats[0].minute_stats.wire_response_bytes, 16.8);
ASSERT_NEAR(agg_stats.stats[0].hour_stats.wire_response_bytes, 16.8);
ASSERT_NEAR(agg_stats.stats[0].total_stats.wire_response_bytes, 16.8);
/* Get stats again, results should be the same. */
census_get_client_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 1);
census_aggregated_rpc_stats_set_empty(&agg_stats);
census_shutdown();
/* Record both server (once) and client (twice) stats with different op_ids.
*/
census_init();
id = census_tracing_start_op();
census_add_method_tag(id, "m2");
census_record_rpc_client_stats(id, &stats);
census_tracing_end_op(id);
id = census_tracing_start_op();
census_add_method_tag(id, "m3");
census_record_rpc_server_stats(id, &stats);
census_tracing_end_op(id);
id = census_tracing_start_op();
census_add_method_tag(id, "m4");
census_record_rpc_client_stats(id, &stats);
census_tracing_end_op(id);
/* Check server stats */
census_get_server_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 1);
GPR_ASSERT(strcmp(agg_stats.stats[0].method, "m3") == 0);
GPR_ASSERT(agg_stats.stats[0].minute_stats.app_error_cnt == 3 &&
agg_stats.stats[0].hour_stats.app_error_cnt == 3 &&
agg_stats.stats[0].total_stats.app_error_cnt == 3);
census_aggregated_rpc_stats_set_empty(&agg_stats);
/* Check client stats */
census_get_client_stats(&agg_stats);
GPR_ASSERT(agg_stats.num_entries == 2);
GPR_ASSERT(agg_stats.stats != NULL);
GPR_ASSERT((strcmp(agg_stats.stats[0].method, "m2") == 0 &&
strcmp(agg_stats.stats[1].method, "m4") == 0) ||
(strcmp(agg_stats.stats[0].method, "m4") == 0 &&
strcmp(agg_stats.stats[1].method, "m2") == 0));
GPR_ASSERT(agg_stats.stats[0].minute_stats.cnt == 1 &&
agg_stats.stats[1].minute_stats.cnt == 1);
census_aggregated_rpc_stats_set_empty(&agg_stats);
census_shutdown();
}
