1 //
2 // Copyright 2023 gRPC authors.
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16
17 #include "src/core/resolver/endpoint_addresses.h"
18
19 #include <grpc/support/port_platform.h>
20
21 #include <set>
22
23 #include "absl/log/check.h"
24 #include "absl/status/statusor.h"
25 #include "absl/strings/string_view.h"
26 #include "gmock/gmock.h"
27 #include "gtest/gtest.h"
28 #include "src/core/lib/address_utils/parse_address.h"
29 #include "src/core/lib/address_utils/sockaddr_utils.h"
30 #include "src/core/lib/iomgr/resolved_address.h"
31 #include "src/core/util/uri.h"
32 #include "test/core/test_util/test_config.h"
33
34 namespace grpc_core {
35 namespace testing {
36 namespace {
37
MakeAddress(absl::string_view address_uri)38 grpc_resolved_address MakeAddress(absl::string_view address_uri) {
39 auto uri = URI::Parse(address_uri);
40 CHECK(uri.ok());
41 grpc_resolved_address address;
42 CHECK(grpc_parse_uri(*uri, &address));
43 return address;
44 }
45
46 MATCHER_P(EqualsAddress, address_str, "") {
47 auto addr = grpc_sockaddr_to_uri(&arg);
48 if (!addr.ok()) {
49 *result_listener << "grpc_sockaddr_to_uri() failed";
50 return false;
51 }
52 return ::testing::ExplainMatchResult(*addr, address_str, result_listener);
53 }
54
TEST(ResolvedAddressLessThan,Basic)55 TEST(ResolvedAddressLessThan, Basic) {
56 std::set<grpc_resolved_address, ResolvedAddressLessThan> address_set;
57 address_set.insert(MakeAddress("ipv4:127.0.0.2:443"));
58 address_set.insert(MakeAddress("ipv4:127.0.0.3:443"));
59 address_set.insert(MakeAddress("ipv4:127.0.0.1:443"));
60 EXPECT_THAT(address_set,
61 ::testing::ElementsAre(EqualsAddress("ipv4:127.0.0.1:443"),
62 EqualsAddress("ipv4:127.0.0.2:443"),
63 EqualsAddress("ipv4:127.0.0.3:443")));
64 }
65
TEST(EndpointAddressSet,Basic)66 TEST(EndpointAddressSet, Basic) {
67 EndpointAddressSet set1({MakeAddress("ipv4:127.0.0.2:443"),
68 MakeAddress("ipv4:127.0.0.3:443"),
69 MakeAddress("ipv4:127.0.0.1:443")});
70 EXPECT_TRUE(set1 == set1);
71 EXPECT_FALSE(set1 < set1);
72 EXPECT_EQ(set1.ToString(), "{127.0.0.1:443, 127.0.0.2:443, 127.0.0.3:443}");
73 EndpointAddressSet set2({MakeAddress("ipv4:127.0.0.4:443"),
74 MakeAddress("ipv4:127.0.0.6:443"),
75 MakeAddress("ipv4:127.0.0.5:443")});
76 EXPECT_FALSE(set1 == set2);
77 EXPECT_TRUE(set1 < set2);
78 EXPECT_FALSE(set2 < set1);
79 EXPECT_EQ(set2.ToString(), "{127.0.0.4:443, 127.0.0.5:443, 127.0.0.6:443}");
80 }
81
TEST(EndpointAddressSet,Subset)82 TEST(EndpointAddressSet, Subset) {
83 EndpointAddressSet set1({MakeAddress("ipv4:127.0.0.2:443"),
84 MakeAddress("ipv4:127.0.0.3:443"),
85 MakeAddress("ipv4:127.0.0.1:443")});
86 EXPECT_EQ(set1.ToString(), "{127.0.0.1:443, 127.0.0.2:443, 127.0.0.3:443}");
87 EndpointAddressSet set2(
88 {MakeAddress("ipv4:127.0.0.2:443"), MakeAddress("ipv4:127.0.0.1:443")});
89 EXPECT_EQ(set2.ToString(), "{127.0.0.1:443, 127.0.0.2:443}");
90 EXPECT_FALSE(set1 == set2);
91 EXPECT_FALSE(set1 < set2);
92 EXPECT_TRUE(set2 < set1);
93 }
94
95 } // namespace
96 } // namespace testing
97 } // namespace grpc_core
98
main(int argc,char ** argv)99 int main(int argc, char** argv) {
100 ::testing::InitGoogleTest(&argc, argv);
101 grpc::testing::TestEnvironment env(&argc, argv);
102 return RUN_ALL_TESTS();
103 }
104