From bec0ab18315483defecaa5ad9612f85e721e10fc Mon Sep 17 00:00:00 2001
From: Alex Hultman <alexhultman@gmail.com>
Date: Sat, 22 Jun 2019 12:43:33 +0200
Subject: [PATCH] Cover more code in mocked hello world

---
 fuzzing/MockedHelloWorld.cpp | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/fuzzing/MockedHelloWorld.cpp b/fuzzing/MockedHelloWorld.cpp
index 5a79513..f58fc59 100644
--- a/fuzzing/MockedHelloWorld.cpp
+++ b/fuzzing/MockedHelloWorld.cpp
@@ -11,7 +11,24 @@ us_listen_socket_t *listenSocket;
 extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
 
     app = new uWS::TemplatedApp<false>(uWS::App().get("/*", [](auto *res, auto *req) {
-        res->end("Hello world!");
+        if (req->getHeader("use_write").length()) {
+            res->writeStatus("200 OK")->writeHeader("write", "true")->write("Hello");
+            res->write(" world!");
+            res->end();
+        } else if (req->getQuery().length()) {
+            res->close();
+        } else {
+            res->end("Hello world!");
+        }
+    }).post("/*", [](auto *res, auto *req) {
+        res->onAborted([]() {
+
+        });
+        res->onData([res](std::string_view chunk, bool isEnd) {
+            if (isEnd) {
+                res->end(chunk);
+            }
+        });
     }).listen(9001, [](us_listen_socket_t *listenSocket) {
         if (listenSocket) {
             std::cout << "Listening on port " << 9001 << std::endl;