1 //
2 //
3 // Copyright 2023 gRPC authors.
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 // http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 //
17 //
18
19 #include "src/cpp/ext/csm/csm_observability.h"
20
21 #include <grpcpp/ext/csm_observability.h>
22 #include <grpcpp/ext/otel_plugin.h>
23
24 #include "google/cloud/opentelemetry/resource_detector.h"
25 #include "gtest/gtest.h"
26 #include "opentelemetry/sdk/metrics/meter_provider.h"
27 #include "src/core/xds/grpc/xds_enabled_server.h"
28 #include "test/core/test_util/test_config.h"
29
30 namespace grpc {
31 namespace testing {
32 namespace {
33
TEST(CsmObservabilityBuilderTest,Basic)34 TEST(CsmObservabilityBuilderTest, Basic) {
35 EXPECT_EQ(
36 CsmObservabilityBuilder()
37 .SetMeterProvider(
38 std::make_shared<opentelemetry::sdk::metrics::MeterProvider>())
39 .BuildAndRegister()
40 .status(),
41 absl::OkStatus());
42 }
43
TEST(CsmDependencyTest,GoogleCloudOpenTelemetryDependency)44 TEST(CsmDependencyTest, GoogleCloudOpenTelemetryDependency) {
45 EXPECT_NE(google::cloud::otel::MakeResourceDetector(), nullptr);
46 }
47
TEST(CsmChannelTargetSelectorTest,NonXdsTargets)48 TEST(CsmChannelTargetSelectorTest, NonXdsTargets) {
49 auto obs = CsmObservabilityBuilder().BuildAndRegister();
50 EXPECT_FALSE(internal::CsmChannelTargetSelector("foo.bar.google.com"));
51 EXPECT_FALSE(internal::CsmChannelTargetSelector("dns:///foo.bar.google.com"));
52 EXPECT_FALSE(
53 internal::CsmChannelTargetSelector("dns:///foo.bar.google.com:1234"));
54 EXPECT_FALSE(internal::CsmChannelTargetSelector(
55 "dns://authority/foo.bar.google.com:1234"));
56 }
57
TEST(CsmChannelTargetSelectorTest,XdsTargets)58 TEST(CsmChannelTargetSelectorTest, XdsTargets) {
59 auto obs = CsmObservabilityBuilder().BuildAndRegister();
60 EXPECT_TRUE(internal::CsmChannelTargetSelector("xds:///foo"));
61 EXPECT_TRUE(internal::CsmChannelTargetSelector("xds:///foo.bar"));
62 }
63
TEST(CsmChannelTargetSelectorTest,XdsTargetsWithNonTDAuthority)64 TEST(CsmChannelTargetSelectorTest, XdsTargetsWithNonTDAuthority) {
65 auto obs = CsmObservabilityBuilder().BuildAndRegister();
66 EXPECT_FALSE(internal::CsmChannelTargetSelector("xds://authority/foo"));
67 }
68
TEST(CsmChannelTargetSelectorTest,XdsTargetsWithTDAuthority)69 TEST(CsmChannelTargetSelectorTest, XdsTargetsWithTDAuthority) {
70 auto obs = CsmObservabilityBuilder().BuildAndRegister();
71 EXPECT_TRUE(internal::CsmChannelTargetSelector(
72 "xds://traffic-director-global.xds.googleapis.com/foo"));
73 }
74
TEST(CsmChannelTargetSelectorTest,CsmObservabilityOutOfScope)75 TEST(CsmChannelTargetSelectorTest, CsmObservabilityOutOfScope) {
76 {
77 auto obs =
78 CsmObservabilityBuilder()
79 .SetMeterProvider(
80 std::make_shared<opentelemetry::sdk::metrics::MeterProvider>())
81 .BuildAndRegister();
82 }
83 // When CsmObservability goes out of scope, the target selector should return
84 // false as well.
85 EXPECT_FALSE(internal::CsmChannelTargetSelector("foo.bar.google.com"));
86 EXPECT_FALSE(internal::CsmChannelTargetSelector("xds:///foo"));
87 EXPECT_FALSE(internal::CsmChannelTargetSelector(
88 "xds://traffic-director-global.xds.googleapis.com/foo"));
89 }
90
TEST(CsmServerSelectorTest,ChannelArgs)91 TEST(CsmServerSelectorTest, ChannelArgs) {
92 auto obs = CsmObservabilityBuilder().BuildAndRegister();
93 EXPECT_TRUE(internal::CsmServerSelector(grpc_core::ChannelArgs()));
94 }
95
TEST(CsmServerSelectorTest,CsmObservabilityOutOfScope)96 TEST(CsmServerSelectorTest, CsmObservabilityOutOfScope) {
97 {
98 auto obs =
99 CsmObservabilityBuilder()
100 .SetMeterProvider(
101 std::make_shared<opentelemetry::sdk::metrics::MeterProvider>())
102 .BuildAndRegister();
103 }
104 // When CsmObservability goes out of scope, the server selector should return
105 // false as well.
106 EXPECT_FALSE(internal::CsmServerSelector(grpc_core::ChannelArgs()));
107 EXPECT_FALSE(internal::CsmServerSelector(
108 grpc_core::ChannelArgs().Set(GRPC_ARG_XDS_ENABLED_SERVER, true)));
109 }
110
111 } // namespace
112 } // namespace testing
113 } // namespace grpc
114
main(int argc,char ** argv)115 int main(int argc, char** argv) {
116 grpc::testing::TestEnvironment env(&argc, argv);
117 ::testing::InitGoogleTest(&argc, argv);
118 return RUN_ALL_TESTS();
119 }
120