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 "google/cloud/opentelemetry/resource_detector.h"
22 #include "gtest/gtest.h"
23
24 #include <grpcpp/ext/csm_observability.h>
25 #include <grpcpp/ext/otel_plugin.h>
26
27 #include "src/core/ext/xds/xds_enabled_server.h"
28 #include "test/core/util/test_config.h"
29
30 namespace grpc {
31 namespace testing {
32 namespace {
33
TEST(CsmObservabilityBuilderTest,Basic)34 TEST(CsmObservabilityBuilderTest, Basic) {
35 EXPECT_EQ(CsmObservabilityBuilder().BuildAndRegister().status(),
36 absl::OkStatus());
37 }
38
TEST(GsmDependencyTest,GoogleCloudOpenTelemetryDependency)39 TEST(GsmDependencyTest, GoogleCloudOpenTelemetryDependency) {
40 EXPECT_NE(google::cloud::otel::MakeResourceDetector(), nullptr);
41 }
42
TEST(CsmChannelTargetSelectorTest,NonXdsTargets)43 TEST(CsmChannelTargetSelectorTest, NonXdsTargets) {
44 auto obs = CsmObservabilityBuilder().BuildAndRegister();
45 EXPECT_FALSE(internal::CsmChannelTargetSelector("foo.bar.google.com"));
46 EXPECT_FALSE(internal::CsmChannelTargetSelector("dns:///foo.bar.google.com"));
47 EXPECT_FALSE(
48 internal::CsmChannelTargetSelector("dns:///foo.bar.google.com:1234"));
49 EXPECT_FALSE(internal::CsmChannelTargetSelector(
50 "dns://authority/foo.bar.google.com:1234"));
51 }
52
TEST(CsmChannelTargetSelectorTest,XdsTargets)53 TEST(CsmChannelTargetSelectorTest, XdsTargets) {
54 auto obs = CsmObservabilityBuilder().BuildAndRegister();
55 EXPECT_TRUE(internal::CsmChannelTargetSelector("xds:///foo"));
56 EXPECT_TRUE(internal::CsmChannelTargetSelector("xds:///foo.bar"));
57 }
58
TEST(CsmChannelTargetSelectorTest,XdsTargetsWithNonTDAuthority)59 TEST(CsmChannelTargetSelectorTest, XdsTargetsWithNonTDAuthority) {
60 auto obs = CsmObservabilityBuilder().BuildAndRegister();
61 EXPECT_FALSE(internal::CsmChannelTargetSelector("xds://authority/foo"));
62 }
63
TEST(CsmChannelTargetSelectorTest,XdsTargetsWithTDAuthority)64 TEST(CsmChannelTargetSelectorTest, XdsTargetsWithTDAuthority) {
65 auto obs = CsmObservabilityBuilder().BuildAndRegister();
66 EXPECT_TRUE(internal::CsmChannelTargetSelector(
67 "xds://traffic-director-global.xds.googleapis.com/foo"));
68 }
69
TEST(CsmChannelTargetSelectorTest,CsmObservabilityOutOfScope)70 TEST(CsmChannelTargetSelectorTest, CsmObservabilityOutOfScope) {
71 { auto obs = CsmObservabilityBuilder().BuildAndRegister(); }
72 // When CsmObservability goes out of scope, the target selector should return
73 // false as well.
74 EXPECT_FALSE(internal::CsmChannelTargetSelector("foo.bar.google.com"));
75 EXPECT_FALSE(internal::CsmChannelTargetSelector("xds:///foo"));
76 EXPECT_FALSE(internal::CsmChannelTargetSelector(
77 "xds://traffic-director-global.xds.googleapis.com/foo"));
78 }
79
TEST(CsmServerSelectorTest,ChannelArgsWithoutXdsServerArg)80 TEST(CsmServerSelectorTest, ChannelArgsWithoutXdsServerArg) {
81 auto obs = CsmObservabilityBuilder().BuildAndRegister();
82 EXPECT_FALSE(internal::CsmServerSelector(grpc_core::ChannelArgs()));
83 }
84
TEST(CsmServerSelectorTest,ChannelArgsWithXdsServerArg)85 TEST(CsmServerSelectorTest, ChannelArgsWithXdsServerArg) {
86 auto obs = CsmObservabilityBuilder().BuildAndRegister();
87 EXPECT_TRUE(internal::CsmServerSelector(
88 grpc_core::ChannelArgs().Set(GRPC_ARG_XDS_ENABLED_SERVER, true)));
89 }
90
TEST(CsmServerSelectorTest,CsmObservabilityOutOfScope)91 TEST(CsmServerSelectorTest, CsmObservabilityOutOfScope) {
92 { auto obs = CsmObservabilityBuilder().BuildAndRegister(); }
93 // When CsmObservability goes out of scope, the server selector should return
94 // false as well.
95 EXPECT_FALSE(internal::CsmServerSelector(grpc_core::ChannelArgs()));
96 EXPECT_FALSE(internal::CsmServerSelector(
97 grpc_core::ChannelArgs().Set(GRPC_ARG_XDS_ENABLED_SERVER, true)));
98 }
99
100 } // namespace
101 } // namespace testing
102 } // namespace grpc
103
main(int argc,char ** argv)104 int main(int argc, char** argv) {
105 grpc::testing::TestEnvironment env(&argc, argv);
106 ::testing::InitGoogleTest(&argc, argv);
107 return RUN_ALL_TESTS();
108 }
109