test/core/promise/loop_test.cc

   1 // Copyright 2021 gRPC authors.
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 #include "src/core/lib/promise/loop.h"
  16
  17 #include <gtest/gtest.h>
  18
  19 namespace grpc_core {
  20
  21 TEST(LoopTest, CountToFive) {
  22   int i = 0;
  23   Loop([&i]() -> LoopCtl<int> {
  24     i++;
  25     if (i < 5) return Continue();
  26     return i;
  27   })();
  28   EXPECT_EQ(i, 5);
  29 }
  30
  31 TEST(LoopTest, FactoryCountToFive) {
  32   int i = 0;
  33   Loop([&i]() {
  34     return [&i]() -> LoopCtl<int> {
  35       i++;
  36       if (i < 5) return Continue();
  37       return i;
  38     };
  39   })();
  40   EXPECT_EQ(i, 5);
  41 }
  42
  43 }  // namespace grpc_core
  44
  45 int main(int argc, char** argv) {
  46   ::testing::InitGoogleTest(&argc, argv);
  47   return RUN_ALL_TESTS();
  48 }