From: Remi Gacogne <remi.gacogne@powerdns.com>
Date: Tue, 9 Dec 2025 09:06:51 +0000 (+0100)
Subject: test-luawrapper: More complicated variant tests
X-Git-Tag: rec-5.4.0-beta1~57^2~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=1d17b0cf0310e0e375a157fd941552be19a137b1;p=thirdparty%2Fpdns.git

test-luawrapper: More complicated variant tests

Signed-off-by: Remi Gacogne <remi.gacogne@powerdns.com>
---

diff --git a/pdns/test-luawrapper.cc b/pdns/test-luawrapper.cc
index 9aae68dca8..b14dc3df66 100644
--- a/pdns/test-luawrapper.cc
+++ b/pdns/test-luawrapper.cc
@@ -70,28 +70,62 @@ BOOST_AUTO_TEST_CASE(test_std_optional)
   }
 }
 
+BOOST_AUTO_TEST_CASE(test_boost_variant)
+{
+  using MyVariantType = boost::variant<int, const std::string, std::string, std::string*>;
+
+  LuaContext context;
+  context.writeFunction("testVariant", [](MyVariantType incoming) -> MyVariantType {
+    return incoming;
+  });
+
+  {
+    auto result = context.executeCode<MyVariantType>("return testVariant(1)");
+    const auto* content = boost::get<int>(&result);
+    BOOST_REQUIRE(content);
+    BOOST_CHECK_EQUAL(*content, 1);
+  }
+
+  {
+    auto result = context.executeCode<MyVariantType>("return testVariant('foo')");
+    const auto* content = boost::get<const std::string>(&result);
+    BOOST_REQUIRE(content);
+    BOOST_CHECK_EQUAL(*content, "foo");
+  }
+
+  {
+    auto func = [&]() {
+      context.executeCode<MyVariantType>("return testVariant(nil)");
+    };
+    BOOST_CHECK_THROW(func(), LuaContext::ExecutionErrorException);
+  }
+}
+
+
 BOOST_AUTO_TEST_CASE(test_std_variant)
 {
+  using MyVariantType = std::variant<int, const std::string, std::string, std::string*>;
+
   LuaContext context;
-  context.writeFunction("testVariant", [](std::variant<int, std::string> incoming) -> std::variant<int, std::string> {
+  context.writeFunction("testVariant", [](MyVariantType incoming) -> MyVariantType {
     return incoming;
   });
 
   {
-    auto result = context.executeCode<std::variant<int, std::string>>("return testVariant(1)");
+    const auto result = context.executeCode<MyVariantType>("return testVariant(1)");
     BOOST_REQUIRE(std::holds_alternative<int>(result));
     BOOST_CHECK_EQUAL(std::get<int>(result), 1);
   }
 
   {
-    auto result = context.executeCode<std::variant<int, std::string>>("return testVariant('foo')");
-    BOOST_REQUIRE(std::holds_alternative<std::string>(result));
-    BOOST_CHECK_EQUAL(std::get<std::string>(result), "foo");
+    const auto result = context.executeCode<MyVariantType>("return testVariant('foo')");
+    BOOST_REQUIRE(std::holds_alternative<const std::string>(result));
+    BOOST_CHECK_EQUAL(std::get<const std::string>(result), "foo");
   }
 
   {
     auto func = [&]() {
-      context.executeCode<std::variant<int, std::string>>("return testVariant(nil)");
+      context.executeCode<MyVariantType>("return testVariant(nil)");
     };
     BOOST_CHECK_THROW(func(), LuaContext::ExecutionErrorException);
   }