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 #include "src/core/lib/event_engine/query_extensions.h"
15
16 #include <string>
17
18 #include "absl/functional/any_invocable.h"
19 #include "absl/status/status.h"
20 #include "gtest/gtest.h"
21
22 #include <grpc/event_engine/event_engine.h>
23 #include <grpc/event_engine/slice_buffer.h>
24 #include <grpc/support/port_platform.h>
25
26 #include "src/core/lib/gprpp/crash.h"
27
28 namespace grpc_event_engine {
29 namespace experimental {
30 namespace {
31
32 template <int i>
33 class TestExtension {
34 public:
35 TestExtension() = default;
36 ~TestExtension() = default;
37
EndpointExtensionName()38 static std::string EndpointExtensionName() {
39 return "grpc.test.test_extension" + std::to_string(i);
40 }
41
GetValue() const42 int GetValue() const { return val_; }
43
44 private:
45 int val_ = i;
46 };
47
48 class ExtendedTestEndpoint
49 : public ExtendedType<EventEngine::Endpoint, TestExtension<0>,
50 TestExtension<1>, TestExtension<2>> {
51 public:
52 ExtendedTestEndpoint() = default;
53 ~ExtendedTestEndpoint() override = default;
Read(absl::AnyInvocable<void (absl::Status)>,SliceBuffer *,const ReadArgs *)54 bool Read(absl::AnyInvocable<void(absl::Status)> /*on_read*/,
55 SliceBuffer* /*buffer*/, const ReadArgs* /*args*/) override {
56 grpc_core::Crash("Not implemented");
57 };
Write(absl::AnyInvocable<void (absl::Status)>,SliceBuffer *,const WriteArgs *)58 bool Write(absl::AnyInvocable<void(absl::Status)> /*on_writable*/,
59 SliceBuffer* /*data*/, const WriteArgs* /*args*/) override {
60 grpc_core::Crash("Not implemented");
61 }
62 /// Returns an address in the format described in DNSResolver. The returned
63 /// values are expected to remain valid for the life of the Endpoint.
GetPeerAddress() const64 const EventEngine::ResolvedAddress& GetPeerAddress() const override {
65 grpc_core::Crash("Not implemented");
66 }
GetLocalAddress() const67 const EventEngine::ResolvedAddress& GetLocalAddress() const override {
68 grpc_core::Crash("Not implemented");
69 };
70 };
71
TEST(QueryExtensionsTest,EndpointSupportsMultipleExtensions)72 TEST(QueryExtensionsTest, EndpointSupportsMultipleExtensions) {
73 ExtendedTestEndpoint endpoint;
74 TestExtension<0>* extension_0 = QueryExtension<TestExtension<0>>(&endpoint);
75 TestExtension<1>* extension_1 = QueryExtension<TestExtension<1>>(&endpoint);
76 TestExtension<2>* extension_2 = QueryExtension<TestExtension<2>>(&endpoint);
77
78 EXPECT_NE(extension_0, nullptr);
79 EXPECT_NE(extension_1, nullptr);
80 EXPECT_NE(extension_2, nullptr);
81
82 EXPECT_EQ(extension_0->GetValue(), 0);
83 EXPECT_EQ(extension_1->GetValue(), 1);
84 EXPECT_EQ(extension_2->GetValue(), 2);
85 }
86 } // namespace
87
88 } // namespace experimental
89 } // namespace grpc_event_engine
90
main(int argc,char ** argv)91 int main(int argc, char** argv) {
92 testing::InitGoogleTest(&argc, argv);
93 return RUN_ALL_TESTS();
94 }
95