32 lines · cpp
1//===-- Unittests for malloc ----------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "src/stdlib/free.h"10#include "src/stdlib/malloc.h"11#include "test/UnitTest/Test.h"12 13TEST(LlvmLibcMallocTest, Allocate) {14 int *ptr = reinterpret_cast<int *>(LIBC_NAMESPACE::malloc(sizeof(int)));15 EXPECT_NE(reinterpret_cast<void *>(ptr), static_cast<void *>(nullptr));16 *ptr = 1;17 EXPECT_EQ(*ptr, 1);18 LIBC_NAMESPACE::free(ptr);19}20 21TEST(LlvmLibcMallocTest, Nullptr) {22 int *ptr = reinterpret_cast<int *>(LIBC_NAMESPACE::malloc(0));23 EXPECT_EQ(reinterpret_cast<void *>(ptr), static_cast<void *>(nullptr));24 LIBC_NAMESPACE::free(ptr);25}26 27TEST(LlvmLibcMallocTest, LargeAllocation) {28 int *ptr = reinterpret_cast<int *>(LIBC_NAMESPACE::malloc(2ul * 1024 * 1024));29 EXPECT_NE(reinterpret_cast<void *>(ptr), static_cast<void *>(nullptr));30 LIBC_NAMESPACE::free(ptr);31}32