1 // Copyright 2023 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/switch.h"
16
17 #include "gtest/gtest.h"
18
19 namespace grpc_core {
20
TEST(SwitchTest,JustDefault)21 TEST(SwitchTest, JustDefault) {
22 EXPECT_EQ(Switch(42, Default([] { return 1; }))(), Poll<int>(1));
23 }
24
TEST(SwitchTest,ThreeCases)25 TEST(SwitchTest, ThreeCases) {
26 auto test_switch = [](int d) {
27 return Switch(d, Case(1, [] { return 25; }), Case(2, [] { return 95; }),
28 Case(3, [] { return 68; }), Default([] { return 52; }));
29 };
30 EXPECT_EQ(test_switch(0)(), Poll<int>(52));
31 EXPECT_EQ(test_switch(1)(), Poll<int>(25));
32 EXPECT_EQ(test_switch(2)(), Poll<int>(95));
33 EXPECT_EQ(test_switch(3)(), Poll<int>(68));
34 EXPECT_EQ(test_switch(4)(), Poll<int>(52));
35 }
36
TEST(SwitchTest,Pending)37 TEST(SwitchTest, Pending) {
38 auto test_switch = [](int d) {
39 return Switch(d, Case(42, []() -> Poll<int> { return Pending{}; }),
40 Case(1, [] { return 25; }), Case(2, [] { return 95; }),
41 Case(3, [] { return 68; }), Default([] { return 52; }));
42 };
43 EXPECT_EQ(test_switch(0)(), Poll<int>(52));
44 EXPECT_EQ(test_switch(1)(), Poll<int>(25));
45 EXPECT_EQ(test_switch(2)(), Poll<int>(95));
46 EXPECT_EQ(test_switch(3)(), Poll<int>(68));
47 EXPECT_EQ(test_switch(4)(), Poll<int>(52));
48 EXPECT_EQ(test_switch(42)(), Poll<int>(Pending{}));
49 }
50
TEST(SwitchTest,ThreeCasesFromEnum)51 TEST(SwitchTest, ThreeCasesFromEnum) {
52 enum class X : uint8_t { A, B, C };
53
54 auto test_switch = [](X d) {
55 return Switch(d, Case(X::A, [] { return 25; }),
56 Case(X::B, [] { return 95; }), Case(X::C, [] { return 68; }),
57 Default([] { return 52; }));
58 };
59 EXPECT_EQ(test_switch(X::A)(), Poll<int>(25));
60 EXPECT_EQ(test_switch(X::B)(), Poll<int>(95));
61 EXPECT_EQ(test_switch(X::C)(), Poll<int>(68));
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