//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++03, c++11, c++14
// TODO: Change to XFAIL once https://github.com/llvm/llvm-project/issues/40340 is fixed
// UNSUPPORTED: availability-pmr-missing
// <memory_resource>
// memory_resource *new_delete_resource()
#include <memory_resource>
#include <cassert>
#include <type_traits>
#include "count_new.h"
#include "test_macros.h"
class assert_on_compare : public std::pmr::memory_resource {
void* do_allocate(std::size_t, size_t) override {
assert(false);
return nullptr;
}
void do_deallocate(void*, std::size_t, size_t) override { assert(false); }
bool do_is_equal(const std::pmr::memory_resource&) const noexcept override {
assert(false);
return true;
}
};
void test_return() {
{ ASSERT_SAME_TYPE(decltype(std::pmr::new_delete_resource()), std::pmr::memory_resource*); }
// assert not null
{ assert(std::pmr::new_delete_resource()); }
// assert same return value
{ assert(std::pmr::new_delete_resource() == std::pmr::new_delete_resource()); }
}
void test_equality() {
// Same object
{
std::pmr::memory_resource& r1 = *std::pmr::new_delete_resource();
std::pmr::memory_resource& r2 = *std::pmr::new_delete_resource();
// check both calls returned the same object
assert(&r1 == &r2);
// check for proper equality semantics
assert(r1 == r2);
assert(r2 == r1);
assert(!(r1 != r2));
assert(!(r2 != r1));
}
// Different types
{
std::pmr::memory_resource& r1 = *std::pmr::new_delete_resource();
assert_on_compare c;
std::pmr::memory_resource& r2 = c;
assert(r1 != r2);
assert(!(r1 == r2));
}
}
void test_allocate_deallocate() {
std::pmr::memory_resource& r1 = *std::pmr::new_delete_resource();
globalMemCounter.reset();
void* ret = r1.allocate(50);
assert(ret);
ASSERT_WITH_LIBRARY_INTERNAL_ALLOCATIONS(globalMemCounter.checkOutstandingNewEq(1));
ASSERT_WITH_LIBRARY_INTERNAL_ALLOCATIONS(globalMemCounter.checkLastNewSizeEq(50));
r1.deallocate(ret, 50);
assert(globalMemCounter.checkOutstandingNewEq(0));
ASSERT_WITH_LIBRARY_INTERNAL_ALLOCATIONS(globalMemCounter.checkDeleteCalledEq(1));
}
int main(int, char**) {
ASSERT_NOEXCEPT(std::pmr::new_delete_resource());
test_return();
test_equality();
test_allocate_deallocate();
return 0;
}