refactor the test and fix alignment issue

Author: kerambyte

llvm/unittests/ADT/CountCopyAndMove.cpp

@@ -10,6 +10,7 @@
 
 using namespace llvm;
 
+int CountCopyAndMove::Constructions = 0;
 int CountCopyAndMove::CopyConstructions = 0;
 int CountCopyAndMove::CopyAssignments = 0;
 int CountCopyAndMove::MoveConstructions = 0;

llvm/unittests/ADT/CountCopyAndMove.h

@@ -12,15 +12,16 @@
 namespace llvm {
 
 struct CountCopyAndMove {
+  static int Constructions;
   static int CopyConstructions;
   static int CopyAssignments;
   static int MoveConstructions;
   static int MoveAssignments;
   static int Destructions;
   int val;
 
-  CountCopyAndMove() = default;
-  explicit CountCopyAndMove(int val) : val(val) {}
+  CountCopyAndMove() { ++Constructions; }
+  explicit CountCopyAndMove(int val) : val(val) { ++Constructions; }
   CountCopyAndMove(const CountCopyAndMove &other) : val(other.val) {
     ++CopyConstructions;
   }
@@ -40,13 +41,18 @@ struct CountCopyAndMove {
   ~CountCopyAndMove() { ++Destructions; }
 
   static void ResetCounts() {
+    Constructions = 0;
     CopyConstructions = 0;
     CopyAssignments = 0;
     MoveConstructions = 0;
     MoveAssignments = 0;
     Destructions = 0;
   }
 
+  static int TotalConstructions() {
+    return Constructions + MoveConstructions + CopyConstructions;
+  }
+
   static int TotalCopies() { return CopyConstructions + CopyAssignments; }
 
   static int TotalMoves() { return MoveConstructions + MoveAssignments; }

llvm/unittests/ADT/FunctionExtrasTest.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/ADT/FunctionExtras.h"
+#include "CountCopyAndMove.h"
 #include "gtest/gtest.h"
 
 #include <memory>
@@ -332,20 +333,15 @@ TEST(UniqueFunctionTest, InlineStorageWorks) {
 // Check that the moved-from captured state is properly destroyed during
 // move construction/assignment.
 TEST(UniqueFunctionTest, MovedFromStateIsDestroyedCorrectly) {
-  static int NumOfMovesCalled = 0;
-  static int NumOfDestructorsCalled = 0;
-  struct State {
-    State() = default;
-    State(State &&) { ++NumOfMovesCalled; }
-    ~State() { ++NumOfDestructorsCalled; }
-  };
+  CountCopyAndMove::ResetCounts();
   {
-    unique_function<void()> CapturingFunction{[state = State{}] {}};
+    unique_function<void()> CapturingFunction{
+        [Counter = CountCopyAndMove{}] {}};
     unique_function<void()> CapturingFunctionMoved{
         std::move(CapturingFunction)};
   }
-  printf("%i, %i\n", NumOfMovesCalled, NumOfDestructorsCalled);
-  EXPECT_EQ(NumOfDestructorsCalled, 1 + NumOfMovesCalled);
+  EXPECT_EQ(CountCopyAndMove::TotalConstructions(),
+            CountCopyAndMove::Destructions);
 }
 
 } // anonymous namespace