Added assert_fatal test

coder137 · coder137 · commit 67a74af3b2d7 · 2022-04-21T20:51:46.000-07:00
diff --git a/buildcc/lib/env/CMakeLists.txt b/buildcc/lib/env/CMakeLists.txt
@@ -42,11 +42,15 @@ if (${TESTING})
     add_executable(test_storage test/test_storage.cpp)
     target_link_libraries(test_storage PRIVATE mock_env)
 
+    add_executable(test_assert_fatal test/test_assert_fatal.cpp)
+    target_link_libraries(test_assert_fatal PRIVATE mock_env)
+
     add_test(NAME test_static_project COMMAND test_static_project)
     add_test(NAME test_env_util COMMAND test_env_util)
     add_test(NAME test_task_state COMMAND test_task_state)
     add_test(NAME test_command COMMAND test_command)
     add_test(NAME test_storage COMMAND test_storage)
+    add_test(NAME test_assert_fatal COMMAND test_assert_fatal)
 endif()
 
 set(ENV_SRCS
diff --git a/buildcc/lib/env/test/test_assert_fatal.cpp b/buildcc/lib/env/test/test_assert_fatal.cpp
@@ -0,0 +1,72 @@
+#include <thread>
+
+#include "taskflow/taskflow.hpp"
+
+// NOTE, Make sure all these includes are AFTER the system and header includes
+#include "CppUTest/CommandLineTestRunner.h"
+#include "CppUTest/MemoryLeakDetectorNewMacros.h"
+#include "CppUTest/TestHarness.h"
+#include "CppUTest/Utest.h"
+#include "CppUTestExt/MockSupport.h"
+
+namespace {
+
+std::thread::id my_main_thread = std::this_thread::get_id();
+
+bool IsRunningInThread() {
+  bool threaded = true;
+  if (std::this_thread::get_id() == my_main_thread) {
+    threaded = false;
+  }
+  return threaded;
+}
+
+void assert_handle_fatal() {
+  if (IsRunningInThread()) {
+    mock().actualCall("assert_handle_fatal_threaded");
+  } else {
+    mock().actualCall("assert_handle_fatal_main");
+  }
+}
+
+} // namespace
+
+// clang-format off
+TEST_GROUP(AssertFatalTestGroup)
+{
+  void teardown() {
+      mock().clear();
+  }
+};
+// clang-format on
+
+TEST(AssertFatalTestGroup, AssertFatal_IsThreadedCheck) {
+  CHECK_FALSE(IsRunningInThread());
+
+  tf::Taskflow tf;
+  tf.emplace([]() { CHECK_TRUE(IsRunningInThread()); });
+
+  tf::Executor ex(1);
+  ex.run(tf);
+  ex.wait_for_all();
+}
+
+TEST(AssertFatalTestGroup, AssertFatal_Threaded) {
+  mock().expectOneCall("assert_handle_fatal_threaded");
+
+  tf::Taskflow tf;
+  tf.emplace([]() { assert_handle_fatal(); });
+
+  tf::Executor ex(1);
+  ex.run(tf);
+  ex.wait_for_all();
+}
+
+TEST(AssertFatalTestGroup, AssertFatal_NotThreaded) {
+  mock().expectOneCall("assert_handle_fatal_main");
+  assert_handle_fatal();
+}
+
+int main(int ac, char **av) {
+  return CommandLineTestRunner::RunAllTests(ac, av);
+}
