Enable -Wextra-semi

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

diff --git a/folly/test/PartialTest.cpp b/folly/test/PartialTest.cpp
index 025cd25f51cfd54d9db97ec59d22069a894d51a1..cbe1851bbc9d121b6ab7ac3f83401dbe205643aa 100644 (file)
--- a/folly/test/PartialTest.cpp
+++ b/folly/test/PartialTest.cpp
@@ -16,8 +16,10 @@
 
 #include <memory>
 
+#include <folly/Function.h>
 #include <folly/Partial.h>
-#include <gtest/gtest.h>
+
+#include <folly/portability/GTest.h>
 
 using folly::partial;
 
@@ -131,3 +133,15 @@ TEST(Partial, MoveOnly) {
 
   EXPECT_EQ(560, *result);
 }
+
+TEST(Partial, WrapInStdFunction) {
+  auto p1 = partial(&add3, 2);
+  std::function<int(int, int)> func = p1;
+  EXPECT_EQ(234, func(3, 4));
+}
+
+TEST(Partial, WrapInFollyFunction) {
+  auto p1 = partial(&add3, 2);
+  folly::Function<int(int, int)> func = p1;
+  EXPECT_EQ(234, func(3, 4));
+}