Future::unit()

[folly.git] / folly / futures / test / UnitTest.cpp

diff --git a/folly/futures/test/UnitTest.cpp b/folly/futures/test/UnitTest.cpp
index 930b1231ae1edf5ce48cea9fbe0743d613978e7f..965a830e98be6a6e7af4f8f06c4f3e172c7fab9b 100644 (file)
--- a/folly/futures/test/UnitTest.cpp
+++ b/folly/futures/test/UnitTest.cpp
@@ -14,12 +14,15 @@
  * limitations under the License.
  */
 
-#include <folly/futures/Future.h>
 #include <gtest/gtest.h>
 
+#include <folly/futures/Future.h>
+
 using namespace folly;
 
-TEST(Unit, FutureDefaultCtor) {
+std::runtime_error eggs("eggs");
+
+TEST(Unit, futureDefaultCtor) {
   Future<Unit>();
 }
 
@@ -29,21 +32,47 @@ TEST(Unit, voidOrUnit) {
   EXPECT_FALSE(is_void_or_unit<int>::value);
 }
 
-TEST(Unit, PromiseSetValue) {
+TEST(Unit, promiseSetValue) {
   Promise<Unit> p;
   p.setValue();
 }
 
-TEST(Unit, LiftInt) {
+TEST(Unit, liftInt) {
   using Lifted = Unit::Lift<int>;
   EXPECT_FALSE(Lifted::value);
   auto v = std::is_same<int, Lifted::type>::value;
   EXPECT_TRUE(v);
 }
 
-TEST(Unit, LiftVoid) {
+TEST(Unit, liftVoid) {
   using Lifted = Unit::Lift<void>;
   EXPECT_TRUE(Lifted::value);
   auto v = std::is_same<Unit, Lifted::type>::value;
   EXPECT_TRUE(v);
 }
+
+TEST(Unit, futureToUnit) {
+  Future<Unit> fu = makeFuture(42).unit();
+  fu.value();
+  EXPECT_TRUE(makeFuture<int>(eggs).unit().hasException());
+}
+
+TEST(Unit, voidFutureToUnit) {
+  Future<Unit> fu = makeFuture().unit();
+  fu.value();
+  EXPECT_TRUE(makeFuture<void>(eggs).unit().hasException());
+}
+
+TEST(Unit, unitFutureToUnitIdentity) {
+  Future<Unit> fu = makeFuture(Unit{}).unit();
+  fu.value();
+  EXPECT_TRUE(makeFuture<Unit>(eggs).unit().hasException());
+}
+
+TEST(Unit, toUnitWhileInProgress) {
+  Promise<int> p;
+  Future<Unit> fu = p.getFuture().unit();
+  EXPECT_FALSE(fu.isReady());
+  p.setValue(42);
+  EXPECT_TRUE(fu.isReady());
+}