/* Ensure all possible state transitions are called without causing problem */ static void test_init_shutdown(void) { census_tracing_init(); census_tracing_init(); census_tracing_shutdown(); census_tracing_shutdown(); census_tracing_init(); }
static void test_add_method_tag_to_unknown_op_id(void) { census_op_id unknown_id = {0xDEAD, 0xBEEF}; int ret = 0; census_tracing_init(); ret = census_add_method_tag(unknown_id, "foo"); GPR_ASSERT(ret != 0); census_tracing_shutdown(); }
static void test_get_trace_method_name(void) { census_op_id id; const char write_name[] = "service/method"; census_tracing_init(); id = census_tracing_start_op(); census_add_method_tag(id, write_name); census_internal_lock_trace_store(); { const char* read_name = census_get_trace_method_name(census_get_trace_obj_locked(id)); GPR_ASSERT(strcmp(read_name, write_name) == 0); } census_internal_unlock_trace_store(); census_tracing_shutdown(); }
/* Test that record stats is noop when trace store is uninitialized. */ static void test_record_stats_with_trace_store_uninitialized(void) { census_rpc_stats stats = {1, 2, 3, 4, 5.1, 6.2, 7.3, 8.4}; census_op_id id = {0, 0}; census_aggregated_rpc_stats agg_stats = {0, NULL}; census_init(); id = census_tracing_start_op(); census_add_method_tag(id, "m"); census_tracing_end_op(id); /* shuts down trace store only. */ census_tracing_shutdown(); census_record_rpc_client_stats(id, &stats); census_get_client_stats(&agg_stats); GPR_ASSERT(agg_stats.num_entries == 0); census_stats_store_shutdown(); }
static void test_trace_print(void) { census_op_id id; int i; const char* annotation_txt[4] = {"abc", "", "$%^ *()_"}; char long_txt[CENSUS_MAX_ANNOTATION_LENGTH + 10]; memset(long_txt, 'a', GPR_ARRAY_SIZE(long_txt)); long_txt[CENSUS_MAX_ANNOTATION_LENGTH + 9] = '\0'; annotation_txt[3] = long_txt; census_tracing_init(); id = census_tracing_start_op(); /* Adds large number of annotations to each trace */ for (i = 0; i < 1000; i++) { census_tracing_print(id, annotation_txt[i % GPR_ARRAY_SIZE(annotation_txt)]); } census_tracing_end_op(id); census_tracing_shutdown(); }
static void test_concurrency(void) { #define NUM_THREADS 1000 gpr_thd_id tid[NUM_THREADS]; int i = 0; thd_arg arg; arg.num_done = 0; gpr_mu_init(&arg.mu); gpr_cv_init(&arg.done); census_tracing_init(); for (i = 0; i < NUM_THREADS; ++i) { gpr_thd_new(tid + i, mimic_trace_op_sequences, &arg, NULL); } gpr_mu_lock(&arg.mu); while (arg.num_done < NUM_THREADS) { gpr_log(GPR_INFO, "num done %d", arg.num_done); gpr_cv_wait(&arg.done, &arg.mu, gpr_inf_future); } gpr_mu_unlock(&arg.mu); census_tracing_shutdown(); #undef NUM_THREADS }
static void test_get_active_ops(void) { census_op_id id_1, id_2, id_3; census_trace_obj** active_ops; const char* annotation_txt[] = {"annotation 1", "a2"}; int i = 0; int n = 0; gpr_log(GPR_INFO, "test_get_active_ops"); census_tracing_init(); /* No active ops before calling start_op(). */ active_ops = census_get_active_ops(&n); GPR_ASSERT(active_ops == NULL); GPR_ASSERT(n == 0); /* Starts one op */ id_1 = census_tracing_start_op(); census_add_method_tag(id_1, "foo_1"); active_ops = census_get_active_ops(&n); GPR_ASSERT(active_ops != NULL); GPR_ASSERT(n == 1); GPR_ASSERT(ids_equal(active_ops[0]->id, id_1)); census_trace_obj_destroy(active_ops[0]); gpr_free(active_ops); active_ops = NULL; /* Start the second and the third ops */ id_2 = census_tracing_start_op(); census_add_method_tag(id_2, "foo_2"); id_3 = census_tracing_start_op(); census_add_method_tag(id_3, "foo_3"); active_ops = census_get_active_ops(&n); GPR_ASSERT(n == 3); for (i = 0; i < 3; i++) { census_trace_obj_destroy(active_ops[i]); } gpr_free(active_ops); active_ops = NULL; /* End the second op and add annotations to the third ops*/ census_tracing_end_op(id_2); census_tracing_print(id_3, annotation_txt[0]); census_tracing_print(id_3, annotation_txt[1]); active_ops = census_get_active_ops(&n); GPR_ASSERT(active_ops != NULL); GPR_ASSERT(n == 2); for (i = 0; i < 2; i++) { census_trace_obj_destroy(active_ops[i]); } gpr_free(active_ops); active_ops = NULL; /* End all ops. */ census_tracing_end_op(id_1); census_tracing_end_op(id_3); active_ops = census_get_active_ops(&n); GPR_ASSERT(active_ops == NULL); GPR_ASSERT(n == 0); census_tracing_shutdown(); }
void census_shutdown(void) { census_stats_store_shutdown(); census_tracing_shutdown(); }