From 0ff230108329e869cf37577fbf27c819cebff6fb Mon Sep 17 00:00:00 2001
From: Theo Butler <theodusbutler@gmail.com>
Date: Wed, 13 Feb 2019 08:12:07 -0500
Subject: [PATCH] use malloc shim in malloc tests

---
 CMakeLists.txt | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d0d5629..5d09dca 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -81,7 +81,11 @@ foreach(SUPER_SLAB_SIZE 1;16)
       unset(SRC)
       aux_source_directory(${TESTDIR}/${TEST_CATEGORY}/${TEST} SRC)
       set(TESTNAME "${TEST_CATEGORY}-${TEST}-${SUPER_SLAB_SIZE}")
-      add_executable(${TESTNAME} ${SRC} src/override/new.cc)
+      if(${TEST} STREQUAL "malloc")
+        add_executable(${TESTNAME} ${SRC} src/override/malloc.cc)
+      else()
+        add_executable(${TESTNAME} ${SRC} src/override/new.cc)
+      endif()
       if (${SUPER_SLAB_SIZE} EQUAL 1)
         target_compile_definitions(${TESTNAME} PRIVATE IS_ADDRESS_SPACE_CONSTRAINED)
       endif()