From efd5318b2d9132996bf35a6af2e82706665890ff Mon Sep 17 00:00:00 2001
From: Benoit Germain <benoit.germain@ubisoft.com>
Date: Mon, 16 Dec 2024 09:50:52 +0100
Subject: Unit tests for thread name

---
 unit_tests/lane_tests.cpp | 61 +++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 56 insertions(+), 5 deletions(-)

(limited to 'unit_tests')

diff --git a/unit_tests/lane_tests.cpp b/unit_tests/lane_tests.cpp
index 14a327f..8b22fcd 100644
--- a/unit_tests/lane_tests.cpp
+++ b/unit_tests/lane_tests.cpp
@@ -104,17 +104,16 @@ TEST_F(LaneTests, UncooperativeShutdown)
     lua_pushcfunction(S, _finallyCB);
     lua_setglobal(S, "finallyCB");
     // start a lane that lasts a long time
-    std::string_view const script{
+    std::string_view const _script{
         " lanes.finally(finallyCB)"
-        " print ('in Master')"
-        " f = lanes.gen('*',"
+        " g = lanes.gen('*',"
         "     {name = 'auto'},"
         "     function()"
         "         for i = 1,1e37 do end" // no cooperative cancellation checks here!
         "     end)"
-        " f()"
+        " g()"
     };
-    ASSERT_EQ(S.doString(script), LuaError::OK) << S;
+    ASSERT_EQ(S.doString(_script), LuaError::OK) << S;
     // close the state before the lane ends.
     // since we don't wait at all, it is possible that the OS thread for the lane hasn't even started at that point
     S.close();
@@ -123,6 +122,58 @@ TEST_F(LaneTests, UncooperativeShutdown)
     ASSERT_EQ(_allLanesTerminated, true) << S;
 }
 
+// #################################################################################################
+
+TEST_F(LaneTests, DefaultThreadNameIsUnnamed)
+{
+    std::string_view const _script{
+        " g = lanes.gen('*',"
+        "     function()"
+        "         return lane_threadname()"
+        "     end)"
+        " h = g()"
+        " local tn = h[1]"
+        " assert(tn == h:get_threadname())"
+        " return tn"
+    };
+    ASSERT_EQ(S.doStringAndRet(_script), "<unnamed>") << S;
+}
+
+// #################################################################################################
+
+TEST_F(LaneTests, UserThreadNameFromGenerator)
+{
+    std::string_view const _script{
+        " g = lanes.gen('*',"
+        "     { name = 'user name'},"
+        "     function()"
+        "         return lane_threadname()"
+        "     end)"
+        " h = g()"
+        " local tn = h[1]"
+        " assert(tn == h:get_threadname())"
+        " return tn"
+    };
+    ASSERT_EQ(S.doStringAndRet(_script), "user name") << S;
+}
+
+// #################################################################################################
+
+TEST_F(LaneTests, UserThreadNameFromInside)
+{
+    std::string_view const _script{
+        " g = lanes.gen('*',"
+        "     function()"
+        "         lane_threadname('user name')"
+        "         return true"
+        "     end)"
+        " h = g()"
+        " h:join()"
+        " return h:get_threadname()"
+    };
+    ASSERT_EQ(S.doStringAndRet(_script), "user name") << S;
+}
+
 // #################################################################################################
 // #################################################################################################
 
-- 
cgit v1.2.3-55-g6feb