[WIP] coverage increased due new malloc() interceptor.

Author: Gonzalo Diaz

src/tests/unit/lib/hackerrank/warmup/staircase.test.cpp

@@ -1,39 +1,68 @@
+// 3rd party libs
 #include <catch2/catch_test_macros.hpp>
 
+#include <nlohmann/json.hpp>
+
+// local libs
+#if defined(__linux__)
+#include "../../../../tools/memory_injector.hpp"
+#endif
 #include <exercises/hackerrank/warmup/staircase.h>
+
+// std libs
 #include <filesystem>
 #include <fstream>
 #include <nlohmann/json.hpp>
 #include <vector>
 
 using json = nlohmann::json;
 
-TEST_CASE("staircase", "[warmup]") {
-  std::filesystem::path cwd = std::filesystem::current_path();
-  std::string path =
-      cwd.string() + "/unit/lib/hackerrank/warmup/staircase.testcases.json";
+TEST_CASE("staircase JSON Test Cases", "[hackerrank] [jsontestcase] [warmup]") {
+#if defined(__linux__)
+  SECTION("staircase: Failure handling during Out of Memory (OOM)") {
+    int input = 4;
 
-  INFO("staircase JSON test cases FILE: " << path);
+    MemoryInjector::enable_oom_fault();
+    REQUIRE(HACKERRANK_WARMUP_staircaseCalculate(input) == nullptr);
+    MemoryInjector::disable_fault();
 
-  std::ifstream f(path);
-  json data = json::parse(f);
+    MemoryInjector::fail_on_allocation_number(2);
+    REQUIRE(HACKERRANK_WARMUP_staircaseCalculate(input) == nullptr);
+    MemoryInjector::disable_fault();
 
-  for (auto testcase : data) {
-    auto input = static_cast<int>(testcase["input"]);
+    MemoryInjector::enable_oom_fault();
+    REQUIRE_NOTHROW(HACKERRANK_WARMUP_staircase(input));
+    MemoryInjector::disable_fault();
+  }
+#endif
 
-    char **result = HACKERRANK_WARMUP_staircaseCalculate(input);
+  SECTION("plusMinus: Normal memory allocation succeeds") {
+    std::filesystem::path cwd = std::filesystem::current_path();
+    std::string path =
+        cwd.string() + "/unit/lib/hackerrank/warmup/staircase.testcases.json";
 
-    std::vector<std::string> result_as_vector;
+    INFO("staircase JSON test cases FILE: " << path);
 
-    for (int i = 0; i < input; i++) {
-      result_as_vector.emplace_back(result[i]);
-    }
+    std::ifstream f(path);
+    json data = json::parse(f);
 
-    HACKERRANK_WARMUP_freeStaircase(result, input);
+    for (auto testcase : data) {
+      auto input = static_cast<int>(testcase["input"]);
 
-    CHECK(result_as_vector == testcase["expected"]);
+      char **result = HACKERRANK_WARMUP_staircaseCalculate(input);
 
-    // Just call void function, to collect coverage
-    HACKERRANK_WARMUP_staircase(input);
+      std::vector<std::string> result_as_vector;
+
+      for (int i = 0; i < input; i++) {
+        result_as_vector.emplace_back(result[i]);
+      }
+
+      HACKERRANK_WARMUP_freeStaircase(result, input);
+
+      CHECK(result_as_vector == testcase["expected"]);
+
+      // Just call void function, to collect coverage
+      HACKERRANK_WARMUP_staircase(input);
+    }
   }
 }