adding tests for upload_handler

Author: StokesAsselborn

tests/upload_handler_unittest.cpp

@@ -0,0 +1,124 @@
+
+#include "handler/upload_handler.hpp"
+#include "http/http_response.hpp"
+#include "stdio.h"
+#include "utest/utest.h"
+
+static void write_all(Handler* handler, const std::string& content)
+{
+    size_t written = 0;
+
+    while (written < content.size()) {
+        size_t n = handler->write_data(content.c_str() + written, content.size() - written);
+
+        // write_data returns 0 when upload is complete OR on error.
+        // If upload is not done but 0 is returned, break to avoid infinite loop.
+        if (n == 0 && handler->needs_input())
+            break;
+
+        written += n;
+    }
+}
+
+UTEST(UploadHandlerTest, needs_input_upload_ongoing)
+{
+    std::string upload_path = "www/example/needs_input_upload_ongoing.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    ASSERT_TRUE(handler->needs_input());
+    delete (handler);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, has_output_upload_ongoing)
+{
+    std::string upload_path = "www/example/has_output_upload_ongoing.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    ASSERT_FALSE(handler->has_output());
+    delete (handler);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, needs_input_upload_done)
+{
+    std::string upload_path = "www/example/needs_input_upload_done.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    write_all(handler, content);
+    ASSERT_TRUE(handler->has_output());
+    delete (handler);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, has_output_upload_done)
+{
+    std::string upload_path = "www/example/has_output_upload_done.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    write_all(handler, content);
+    ASSERT_TRUE(handler->has_output());
+    delete (handler);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, check_content_uploaded)
+{
+    std::string upload_path = "www/example/check_content_uploaded.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    write_all(handler, content);
+    ASSERT_TRUE(handler->has_output());
+    delete (handler);
+
+    // validating content
+    FILE* f = fopen(upload_path.c_str(), "r");
+    ASSERT_TRUE(f != NULL);
+    char buffer[1028];
+    size_t n = fread(buffer, 1, sizeof(buffer), f);
+    fclose(f);
+    std::string file_content(buffer, n);
+    ASSERT_EQ(content.c_str(), file_content.c_str());
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, return_201)
+{
+    std::string upload_path = "www/example/return_201.txt";
+    std::string content = "some content to be written to the file";
+    Handler* handler = new UploadHandler(upload_path, content.size());
+    ASSERT_FALSE(handler->has_output());
+    write_all(handler, content);
+    ASSERT_TRUE(handler->has_output());
+    char buffer[1028];
+    size_t n = handler->read_data(buffer, sizeof(buffer));
+    std::string http_res(buffer, n);
+    ASSERT_TRUE(http_res.find("201") != std::string::npos);
+    delete (handler);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, return_409)
+{
+    std::string upload_path = "www/example/return_409.txt";
+    std::string content = "some content to be written to the file";
+    {
+        Handler* handler1 = new UploadHandler(upload_path, content.size());
+        write_all(handler1, content);
+        ASSERT_TRUE(handler1->has_output());
+        delete handler1;
+    }
+    Handler* handler2 = new UploadHandler(upload_path, content.size());
+    ASSERT_TRUE(handler2->has_output());
+    char buffer[1028];
+    size_t n = handler2->read_data(buffer, sizeof(buffer));
+    std::string http_res(buffer, n);
+    ASSERT_TRUE(http_res.find("409") != std::string::npos);
+    delete (handler2);
+    remove(upload_path.c_str());
+}
+
+UTEST(UploadHandlerTest, return_507)
+{
+    UTEST_SKIP("TODO: Test Disk full without making your machine explode");
+}