xref: /aosp_15_r20/external/grpc-grpc/test/core/promise/exec_ctx_wakeup_scheduler_test.cc (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
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/exec_ctx_wakeup_scheduler.h"
16 
17 #include <stdlib.h>
18 
19 #include <memory>
20 
21 #include "absl/status/status.h"
22 #include "gtest/gtest.h"
23 
24 #include "src/core/lib/promise/activity.h"
25 #include "src/core/lib/promise/poll.h"
26 
27 namespace grpc_core {
28 
TEST(ExecCtxWakeupSchedulerTest,Works)29 TEST(ExecCtxWakeupSchedulerTest, Works) {
30   int state = 0;
31   bool done = false;
32   auto activity = MakeActivity(
33       [&state]() mutable -> Poll<absl::Status> {
34         ++state;
35         switch (state) {
36           case 1:
37             return Pending();
38           case 2:
39             return absl::OkStatus();
40           default:
41             abort();
42         }
43       },
44       ExecCtxWakeupScheduler(),
45       [&done](absl::Status status) {
46         EXPECT_EQ(status, absl::OkStatus());
47         done = true;
48       });
49 
50   EXPECT_EQ(state, 1);
51   EXPECT_FALSE(done);
52   {
53     ExecCtx exec_ctx;
54     EXPECT_FALSE(exec_ctx.HasWork());
55     activity->ForceWakeup();
56     EXPECT_TRUE(exec_ctx.HasWork());
57     EXPECT_EQ(state, 1);
58     EXPECT_FALSE(done);
59   }
60   EXPECT_EQ(state, 2);
61   EXPECT_TRUE(done);
62 }
63 
64 }  // namespace grpc_core
65 
main(int argc,char ** argv)66 int main(int argc, char** argv) {
67   ::testing::InitGoogleTest(&argc, argv);
68   return RUN_ALL_TESTS();
69 }
70