Update tests

sogou · Mar 24, 2023 · 891b31e · 891b31e
1 parent 9a2ebdb
commit 891b31e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/test/facilities_unittest.cc b/test/facilities_unittest.cc
@@ -75,7 +75,12 @@ TEST(facilities_unittest, async_request)
 	if (res.task_state == WFT_STATE_SUCCESS)
 	{
 		auto code = atoi(res.resp.get_status_code());
-		EXPECT_EQ(code, HttpStatusOK);
+		EXPECT_TRUE(code == HttpStatusOK ||
+					code == HttpStatusMovedPermanently ||
+					code == HttpStatusFound ||
+					code == HttpStatusSeeOther ||
+					code == HttpStatusTemporaryRedirect ||
+					code == HttpStatusPermanentRedirect);
 	}
 }
 

diff --git a/test/http_unittest.cc b/test/http_unittest.cc
@@ -82,7 +82,12 @@ TEST(http_unittest, WFHttpTask2)
 		if (state == WFT_STATE_SUCCESS)
 		{
 			auto code = atoi(task->get_resp()->get_status_code());
-			EXPECT_EQ(code, HttpStatusOK);
+			EXPECT_TRUE(code == HttpStatusOK ||
+						code == HttpStatusMovedPermanently ||
+						code == HttpStatusFound ||
+						code == HttpStatusSeeOther ||
+						code == HttpStatusTemporaryRedirect ||
+						code == HttpStatusPermanentRedirect);
 		}
 
 		mutex.lock();