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/wait_for_callback.h"
16
17 #include "absl/status/status.h"
18 #include "gtest/gtest.h"
19 #include "src/core/lib/promise/map.h"
20 #include "src/core/util/notification.h"
21 #include "test/core/promise/test_wakeup_schedulers.h"
22
23 namespace grpc_core {
24
TEST(WaitForCallbackTest,Works)25 TEST(WaitForCallbackTest, Works) {
26 WaitForCallback w4cb;
27 auto callback = w4cb.MakeCallback();
28 Notification done;
29 auto activity = MakeActivity(
30 [&w4cb]() {
31 return Map(w4cb.MakeWaitPromise(),
32 [](Empty) { return absl::OkStatus(); });
33 },
34 InlineWakeupScheduler{},
35 [&done](const absl::Status& s) {
36 EXPECT_TRUE(s.ok());
37 done.Notify();
38 });
39 EXPECT_FALSE(done.HasBeenNotified());
40 callback();
41 done.WaitForNotification();
42 }
43
44 } // namespace grpc_core
45
main(int argc,char ** argv)46 int main(int argc, char** argv) {
47 ::testing::InitGoogleTest(&argc, argv);
48 return RUN_ALL_TESTS();
49 }
50