Remove InlineExecutor.cpp

[folly.git] / folly / test / CallOnceTest.cpp

diff --git a/folly/test/CallOnceTest.cpp b/folly/test/CallOnceTest.cpp
index 357b697397e2b5a578dfb2d4f18c89fa44f717ff..e0dccd81f87684fa84c2eea81994bac7d49f1eae 100644 (file)
--- a/folly/test/CallOnceTest.cpp
+++ b/folly/test/CallOnceTest.cpp
@@ -50,6 +50,23 @@ TEST(FollyCallOnce, Simple) {
   ASSERT_EQ(1, out);
 }
 
+TEST(FollyCallOnce, Exception) {
+  struct ExpectedException {};
+  folly::once_flag flag;
+  size_t numCalls = 0;
+  EXPECT_THROW(
+      folly::call_once(
+          flag,
+          [&] {
+            ++numCalls;
+            throw ExpectedException();
+          }),
+      ExpectedException);
+  EXPECT_EQ(1, numCalls);
+  folly::call_once(flag, [&] { ++numCalls; });
+  EXPECT_EQ(2, numCalls);
+}
+
 TEST(FollyCallOnce, Stress) {
   for (int i = 0; i < 100; ++i) {
     folly::once_flag flag;