1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
//===----------------------------------------------------------------------===//
//
// 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
// <experimental/memory_resource>
// memory_resource * new_delete_resource()
#include <experimental/memory_resource>
#include <type_traits>
#include <cassert>
#include "count_new.h"
#include "test_macros.h"
namespace ex = std::experimental::pmr;
struct assert_on_compare : public ex::memory_resource
{
protected:
void * do_allocate(size_t, size_t) override
{ assert(false); return nullptr; }
void do_deallocate(void *, size_t, size_t) override
{ assert(false); }
bool do_is_equal(ex::memory_resource const &) const noexcept override
{ assert(false); return false; }
};
void test_return()
{
{
static_assert(std::is_same<
decltype(ex::new_delete_resource()), ex::memory_resource*
>::value, "");
}
// assert not null
{
assert(ex::new_delete_resource());
}
// assert same return value
{
assert(ex::new_delete_resource() == ex::new_delete_resource());
}
}
void test_equality()
{
// Same object
{
ex::memory_resource & r1 = *ex::new_delete_resource();
ex::memory_resource & r2 = *ex::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
{
ex::memory_resource & r1 = *ex::new_delete_resource();
assert_on_compare c;
ex::memory_resource & r2 = c;
assert(r1 != r2);
assert(!(r1 == r2));
}
}
void test_allocate_deallocate()
{
ex::memory_resource & r1 = *ex::new_delete_resource();
globalMemCounter.reset();
void *ret = r1.allocate(50);
assert(ret);
assert(globalMemCounter.checkOutstandingNewEq(1));
assert(globalMemCounter.checkLastNewSizeEq(50));
r1.deallocate(ret, 1);
assert(globalMemCounter.checkOutstandingNewEq(0));
assert(globalMemCounter.checkDeleteCalledEq(1));
}
int main(int, char**)
{
static_assert(noexcept(ex::new_delete_resource()), "Must be noexcept");
test_return();
test_equality();
test_allocate_deallocate();
return 0;
}
|