folly/futures/test/MapTest.cpp

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