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 <set>
20
21 #include "absl/status/statusor.h"
22 #include "absl/strings/string_view.h"
23 #include "gmock/gmock.h"
24 #include "gtest/gtest.h"
25
26 #include <grpc/support/log.h>
27 #include <grpc/support/port_platform.h>
28
29 #include "src/core/lib/address_utils/parse_address.h"
30 #include "src/core/lib/address_utils/sockaddr_utils.h"
31 #include "src/core/lib/iomgr/resolved_address.h"
32 #include "src/core/lib/uri/uri_parser.h"
33 #include "test/core/util/test_config.h"
34
35 namespace grpc_core {
36 namespace testing {
37 namespace {
38
MakeAddress(absl::string_view address_uri)39 grpc_resolved_address MakeAddress(absl::string_view address_uri) {
40 auto uri = URI::Parse(address_uri);
41 GPR_ASSERT(uri.ok());
42 grpc_resolved_address address;
43 GPR_ASSERT(grpc_parse_uri(*uri, &address));
44 return address;
45 }
46
47 MATCHER_P(EqualsAddress, address_str, "") {
48 auto addr = grpc_sockaddr_to_uri(&arg);
49 if (!addr.ok()) {
50 *result_listener << "grpc_sockaddr_to_uri() failed";
51 return false;
52 }
53 return ::testing::ExplainMatchResult(*addr, address_str, result_listener);
54 }
55
TEST(ResolvedAddressLessThan,Basic)56 TEST(ResolvedAddressLessThan, Basic) {
57 std::set<grpc_resolved_address, ResolvedAddressLessThan> address_set;
58 address_set.insert(MakeAddress("ipv4:127.0.0.2:443"));
59 address_set.insert(MakeAddress("ipv4:127.0.0.3:443"));
60 address_set.insert(MakeAddress("ipv4:127.0.0.1:443"));
61 EXPECT_THAT(address_set,
62 ::testing::ElementsAre(EqualsAddress("ipv4:127.0.0.1:443"),
63 EqualsAddress("ipv4:127.0.0.2:443"),
64 EqualsAddress("ipv4:127.0.0.3:443")));
65 }
66
TEST(EndpointAddressSet,Basic)67 TEST(EndpointAddressSet, Basic) {
68 EndpointAddressSet set1({MakeAddress("ipv4:127.0.0.2:443"),
69 MakeAddress("ipv4:127.0.0.3:443"),
70 MakeAddress("ipv4:127.0.0.1:443")});
71 EXPECT_TRUE(set1 == set1);
72 EXPECT_FALSE(set1 < set1);
73 EXPECT_EQ(set1.ToString(), "{127.0.0.1:443, 127.0.0.2:443, 127.0.0.3:443}");
74 EndpointAddressSet set2({MakeAddress("ipv4:127.0.0.4:443"),
75 MakeAddress("ipv4:127.0.0.6:443"),
76 MakeAddress("ipv4:127.0.0.5:443")});
77 EXPECT_FALSE(set1 == set2);
78 EXPECT_TRUE(set1 < set2);
79 EXPECT_FALSE(set2 < set1);
80 EXPECT_EQ(set2.ToString(), "{127.0.0.4:443, 127.0.0.5:443, 127.0.0.6:443}");
81 }
82
TEST(EndpointAddressSet,Subset)83 TEST(EndpointAddressSet, Subset) {
84 EndpointAddressSet set1({MakeAddress("ipv4:127.0.0.2:443"),
85 MakeAddress("ipv4:127.0.0.3:443"),
86 MakeAddress("ipv4:127.0.0.1:443")});
87 EXPECT_EQ(set1.ToString(), "{127.0.0.1:443, 127.0.0.2:443, 127.0.0.3:443}");
88 EndpointAddressSet set2(
89 {MakeAddress("ipv4:127.0.0.2:443"), MakeAddress("ipv4:127.0.0.1:443")});
90 EXPECT_EQ(set2.ToString(), "{127.0.0.1:443, 127.0.0.2:443}");
91 EXPECT_FALSE(set1 == set2);
92 EXPECT_FALSE(set1 < set2);
93 EXPECT_TRUE(set2 < set1);
94 }
95
96 } // namespace
97 } // namespace testing
98 } // namespace grpc_core
99
main(int argc,char ** argv)100 int main(int argc, char** argv) {
101 ::testing::InitGoogleTest(&argc, argv);
102 grpc::testing::TestEnvironment env(&argc, argv);
103 return RUN_ALL_TESTS();
104 }
105