[#3] Added test for RawJPEGFactory

Clovel · Jun 10, 2019 · d9039c2 · d9039c2
1 parent e9b22e9
commit d9039c2
Show file tree

Hide file tree

Showing 3 changed files with 77 additions and 5 deletions.
diff --git a/ressources/lena30.jpg b/ressources/lena30.jpg
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
@@ -39,6 +39,7 @@ add_test(test_base64_decode tests_${PROJECT_NAME} 2)
 add_test(test_base64_encode tests_${PROJECT_NAME} 3)
 add_test(test_literal_to_string tests_${PROJECT_NAME} 4)
 add_test(test_string_to_literal tests_${PROJECT_NAME} 5)
+add_test(test_raw_jpeg_factory tests_${PROJECT_NAME} 6)
 # add_test(XXXXX tests_${PROJECT_NAME} 0)
 
 add_executable(tests_${PROJECT_NAME}

diff --git a/tests/test_main.cpp b/tests/test_main.cpp
@@ -2,9 +2,13 @@
 #include "Base64Encoder.hpp"
 #include "LiteralConverter.hpp"
 
+#include "RawJPEGFactory.hpp"
+
 /* C++ system */
 #include <iostream>
+#include <fstream>
 #include <string>
+#include <exception>
 
 /* C system */
 #include <cstring>
@@ -33,7 +37,7 @@ int test_base64_encode_decode(void) {
 
     assert(lStr == lDecodedStr);
 
-    return 0;
+    return EXIT_SUCCESS;
 }
 
 int test_base64_decode_encode(void) {
@@ -49,7 +53,7 @@ int test_base64_decode_encode(void) {
 
     assert(lStr == lEncodedStr);
 
-    return 0;
+    return EXIT_SUCCESS;
 }
 
 int test_base64_decode(void) {
@@ -63,7 +67,7 @@ int test_base64_decode(void) {
 
     assert(lExpectedStr == lDecodedStr);
 
-    return 0;
+    return EXIT_SUCCESS;
 }
 
 int test_base64_encode(void) {
@@ -82,7 +86,7 @@ int test_base64_encode(void) {
 
     assert(lExpectedStr == lEncodedStr);
 
-    return 0;
+    return EXIT_SUCCESS;
 }
 
 int test_literal_to_string(void) {
@@ -93,7 +97,7 @@ int test_literal_to_string(void) {
 
     assert(lExpectedStr == lResult);
 
-    return 0;
+    return EXIT_SUCCESS;
 }
 
 int test_string_to_literal(void) {
@@ -107,6 +111,70 @@ int test_string_to_literal(void) {
     return 0;
 }
 
+int test_raw_jpeg_factory(void) {
+    std::ifstream lInputFileStream;
+    lInputFileStream.open("../../ressources/lena30.jpg", std::ios_base::in | std::ios_base::binary | std::ios_base::ate);
+    if(lInputFileStream.fail() || !lInputFileStream.is_open()) {
+        std::cout << "[ERROR] Failed to open ../../ressources/lena30.jpg" << std::endl;
+        return EXIT_FAILURE;
+    }
+
+    std::string lRawJPEG;
+    size_t lStrRawLength = lInputFileStream.tellg();
+    std::cout << "[INFO ] lStrRawLength = " << lStrRawLength << std::endl;
+
+    /* Allocating memory for the string in advance */
+    try {
+        lRawJPEG.reserve(lStrRawLength);
+        } catch (const std::exception &e) {
+        std::cout << "[ERROR] Excpetion \"" << e.what() << "\" occured during allocation !" << std::endl;
+        return EXIT_FAILURE;
+    }
+    try {
+        lInputFileStream.seekg(0, std::ios::beg);
+    } catch (const std::bad_alloc &e) {
+        std::cout << "[ERROR] Excpetion \"" << e.what() << "\" occured during seekg back to BOF !" << std::endl;
+        return EXIT_FAILURE;
+    }
+
+    /* Getting the string */
+    try {
+        lRawJPEG.assign((std::istreambuf_iterator<char>(lInputFileStream)),
+                std::istreambuf_iterator<char>());
+    } catch (const std::exception &e) {
+        std::cout << "[ERROR] Excpetion \"" << e.what() << "\" occured during data read !" << std::endl;
+        return EXIT_FAILURE;
+    }
+
+    //std::cout << "[DEBUG] Raw contents are : " << lRawJPEG << std::endl;
+
+    /* Instanciate a RawJPEGFactory */
+    RawJPEGFactory lFactory;
+
+    /* Segmenting the string */
+    size_t lSegmentSize = lStrRawLength/100;
+    size_t lLeftSize = lStrRawLength % 100;
+    std::cout << "[DEBUG] lSegmentSize = " << lSegmentSize << ", lSegmentSize*100 = " << lSegmentSize*100 << ", left = " << lLeftSize << std::endl;
+
+    /* Getting segments and appending them to the Facotry string */
+    std::string lTmpStr, lAppendResult;
+    for(unsigned int i = 0U; 100 > i; ++i) {
+        lTmpStr = lRawJPEG.substr(i*lSegmentSize, lSegmentSize);
+        lAppendResult = lFactory.append(lTmpStr);
+        assert(std::string() == lAppendResult);
+    }
+    lTmpStr = lRawJPEG.substr(lStrRawLength - lLeftSize, lSegmentSize);
+    lAppendResult = lFactory.append(lTmpStr);
+
+    /* Check if the Factory returned the raw string */
+    assert(std::string() != lAppendResult);
+
+    /* Check if the Factory's result is equal to the original raw data */
+    assert(lRawJPEG == lAppendResult);
+
+    return EXIT_SUCCESS;
+}
+
 
 /* ----------------------------------------------------- */
 /* main function --------------------------------------- */
@@ -147,6 +215,9 @@ int main(const int argc, const char * const * const argv)
         case 5:
             result = test_string_to_literal();
             break;
+        case 6:
+            result = test_raw_jpeg_factory();
+            break;
         default:
             fprintf(stdout, "[INFO ] test #%d not available\n", test_num);
             fflush(stdout);