restore fail func

google · Feb 5, 2024 · 7c402e2 · 7c402e2
1 parent 8d2b59b
commit 7c402e2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/logging_unittest.cc b/src/logging_unittest.cc
@@ -1582,6 +1582,8 @@ TEST(Logging, FatalThrow) {
 #  endif  // __has_attribute(noreturn)
 #endif    // defined(__has_attribute)
                              { throw std::logic_error{"fail"}; });
-  EXPECT_THROW(LOG(FATAL) << "must throw to fail", std::logic_error);
-  InstallFailureFunction(fail_func);
+  auto restore_fail = [fail_func] { InstallFailureFunction(fail_func); };
+  ScopedExit<decltype(restore_fail)> restore{restore_fail};
+  EXPECT_THROW({ LOG(FATAL) << "must throw to fail"; }, std::logic_error);
+
 }