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