From 2d96b05403662273ed202f02cd64bf81c038aa37 Mon Sep 17 00:00:00 2001
From: Max Kellermann <mk@cm4all.com>
Date: Mon, 1 Mar 2021 23:05:58 +0100
Subject: [PATCH] test/util/TestException: add CheckFindRetrowNested() unit
 test

---
 test/util/TestException.cxx | 113 ++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)

diff --git a/test/util/TestException.cxx b/test/util/TestException.cxx
index bc0b50bf0..be4e557e5 100644
--- a/test/util/TestException.cxx
+++ b/test/util/TestException.cxx
@@ -49,3 +49,116 @@ TEST(ExceptionTest, DerivedError)
 
 	ASSERT_EQ(GetFullMessage(std::make_exception_ptr(DerivedError("Foo"))), "Foo");
 }
+
+template<typename T>
+static bool
+CheckFindRetrowNested(std::exception_ptr e) noexcept
+{
+	try {
+		FindRetrowNested<T>(e);
+	} catch (const T &) {
+		return true;
+	}
+
+	return false;
+}
+
+TEST(ExceptionTest, FindRetrowNestedDirect)
+{
+	struct Foo {};
+	struct Bar {};
+	struct Derived : Foo {};
+
+	try {
+		throw Foo{};
+	} catch (...) {
+		EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+	}
+
+	try {
+		throw Bar{};
+	} catch (...) {
+		EXPECT_FALSE(CheckFindRetrowNested<Foo>(std::current_exception()));
+	}
+
+	try {
+		throw Derived{};
+	} catch (...) {
+		EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+	}
+}
+
+TEST(ExceptionTest, FindRetrowNestedIndirect)
+{
+	struct Foo {};
+	struct Bar {};
+	struct Derived : Foo {};
+	struct Outer {};
+
+	try {
+		throw Foo{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(Outer{});
+		} catch (...) {
+			EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+
+	try {
+		throw Bar{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(Outer{});
+		} catch (...) {
+			EXPECT_FALSE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+
+	try {
+		throw Derived{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(Outer{});
+		} catch (...) {
+			EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+}
+
+TEST(ExceptionTest, FindRetrowNestedIndirectRuntimeError)
+{
+	struct Foo {};
+	struct Bar {};
+	struct Derived : Foo {};
+
+	try {
+		throw Foo{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(std::runtime_error("X"));
+		} catch (...) {
+			EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+
+	try {
+		throw Bar{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(std::runtime_error("X"));
+		} catch (...) {
+			EXPECT_FALSE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+
+	try {
+		throw Derived{};
+	} catch (...) {
+		try {
+			std::throw_with_nested(std::runtime_error("X"));
+		} catch (...) {
+			EXPECT_TRUE(CheckFindRetrowNested<Foo>(std::current_exception()));
+		}
+	}
+}