Improvements to libtest

Signed-off-by: Mattias Andrée <m@maandree.se>

1 files changed, 31 insertions, 0 deletions

diff --git a/libtest/libtest_get_alloc_failure_in.c b/libtest/libtest_get_alloc_failure_in.c
new file mode 100644
index 0000000..347c973
--- /dev/null
+++ b/libtest/libtest_get_alloc_failure_in.c
@@ -0,0 +1,31 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+#ifndef TEST
+
+
+size_t
+libtest_get_alloc_failure_in(void)
+{
+	return libtest_malloc_fail_in;
+}
+
+
+#else
+
+
+int
+main(void)
+{
+	EXPECT(libtest_malloc_fail_in == 0u);
+	EXPECT(libtest_get_alloc_failure_in() == 0u);
+	EXPECT(libtest_get_alloc_failure_in() == 0u);
+	EXPECT(libtest_malloc_fail_in == 0u);
+	libtest_malloc_fail_in = 4u;
+	EXPECT(libtest_get_alloc_failure_in() == 4u);
+	EXPECT(libtest_get_alloc_failure_in() == 4u);
+	EXPECT(libtest_malloc_fail_in == 4u);
+	return 0;
+}
+
+
+#endif