Add test for a PAC get without content length

1 files changed, 21 insertions, 0 deletions

diff --git a/libproxy/test/get-pac-test.cpp b/libproxy/test/get-pac-test.cpp
index c7d49d9..a20690b 100644
--- a/libproxy/test/get-pac-test.cpp
+++ b/libproxy/test/get-pac-test.cpp
@@ -143,6 +143,8 @@ class TestServer {
 				sendOverflow(csock);
 			} else if (strstr(buffer, "chunked")) {
 				sendChunked(csock);
+			} else if (strstr(buffer, "without_content_length")) {
+				sendWithoutContentLength(csock);
 			} else {
 				assert(!"Unsupported request");
 			}
@@ -224,6 +226,20 @@ done:
 			close(ret);
 		}
 
+		void sendWithoutContentLength(int csock)
+		{
+			int ret;
+			const char *basic =
+				"HTTP/1.1 200 OK\n" \
+				"Content-Type: text/plain\n" \
+				"\n" \
+				"0123456789";
+			ret = send(csock, (void*)basic, strlen(basic), 0);
+			assert(ret == strlen(basic));
+			shutdown(csock, SHUT_RDWR);
+			close(ret);
+		}
+
 		in_port_t m_port;
 		int m_sock;
 		int m_pipe[2];
@@ -241,6 +257,7 @@ int main()
 	url truncated("http://localhost:1983/truncated.js");
 	url overflow("http://localhost:1983/overflow.js");
 	url chunked("http://localhost:1983/chunked.js");
+	url without_content_length("http://localhost:1983/without_content_length.js");
 
 	server.start();
 
@@ -260,6 +277,10 @@ int main()
 	pac = chunked.get_pac();
 	if (!(pac != NULL && strlen(pac) == 10 && !strcmp("0123456789", pac)))
 		return 4; // Test failed, exit with error code
+
+	pac = without_content_length.get_pac();
+	if (!(pac != NULL && strlen(pac) == 10 && !strcmp("0123456789", pac)))
+		return 5; // Test failed, exit with error code
 	delete[] pac;
 
 	server.stop();