mlir/unittests/Support/StorageUniquerTest.cpp

   1 //===- StorageUniquerTest.cpp - StorageUniquer Tests ----------------------===//
   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-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "mlir/Support/StorageUniquer.h"
  10 #include "gmock/gmock.h"
  11
  12 using namespace mlir;
  13
  14 namespace {
  15 /// Simple storage class used for testing.
  16 template <typename ConcreteT, typename... Args>
  17 struct SimpleStorage : public StorageUniquer::BaseStorage {
  18   using Base = SimpleStorage<ConcreteT, Args...>;
  19   using KeyTy = std::tuple<Args...>;
  20
  21   SimpleStorage(KeyTy key) : key(key) {}
  22
  23   /// Get an instance of this storage instance.
  24   template <typename... ParamsT>
  25   static ConcreteT *get(StorageUniquer &uniquer, ParamsT &&...params) {
  26     return uniquer.get<ConcreteT>(
  27         /*initFn=*/{}, std::make_tuple(std::forward<ParamsT>(params)...));
  28   }
  29
  30   /// Construct an instance with the given storage allocator.
  31   static ConcreteT *construct(StorageUniquer::StorageAllocator &alloc,
  32                               KeyTy key) {
  33     return new (alloc.allocate<ConcreteT>())
  34         ConcreteT(std::forward<KeyTy>(key));
  35   }
  36   bool operator==(const KeyTy &key) const { return this->key == key; }
  37
  38   KeyTy key;
  39 };
  40 } // namespace
  41
  42 TEST(StorageUniquerTest, NonTrivialDestructor) {
  43   struct NonTrivialStorage : public SimpleStorage<NonTrivialStorage, bool *> {
  44     using Base::Base;
  45     ~NonTrivialStorage() {
  46       bool *wasDestructed = std::get<0>(key);
  47       *wasDestructed = true;
  48     }
  49   };
  50
  51   // Verify that the storage instance destructor was properly called.
  52   bool wasDestructed = false;
  53   {
  54     StorageUniquer uniquer;
  55     uniquer.registerParametricStorageType<NonTrivialStorage>();
  56     NonTrivialStorage::get(uniquer, &wasDestructed);
  57   }
  58
  59   EXPECT_TRUE(wasDestructed);
  60 }