Feature: performance test case

1 files changed, 4 insertions, 37 deletions

diff --git a/test/dns_decoder_test.cpp b/test/dns_decoder_test.cpp
index 44931e5..0472ad5 100644
--- a/test/dns_decoder_test.cpp
+++ b/test/dns_decoder_test.cpp
@@ -85,46 +85,13 @@ void dns_decoder_test_message_cb(struct session *ss, int topic_id, const void *m
 		cJSON_AddNumberToObject(real_result, "dns_rd", (double)(flag->rd));
 	}
 
-	uint16_t n_answer_rr=0;
-	struct dns_resource_record *answer_rr=NULL;
-	dns_message_answer_resource_record_get0(dns_msg, &answer_rr, &n_answer_rr);
-	
-	uint16_t n_authority_rr=0;
-	struct dns_resource_record *authority_rr=NULL;
-	dns_message_authority_resource_record_get0(dns_msg, &authority_rr, &n_authority_rr);
-
-	uint16_t n_additional_rr=0;
-	struct dns_resource_record *additional_rr=NULL;
-	dns_message_additional_resource_record_get0(dns_msg, &additional_rr, &n_additional_rr);
-
-	const char *answer=dns_resource_record_json_exporter(answer_rr, n_answer_rr);
-	const char *authority=dns_resource_record_json_exporter(authority_rr, n_authority_rr);
-	const char *additional=dns_resource_record_json_exporter(additional_rr, n_additional_rr);
-
-	cJSON *rr_array=cJSON_CreateObject();
-	if(answer!=NULL)
+	const char *resource_record_str=dns_message_resource_record_json_exporter(dns_msg);
+	if(resource_record_str!=NULL)
 	{
-		cJSON *rr_object=cJSON_Parse(answer);
-		cJSON_AddItemToObject(rr_array, "answer", rr_object);
-		free((void *)answer);
+		cJSON *rr_array=cJSON_Parse(resource_record_str);
+		cJSON_AddItemToObject(real_result, "rr", rr_array);
 	}
 
-	if(authority!=NULL)
-	{
-		cJSON *rr_object=cJSON_Parse(authority);
-		cJSON_AddItemToObject(rr_array, "authority", rr_object);
-		free((void *)authority);
-	}
-
-	if(additional!=NULL)
-	{
-		cJSON *rr_object=cJSON_Parse(additional);
-		cJSON_AddItemToObject(rr_array, "additional", rr_object);
-		free((void *)additional);
-	}
-
-	cJSON_AddItemToObject(real_result, "rr", rr_array);
-
 	if(plugin_env->write_result_enable==1)
 	{
 		char *real_result_str=cJSON_Print(real_result);