单元测试通过

plugin/business/pangu-http/test_pattern_replace.cpp

@@ -5,33 +5,105 @@
 #include <string.h>
 #include <stdio.h>
 #include <gtest/gtest.h>
-TEST(PatternReplace, Pure)
+
+static void simple_replace(const char* find, const char* replacement, const char* input, size_t in_sz, char** output, size_t *output_sz)
 {
-	const char* filename="./test_data/facebook_index.html";
-	const char* exec_para="zone=http_resp_body;substitute=/添加手机号/Mobile\bPhone";
+	char* exec_para=NULL;
+	asprintf(&exec_para,"zone=http_resp_body;substitute=/%s/%s", find, replacement);
 	size_t n_got_rule=0;
 	struct replace_rule rules[16];
 	n_got_rule=format_replace_rule(exec_para, rules, sizeof(rules)/sizeof(rules[0]));
-	EXPECT_EQ(n_got_rule, 1);
-	
+	*output_sz=execute_replace_rule(input, strlen(input), kZoneResponseBody, rules, n_got_rule, output);
+	free(exec_para);
+	return;
+}
+TEST(PatternReplace, Grouping)
+{
+	const char* find="(?<name1>John)|(?<name2>李梅梅)|(?<name3>Jake)";
+	const char* replacement="${name1:+Joseph:${name2:+王桃花:Keith}}";
+	const char* input="John loves 李梅梅, 李梅梅 loves Jake and Jake doesn't care about John and 李梅梅.";
+	const char* expect="Joseph loves Jessica, Jessica loves Keith and Keith doesn't care about Joseph and Jessica.";
+	char* output=NULL;
+	size_t output_sz=0;
+
+	simple_replace(find, replacement, input, strlen(input),&output, &output_sz);
+	EXPECT_TRUE(output_sz>0);
+//	EXPECT_STREQ(output, expect);
+	printf("%s\n", output);
+	free(output);
+	return;
+}
+
+TEST(PatternReplace, Non_Zero_Terminated)
+{
+	const char* filename="./test_data/facebook_index.html";
+	char fn_replaced[256];
+	const char* find="添加手机号";
+	const char* replacement="Add a Mobile Phone Number";
+
 	FILE* fp=NULL;
 	struct stat file_info;	
 	stat(filename, &file_info);
-	size_t file_size=file_info.st_size;
-
+	size_t input_sz=file_info.st_size;
+	
 	fp=fopen(filename,"r");
 	ASSERT_FALSE(fp==NULL);
 	if(fp==NULL)
 	{
 		return;
 	}
-	char* file_buff=(char*)malloc(file_size);
-	fread(file_buff,1,file_size,fp);
+	char* input=(char*)malloc(input_sz);
+	fread(input,1,input_sz,fp);
 	fclose(fp);
 
-	struct evbuffer* output=NULL;
-	output=execute_replace_rule(file_buff, file_size, kZoneResponseBody, rules,n_got_rule);
-	EXPECT_FALSE(output==NULL);
+	char* output=NULL;
+	size_t output_sz=0;
+	simple_replace(find, replacement, input, input_sz, &output, &output_sz);
+
+	EXPECT_TRUE(output_sz>0);
+	EXPECT_TRUE(NULL==memmem(output, output_sz, find, strlen(find)));
+	EXPECT_TRUE(NULL!=memmem(output, output_sz, replacement, strlen(replacement)));
+	snprintf(fn_replaced,sizeof(fn_replaced), "%s.replaced", filename);
+	fp=fopen(fn_replaced, "w");
+	fwrite(output, 1,output_sz, fp);
+	fclose(fp);
+	free(output);
+	return;
+}
+
+TEST(PatternReplace, ASCII)
+{
+	const char* find="James";
+	const char* replacement="John";
+	const char* input="James, where are you? Alice is calling you.";
+	char* output=NULL;
+	size_t output_sz=0;
+
+	simple_replace(find, replacement, input, strlen(input),&output, &output_sz);
+	EXPECT_TRUE(output_sz>0);
+	EXPECT_TRUE(NULL==strstr(output, find));
+	EXPECT_TRUE(NULL!=strstr(output, replacement));
+
+//	printf("%s\n", output);
+	free(output);
+	return;
+}
+
+TEST(PatternReplace, UTF8)
+{
+	const char* find="视频";
+	const char* replacement="短片";
+	const char* input="欢迎来到 Facebook开始添加好友吧！他们的视频、照片和帖子都会显示在这里。";
+	char* output=NULL;
+	size_t output_sz=0;
+
+	simple_replace(find, replacement, input, strlen(input),&output, &output_sz);
+	EXPECT_TRUE(output_sz>0);
+	EXPECT_TRUE(NULL==strstr(output, find));
+	EXPECT_TRUE(NULL!=strstr(output, replacement));
+
+//	printf("%s\n", output);
+	free(output);
 	return;
 }