Convert mp_unit tests from gtest to framework.hpp

- Modified test/mp_unit/mp_unit_tests.hpp to use ../framework.hpp instead of gtest/gtest.h
- Enhanced test/framework.hpp with GTest-compatible APIs:
  - Added Environment base class for global test setup/teardown
  - Added TestInfo and UnitTest classes for test metadata access
  - Added GTEST_SKIP macro support via SkipHelper class
  - Added namespace alias 'testing' for compatibility
  - Added InitGoogleTest and AddGlobalTestEnvironment helper functions
- Updated test/framework.cc with implementations for new classes
- All mp_unit test files now use framework.hpp through mp_unit_tests.hpp
- Formatting applied via lint.sh

Co-authored-by: chhwang <8018170+chhwang@users.noreply.github.com>

Author: Copilot

test/executor_test.cc

@@ -93,11 +93,8 @@ double benchTime(int rank, std::shared_ptr<mscclpp::Bootstrap> bootstrap, std::s
 
 int main(int argc, char* argv[]) {
   if (argc != 5 && argc != 6) {
-    std::cerr << "Usage: " << argv[0] << " <buffer size>"
-              << " <execution plan path>"
-              << " <number of iterations>"
-              << " <number of graph iterations>"
-              << " (optional) <packet type>" << std::endl;
+    std::cerr << "Usage: " << argv[0] << " <buffer size>" << " <execution plan path>" << " <number of iterations>"
+              << " <number of graph iterations>" << " (optional) <packet type>" << std::endl;
     return 1;
   }
 

test/framework.cc

@@ -161,26 +161,51 @@ int runMultipleTests(
 
 }  // namespace utils
 
+// UnitTest implementation
+UnitTest* UnitTest::GetInstance() {
+  static UnitTest instance;
+  return &instance;
+}
+
 // TestRegistry implementation
 TestRegistry& TestRegistry::instance() {
   static TestRegistry registry;
   return registry;
 }
 
 void TestRegistry::registerTest(const std::string& test_suite, const std::string& test_name, TestFactory factory) {
-  TestInfo info;
+  TestInfoInternal info;
   info.suite_name = test_suite;
   info.test_name = test_name;
   info.factory = factory;
   tests_.push_back(info);
 }
 
+void TestRegistry::addGlobalTestEnvironment(Environment* env) { environments_.push_back(env); }
+
+void TestRegistry::initGoogleTest(int* argc, char** argv) {
+  // Parse command-line arguments if needed
+  // For now, this is a no-op placeholder for compatibility
+}
+
 int TestRegistry::runAllTests(int argc, char* argv[]) {
   // Initialize MPI if not already initialized
   if (!g_mpi_initialized) {
     utils::initializeMPI(argc, argv);
   }
 
+  // Set up global test environments
+  for (auto* env : environments_) {
+    try {
+      env->SetUp();
+    } catch (const std::exception& e) {
+      if (g_mpi_rank == 0) {
+        std::cerr << "Failed to set up test environment: " << e.what() << std::endl;
+      }
+      return 1;
+    }
+  }
+
   int passed = 0;
   int failed = 0;
 
@@ -196,6 +221,10 @@ int TestRegistry::runAllTests(int argc, char* argv[]) {
       std::cout << "[ RUN      ] " << test_info.suite_name << "." << test_info.test_name << std::endl;
     }
 
+    // Set current test info for UnitTest::GetInstance()->current_test_info()
+    TestInfo current_info(test_info.suite_name, test_info.test_name);
+    UnitTest::GetInstance()->set_current_test_info(&current_info);
+
     TestCase* test_case = nullptr;
     try {
       test_case = test_info.factory();
@@ -216,6 +245,9 @@ int TestRegistry::runAllTests(int argc, char* argv[]) {
 
     delete test_case;
 
+    // Clear current test info
+    UnitTest::GetInstance()->set_current_test_info(nullptr);
+
     // Synchronize test status across all MPI processes
     int local_passed = g_current_test_passed ? 1 : 0;
     int global_passed = 1;
@@ -246,6 +278,17 @@ int TestRegistry::runAllTests(int argc, char* argv[]) {
     }
   }
 
+  // Tear down global test environments (in reverse order)
+  for (auto it = environments_.rbegin(); it != environments_.rend(); ++it) {
+    try {
+      (*it)->TearDown();
+    } catch (const std::exception& e) {
+      if (g_mpi_rank == 0) {
+        std::cerr << "Failed to tear down test environment: " << e.what() << std::endl;
+      }
+    }
+  }
+
   return failed > 0 ? 1 : 0;
 }
 

test/framework.hpp

@@ -4,15 +4,14 @@
 #ifndef MSCCLPP_MP_UNIT_TESTS_HPP_
 #define MSCCLPP_MP_UNIT_TESTS_HPP_
 
-#include <gtest/gtest.h>
-
 #include <mscclpp/core.hpp>
 #include <mscclpp/executor.hpp>
 #include <mscclpp/memory_channel.hpp>
 #include <mscclpp/packet_device.hpp>
 #include <mscclpp/port_channel.hpp>
 #include <mscclpp/utils.hpp>
 
+#include "../framework.hpp"
 #include "ib.hpp"
 #include "utils_internal.hpp"
 

test/mp_unit/mp_unit_tests.hpp

@@ -12,15 +12,15 @@ namespace mscclpp {
 namespace test {
 
 // Global state for performance test results
-static std::vector<struct PerfTestResult {
+static std::vector < struct PerfTestResult {
   std::string test_name;
   std::string test_category;
   std::map<std::string, std::string> test_params;
   nlohmann::ordered_json metrics;
   int num_processes;
   int process_rank;
   std::string timestamp;
-}> g_perf_results;
+} > g_perf_results;
 
 static std::string getCurrentTimestamp() {
   auto now = std::chrono::system_clock::now();

test/perf/framework.cc

@@ -7,10 +7,10 @@
 // This file is kept for backwards compatibility with perf tests
 // The actual framework is now in test/framework.hpp
 
-#include "../framework.hpp"
-
 #include <nlohmann/json.hpp>
 
+#include "../framework.hpp"
+
 namespace mscclpp {
 namespace test {
 

test/perf/framework.hpp

@@ -2,10 +2,11 @@
 // Licensed under the MIT license.
 
 #include <gmock/gmock.h>
-#include "../framework.hpp"
 
 #include <mscclpp/core.hpp>
 
+#include "../framework.hpp"
+
 class LocalCommunicatorTest : public ::testing::Test {
  protected:
   void SetUp() override {

test/unit/core_tests.cc

@@ -1,10 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
-#include "../framework.hpp"
-
 #include <mscclpp/errors.hpp>
 
+#include "../framework.hpp"
+
 TEST(ErrorsTest, SystemError) {
   mscclpp::Error error("test", mscclpp::ErrorCode::SystemError);
   EXPECT_EQ(error.getErrorCode(), mscclpp::ErrorCode::SystemError);

test/unit/errors_tests.cc

@@ -1,13 +1,12 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
-#include "../framework.hpp"
-
 #include <mscclpp/fifo.hpp>
 #include <mscclpp/gpu_utils.hpp>
 #include <mscclpp/numa.hpp>
 #include <mscclpp/utils.hpp>
 
+#include "../framework.hpp"
 #include "utils_internal.hpp"
 
 #define ITER 10000  // should be larger than the FIFO size for proper testing

test/unit/fifo_tests.cu

@@ -1,10 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
-#include "../framework.hpp"
-
 #include <mscclpp/gpu_utils.hpp>
 
+#include "../framework.hpp"
+
 TEST(GpuUtilsTest, StreamPool) {
   auto streamPool = mscclpp::gpuStreamPool();
   cudaStream_t s;