소스 검색

Unit test: Allow all redirect codes

bel2125 7 년 전
부모
커밋
f358b84aaf
1개의 변경된 파일8개의 추가작업 그리고 2개의 파일을 삭제
  1. 8 2
      unittest/public_server.c

+ 8 - 2
unittest/public_server.c

@@ -1383,7 +1383,10 @@ START_TEST(test_request_handlers)
 	client_ri = mg_get_response_info(client_conn);
 
 	ck_assert(client_ri != NULL);
-	ck_assert_int_eq(client_ri->status_code, 302);
+	ck_assert((client_ri->status_code == 301) || (client_ri->status_code == 302)
+	          || (client_ri->status_code == 303)
+	          || (client_ri->status_code == 307)
+	          || (client_ri->status_code == 308)); /* is a redirect code */
 	i = mg_read(client_conn, buf, sizeof(buf));
 	ck_assert_int_eq(i, -1);
 	mg_close_connection(client_conn);
@@ -1411,7 +1414,10 @@ START_TEST(test_request_handlers)
 	client_ri = mg_get_response_info(client_conn);
 
 	ck_assert(client_ri != NULL);
-	ck_assert_int_eq(client_ri->status_code, 302);
+	ck_assert((client_ri->status_code == 301) || (client_ri->status_code == 302)
+	          || (client_ri->status_code == 303)
+	          || (client_ri->status_code == 307)
+	          || (client_ri->status_code == 308)); /* is a redirect code */
 	i = mg_read(client_conn, buf, sizeof(buf));
 	ck_assert_int_eq(i, -1);
 	mg_close_connection(client_conn);