added test on errno = ENOMEM on out of memory

1 files changed, 18 insertions, 0 deletions

diff --git a/src/tests/tcmalloc_unittest.cc b/src/tests/tcmalloc_unittest.cc
index 423cae1..69698bc 100644
--- a/src/tests/tcmalloc_unittest.cc
+++ b/src/tests/tcmalloc_unittest.cc
@@ -1047,6 +1047,23 @@ static void TestSetNewMode() {
   tc_set_new_mode(old_mode);
 }
 
+static void TestErrno(void) {
+  errno = 0;
+  void* ret = memalign(128, kTooBig);
+  EXPECT_EQ(NULL, ret);
+  EXPECT_EQ(ENOMEM, errno);
+
+  errno = 0;
+  ret = malloc(kTooBig);
+  EXPECT_EQ(NULL, ret);
+  EXPECT_EQ(ENOMEM, errno);
+
+  errno = 0;
+  ret = tc_malloc_skip_new_handler(kTooBig);
+  EXPECT_EQ(NULL, ret);
+  EXPECT_EQ(ENOMEM, errno);
+}
+
 static int RunAllTests(int argc, char** argv) {
   // Optional argv[1] is the seed
   AllocatorState rnd(argc > 1 ? atoi(argv[1]) : 100);
@@ -1379,6 +1396,7 @@ static int RunAllTests(int argc, char** argv) {
   TestReleaseToSystem();
   TestAggressiveDecommit();
   TestSetNewMode();
+  TestErrno();
 
   return 0;
 }