test/core/promise/latch_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/latch.h"
  16
  17 #include <gmock/gmock.h>
  18 #include <gtest/gtest.h>
  19
  20 #include "src/core/lib/promise/join.h"
  21 #include "src/core/lib/promise/seq.h"
  22
  23 using testing::MockFunction;
  24 using testing::StrictMock;
  25
  26 namespace grpc_core {
  27
  28 TEST(LatchTest, Works) {
  29   Latch<int> latch;
  30   StrictMock<MockFunction<void(absl::Status)>> on_done;
  31   EXPECT_CALL(on_done, Call(absl::OkStatus()));
  32   MakeActivity(
  33       [&latch] {
  34         return Seq(Join(latch.Wait(),
  35                         [&latch]() {
  36                           latch.Set(42);
  37                           return true;
  38                         }),
  39                    [](std::tuple<int*, bool> result) {
  40                      EXPECT_EQ(*std::get<0>(result), 42);
  41                      return absl::OkStatus();
  42                    });
  43       },
  44       NoCallbackScheduler(),
  45       [&on_done](absl::Status status) { on_done.Call(std::move(status)); });
  46 }
  47
  48 }  // namespace grpc_core
  49
  50 int main(int argc, char** argv) {
  51   ::testing::InitGoogleTest(&argc, argv);
  52   return RUN_ALL_TESTS();
  53 }