SERVER-35682 Add SharedPromise and SharedSemiFuture types

1 files changed, 37 insertions, 1 deletions

diff --git a/src/mongo/util/future_test_future_int.cpp b/src/mongo/util/future_test_future_int.cpp
index 92955d367fb..899310cf69c 100644
--- a/src/mongo/util/future_test_future_int.cpp
+++ b/src/mongo/util/future_test_future_int.cpp
@@ -89,6 +89,17 @@ TEST(Future, Success_getNothrowRvalue) {
                         [](Future<int>&& fut) { ASSERT_EQ(std::move(fut).getNoThrow(), 1); });
 }
 
+TEST(Future, Success_shared_get) {
+    FUTURE_SUCCESS_TEST([] { return 1; },
+                        [](Future<int>&& fut) { ASSERT_EQ(std::move(fut).share().get(), 1); });
+}
+
+TEST(Future, Success_shared_getNothrow) {
+    FUTURE_SUCCESS_TEST(
+        [] { return 1; },
+        [](Future<int>&& fut) { ASSERT_EQ(std::move(fut).share().getNoThrow(), 1); });
+}
+
 TEST(Future, Success_getAsync) {
     FUTURE_SUCCESS_TEST(
         [] { return 1; },
@@ -129,6 +140,16 @@ TEST(Future, Fail_getNothrowRvalue) {
         [](Future<int>&& fut) { ASSERT_EQ(std::move(fut).getNoThrow(), failStatus()); });
 }
 
+TEST(Future, Fail_shared_get) {
+    FUTURE_FAIL_TEST<int>(
+        [](Future<int>&& fut) { ASSERT_THROWS_failStatus(std::move(fut).share().get()); });
+}
+
+TEST(Future, Fail_shared_getNothrow) {
+    FUTURE_FAIL_TEST<int>(
+        [](Future<int>&& fut) { ASSERT_EQ(std::move(fut).share().getNoThrow(), failStatus()); });
+}
+
 TEST(Future, Fail_getAsync) {
     FUTURE_FAIL_TEST<int>([](Future<int>&& fut) {
         auto pf = makePromiseFuture<int>();
@@ -203,9 +224,24 @@ TEST(Future, isReady_TSAN_OK) {
         done = true;
         return 1;
     });
+    //(void)*const_cast<volatile bool*>(&done);  // Data Race! Uncomment to make sure TSAN works.
     while (!fut.isReady()) {
     }
-    // ASSERT(done);  // Data Race! Uncomment to make sure TSAN is working.
+    ASSERT(done);
+    (void)fut.get();
+    ASSERT(done);
+}
+
+TEST(Future, isReady_shared_TSAN_OK) {
+    bool done = false;
+    auto fut = async([&] {
+                   done = true;
+                   return 1;
+               }).share();
+    //(void)*const_cast<volatile bool*>(&done);  // Data Race! Uncomment to make sure TSAN works.
+    while (!fut.isReady()) {
+    }
+    ASSERT(done);
     (void)fut.get();
     ASSERT(done);
 }