folly/futures/test/MapTest.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 <folly/futures/Future.h>
  18 #include <folly/portability/GTest.h>
  19
  20 using namespace folly;
  21
  22 TEST(Map, basic) {
  23   Promise<int> p1;
  24   Promise<int> p2;
  25   Promise<int> p3;
  26
  27   std::vector<Future<int>> fs;
  28   fs.push_back(p1.getFuture());
  29   fs.push_back(p2.getFuture());
  30   fs.push_back(p3.getFuture());
  31
  32   int c = 0;
  33   std::vector<Future<Unit>> fs2 = futures::map(fs, [&](int i){
  34     c += i;
  35   });
  36
  37   // Ensure we call the callbacks as the futures complete regardless of order
  38   p2.setValue(1);
  39   EXPECT_EQ(1, c);
  40   p3.setValue(1);
  41   EXPECT_EQ(2, c);
  42   p1.setValue(1);
  43   EXPECT_EQ(3, c);
  44
  45   EXPECT_TRUE(collect(fs2).isReady());
  46 }