3d6bdc84b85ea4e20abdfbc3ea36555f674e58de
[folly.git] / folly / futures / test / WhenTest.cpp
1 /*
2  * Copyright 2017 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <memory>
18 #include <mutex>
19
20 #include <folly/futures/Future.h>
21 #include <folly/portability/GTest.h>
22
23 using namespace folly;
24
25 TEST(When, predicateFalse) {
26   int i = 0;
27   auto thunk = [&] {
28     return makeFuture().then([&] { i += 1; });
29   };
30
31   // false
32   auto f1 = folly::when(false, thunk);
33   f1.wait();
34   EXPECT_EQ(0, i);
35 }
36
37 TEST(When, predicateTrue) {
38   int i = 0;
39   auto thunk = [&] {
40     return makeFuture().then([&] { i += 1; });
41   };
42
43   // true
44   auto f2 = folly::when(true, thunk);
45   f2.wait();
46   EXPECT_EQ(1, i);
47 }