forked from OSchip/llvm-project
61 lines
1.9 KiB
C++
61 lines
1.9 KiB
C++
//===- StorageUniquerTest.cpp - StorageUniquer Tests ----------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Support/StorageUniquer.h"
|
|
#include "gmock/gmock.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
/// Simple storage class used for testing.
|
|
template <typename ConcreteT, typename... Args>
|
|
struct SimpleStorage : public StorageUniquer::BaseStorage {
|
|
using Base = SimpleStorage<ConcreteT, Args...>;
|
|
using KeyTy = std::tuple<Args...>;
|
|
|
|
SimpleStorage(KeyTy key) : key(key) {}
|
|
|
|
/// Get an instance of this storage instance.
|
|
template <typename... ParamsT>
|
|
static ConcreteT *get(StorageUniquer &uniquer, ParamsT &&...params) {
|
|
return uniquer.get<ConcreteT>(
|
|
/*initFn=*/{}, std::make_tuple(std::forward<ParamsT>(params)...));
|
|
}
|
|
|
|
/// Construct an instance with the given storage allocator.
|
|
static ConcreteT *construct(StorageUniquer::StorageAllocator &alloc,
|
|
KeyTy key) {
|
|
return new (alloc.allocate<ConcreteT>())
|
|
ConcreteT(std::forward<KeyTy>(key));
|
|
}
|
|
bool operator==(const KeyTy &key) const { return this->key == key; }
|
|
|
|
KeyTy key;
|
|
};
|
|
} // namespace
|
|
|
|
TEST(StorageUniquerTest, NonTrivialDestructor) {
|
|
struct NonTrivialStorage : public SimpleStorage<NonTrivialStorage, bool *> {
|
|
using Base::Base;
|
|
~NonTrivialStorage() {
|
|
bool *wasDestructed = std::get<0>(key);
|
|
*wasDestructed = true;
|
|
}
|
|
};
|
|
|
|
// Verify that the storage instance destructor was properly called.
|
|
bool wasDestructed = false;
|
|
{
|
|
StorageUniquer uniquer;
|
|
uniquer.registerParametricStorageType<NonTrivialStorage>();
|
|
NonTrivialStorage::get(uniquer, &wasDestructed);
|
|
}
|
|
|
|
EXPECT_TRUE(wasDestructed);
|
|
}
|