updated tests for null

goatshriek · Oct 20, 2024 · 97cf7ad · 97cf7ad
1 parent 0620624
commit 97cf7ad
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/src/memory.c b/src/memory.c
@@ -137,13 +137,13 @@ alloc_array( size_t item_count, size_t item_size ) {
 }
 
 malloc_func_t stumpless_get_malloc(void) {
-    return stumpless_malloc;
+    return stumpless_malloc ? stumpless_malloc : NULL;
 }
 
 free_func_t stumpless_get_free(void) {
-    return stumpless_free;
+    return stumpless_free ? stumpless_free : NULL;
 }
 
 realloc_func_t stumpless_get_realloc(void) {
-    return stumpless_realloc;
+    return stumpless_realloc ? stumpless_realloc : NULL;
 }
diff --git a/test/function/memory.cpp b/test/function/memory.cpp
@@ -79,4 +79,25 @@ namespace {
     EXPECT_EQ(realloc_function, realloc);
   }
 
+  TEST(MemoryFunctionsTest, GetMalloc_NullFunction) {
+    stumpless_set_malloc(NULL);
+    auto malloc_function = stumpless_get_malloc();
+    EXPECT_EQ(malloc_function, nullptr);
+    EXPECT_ERROR_ID_EQ(STUMPLESS_ARGUMENT_EMPTY);
+  }
+
+  TEST(MemoryFunctionsTest, GetFree_NullFunction) {
+    stumpless_set_free(NULL);
+    auto free_function = stumpless_get_free();
+    EXPECT_EQ(free_function, nullptr);
+    EXPECT_ERROR_ID_EQ(STUMPLESS_ARGUMENT_EMPTY);
+  }
+
+  TEST(MemoryFunctionsTest, GetRealloc_NullFunction) {
+    stumpless_set_realloc(NULL);
+    auto realloc_function = stumpless_get_realloc();
+    EXPECT_EQ(realloc_function, nullptr);
+    EXPECT_ERROR_ID_EQ(STUMPLESS_ARGUMENT_EMPTY);
+  }
+
 }