[Lldb-commits] [lldb] 33f65f3 - [lldb] Add a basic unit test for the SharedCluster class

[Pavel Labath via lldb-commits]

Author: Pavel Labath
Date: 2020-01-31T15:46:41+01:00
New Revision: 33f65f393f6652a0974582f7b13a9aa38a9929a3

URL: https://github.com/llvm/llvm-project/commit/33f65f393f6652a0974582f7b13a9aa38a9929a3
DIFF: https://github.com/llvm/llvm-project/commit/33f65f393f6652a0974582f7b13a9aa38a9929a3.diff

LOG: [lldb] Add a basic unit test for the SharedCluster class

Added: 
    lldb/unittests/Utility/SharedClusterTest.cpp

Modified: 
    lldb/unittests/Utility/CMakeLists.txt

Removed: 
    


################################################################################
diff  --git a/lldb/unittests/Utility/CMakeLists.txt b/lldb/unittests/Utility/CMakeLists.txt
index 8dc13327b9b0..5e27c6725d4b 100644
--- a/lldb/unittests/Utility/CMakeLists.txt
+++ b/lldb/unittests/Utility/CMakeLists.txt
@@ -24,6 +24,7 @@ add_lldb_unittest(UtilityTests
   ReproducerInstrumentationTest.cpp
   ReproducerTest.cpp
   ScalarTest.cpp
+  SharedClusterTest.cpp
   StateTest.cpp
   StatusTest.cpp
   StreamTeeTest.cpp

diff  --git a/lldb/unittests/Utility/SharedClusterTest.cpp b/lldb/unittests/Utility/SharedClusterTest.cpp
new file mode 100644
index 000000000000..a41b12aa18d8
--- /dev/null
+++ b/lldb/unittests/Utility/SharedClusterTest.cpp
@@ -0,0 +1,55 @@
+//===-- SharedClusterTest.cpp ---------------------------------------------===//
+//
+// 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 "lldb/Utility/SharedCluster.h"
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+
+using namespace lldb_private;
+
+namespace {
+class DestructNotifier {
+public:
+  DestructNotifier(std::vector<int> &Queue, int Key) : Queue(Queue), Key(Key) {}
+  ~DestructNotifier() { Queue.push_back(Key); }
+
+  std::vector<int> &Queue;
+  const int Key;
+};
+} // namespace
+
+TEST(SharedCluster, ClusterManager) {
+  std::vector<int> Queue;
+  auto *CM = new ClusterManager<DestructNotifier>();
+  auto *One = new DestructNotifier(Queue, 1);
+  auto *Two = new DestructNotifier(Queue, 2);
+  CM->ManageObject(One);
+  CM->ManageObject(Two);
+
+  ASSERT_THAT(Queue, testing::IsEmpty());
+  {
+    SharingPtr<DestructNotifier> OnePtr = CM->GetSharedPointer(One);
+    ASSERT_EQ(OnePtr->Key, 1);
+    ASSERT_THAT(Queue, testing::IsEmpty());
+
+    {
+      SharingPtr<DestructNotifier> OnePtrCopy = OnePtr;
+      ASSERT_EQ(OnePtrCopy->Key, 1);
+      ASSERT_THAT(Queue, testing::IsEmpty());
+    }
+
+    {
+      SharingPtr<DestructNotifier> TwoPtr = CM->GetSharedPointer(Two);
+      ASSERT_EQ(TwoPtr->Key, 2);
+      ASSERT_THAT(Queue, testing::IsEmpty());
+    }
+
+    ASSERT_THAT(Queue, testing::IsEmpty());
+  }
+  ASSERT_THAT(Queue, testing::ElementsAre(1, 2));
+}