fixes #745 HTTP server redirect handler

1 files changed, 116 insertions, 0 deletions

diff --git a/tests/httpserver.c b/tests/httpserver.c
index 21b961b8..1bf88f12 100644
--- a/tests/httpserver.c
+++ b/tests/httpserver.c
@@ -550,4 +550,120 @@ TestMain("HTTP Server", {
 			nng_url_free(curl);
 		});
 	});
+
+	Convey("Redirect handler works", {
+		char     urlstr[32];
+		nng_url *url;
+
+		trantest_next_address(urlstr, "http://127.0.0.1:%u");
+		So(nng_url_parse(&url, urlstr) == 0);
+		So(nng_http_server_hold(&s, url) == 0);
+
+		Reset({
+			nng_http_server_release(s);
+			nng_url_free(url);
+		});
+
+		Convey("GET redirect works", {
+			char          fullurl[256];
+			nng_http_req *req;
+			nng_http_res *res;
+			nng_url *     curl;
+			const char *  dest;
+			void *        data;
+			size_t        size;
+
+			So(nng_http_handler_alloc_redirect(&h, "/here", 301,
+			       "http://127.0.0.1/there") == 0);
+			So(nng_http_server_add_handler(s, h) == 0);
+			So(nng_http_server_start(s) == 0);
+			nng_msleep(100);
+
+			So(nng_http_res_alloc(&res) == 0);
+			snprintf(fullurl, sizeof(fullurl), "%s/here", urlstr);
+			So(nng_url_parse(&curl, fullurl) == 0);
+			So(nng_http_req_alloc(&req, curl) == 0);
+			So(nng_http_req_set_method(req, "GET") == 0);
+
+			So(httpdo(curl, req, res, &data, &size) == 0);
+			So(nng_http_res_get_status(res) == 301);
+			So((dest = nng_http_res_get_header(res, "Location")) !=
+			    NULL);
+			So(strcmp(dest, "http://127.0.0.1/there") == 0);
+			So(data != NULL);
+			So(size > 0);
+			nng_http_req_free(req);
+			nng_http_res_free(res);
+			nng_url_free(curl);
+		});
+
+		Convey("Tree redirect works", {
+			char          fullurl[256];
+			nng_http_req *req;
+			nng_http_res *res;
+			nng_url *     curl;
+			const char *  dest;
+			void *        data;
+			size_t        size;
+
+			// We'll use a 303 to ensure codes carry thru
+			So(nng_http_handler_alloc_redirect(&h, "/here", 303,
+			       "http://127.0.0.1/there") == 0);
+			So(nng_http_handler_set_tree(h) == 0);
+			So(nng_http_server_add_handler(s, h) == 0);
+			So(nng_http_server_start(s) == 0);
+			nng_msleep(100);
+
+			So(nng_http_res_alloc(&res) == 0);
+			snprintf(fullurl, sizeof(fullurl),
+			    "%s/here/i/go/again", urlstr);
+			So(nng_url_parse(&curl, fullurl) == 0);
+			So(nng_http_req_alloc(&req, curl) == 0);
+			So(nng_http_req_set_method(req, "GET") == 0);
+
+			So(httpdo(curl, req, res, &data, &size) == 0);
+			So(nng_http_res_get_status(res) == 303);
+			So((dest = nng_http_res_get_header(res, "Location")) !=
+			    NULL);
+			So(strcmp(dest, "http://127.0.0.1/there/i/go/again") ==
+			    0);
+			nng_http_req_free(req);
+			nng_http_res_free(res);
+			nng_url_free(curl);
+		});
+
+		Convey("POST Redirect works", {
+			char          fullurl[256];
+			size_t        size;
+			nng_http_req *req;
+			nng_http_res *res;
+			nng_url *     curl;
+			char          txdata[5];
+			const char *  dest;
+			void *        data;
+
+			So(nng_http_handler_alloc_redirect(&h, "/here", 301,
+			       "http://127.0.0.1/there") == 0);
+			So(nng_http_server_add_handler(s, h) == 0);
+			So(nng_http_server_start(s) == 0);
+			nng_msleep(100);
+
+			snprintf(txdata, sizeof(txdata), "1234");
+			So(nng_http_res_alloc(&res) == 0);
+			snprintf(fullurl, sizeof(fullurl), "%s/here", urlstr);
+			So(nng_url_parse(&curl, fullurl) == 0);
+			So(nng_http_req_alloc(&req, curl) == 0);
+			nng_http_req_set_data(req, txdata, strlen(txdata));
+			So(nng_http_req_set_method(req, "POST") == 0);
+			So(httpdo(curl, req, res, (void **) &data, &size) ==
+			    0);
+			So(nng_http_res_get_status(res) == 301);
+			So((dest = nng_http_res_get_header(res, "Location")) !=
+			    NULL);
+			So(strcmp(dest, "http://127.0.0.1/there") == 0);
+			nng_http_req_free(req);
+			nng_http_res_free(res);
+			nng_url_free(curl);
+		});
+	});
 })