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<int, 1>([] { return 25; }),
28 Case<int, 2>([] { return 95; }),
29 Case<int, 3>([] { return 68; }), Default([] { return 52; }));
30 };
31 EXPECT_EQ(test_switch(0)(), Poll<int>(52));
32 EXPECT_EQ(test_switch(1)(), Poll<int>(25));
33 EXPECT_EQ(test_switch(2)(), Poll<int>(95));
34 EXPECT_EQ(test_switch(3)(), Poll<int>(68));
35 EXPECT_EQ(test_switch(4)(), Poll<int>(52));
36 }
37
TEST(SwitchTest,Pending)38 TEST(SwitchTest, Pending) {
39 auto test_switch = [](int d) {
40 return Switch(d, Case<int, 42>([]() -> Poll<int> { return Pending{}; }),
41 Case<int, 1>([] { return 25; }),
42 Case<int, 2>([] { return 95; }),
43 Case<int, 3>([] { return 68; }), Default([] { return 52; }));
44 };
45 EXPECT_EQ(test_switch(0)(), Poll<int>(52));
46 EXPECT_EQ(test_switch(1)(), Poll<int>(25));
47 EXPECT_EQ(test_switch(2)(), Poll<int>(95));
48 EXPECT_EQ(test_switch(3)(), Poll<int>(68));
49 EXPECT_EQ(test_switch(4)(), Poll<int>(52));
50 EXPECT_EQ(test_switch(42)(), Poll<int>(Pending{}));
51 }
52
TEST(SwitchTest,ThreeCasesFromEnum)53 TEST(SwitchTest, ThreeCasesFromEnum) {
54 enum class X : uint8_t { A, B, C };
55
56 auto test_switch = [](X d) {
57 return Switch(d, Case<X, X::A>([] { return 25; }),
58 Case<X, X::B>([] { return 95; }),
59 Case<X, X::C>([] { return 68; }), Default([] { return 52; }));
60 };
61 EXPECT_EQ(test_switch(X::A)(), Poll<int>(25));
62 EXPECT_EQ(test_switch(X::B)(), Poll<int>(95));
63 EXPECT_EQ(test_switch(X::C)(), Poll<int>(68));
64 }
65
66 } // namespace grpc_core
67
main(int argc,char ** argv)68 int main(int argc, char** argv) {
69 ::testing::InitGoogleTest(&argc, argv);
70 return RUN_ALL_TESTS();
71 }
72