1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- // Copyright 2020 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "net/dns/dns_alias_utility.h"
- #include <string>
- #include <vector>
- #include "net/dns/public/dns_protocol.h"
- #include "testing/gmock/include/gmock/gmock.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace net {
- namespace {
- TEST(DnsAliasUtilityTest, FixUpDnsAliases) {
- const std::set<std::string> kAliases = {
- "localhost", "1.2.3.4", "a.com", "", "test",
- "0", "[::1]", "::1", "-www.e.com", "alias.com",
- "s .de", "www-1", "2a", "a-", "b..net",
- "a.com", "b_o.org", "alias.com", "1..3.2", "1.2.3.09",
- "foo.4", "a,b,c", "f/g", "www?", "[3a2:401f::1]",
- "0.0.1.2", "a.b.com", "c.org", "123.tld", "d-e.net",
- "f__g", "h"};
- const std::set<std::string> kExpectedFixedUpAliases = {
- "a.com", "test", "-www.e.com", "alias.com", "s%20.de", "www-1",
- "2a", "a-", "b..net", "b_o.org", "a%2Cb%2Cc", "a.b.com",
- "c.org", "123.tld", "d-e.net", "f__g", "h"};
- std::set<std::string> fixed_up_aliases =
- dns_alias_utility::FixUpDnsAliases(kAliases);
- EXPECT_EQ(kExpectedFixedUpAliases, fixed_up_aliases);
- std::string long_unqualified_alias(dns_protocol::kMaxCharNameLength + 1, 'x');
- std::string long_qualified_alias(dns_protocol::kMaxCharNameLength, 'x');
- long_qualified_alias += ".";
- std::set<std::string> set_with_long_aliases(
- {long_unqualified_alias, long_qualified_alias});
- std::set<std::string> fixed_up_set_with_long_aliases =
- dns_alias_utility::FixUpDnsAliases(set_with_long_aliases);
- EXPECT_THAT(fixed_up_set_with_long_aliases,
- testing::ElementsAre(long_qualified_alias));
- std::set<std::string> empty_aliases;
- std::set<std::string> fixed_up_empty_aliases =
- dns_alias_utility::FixUpDnsAliases(empty_aliases);
- EXPECT_TRUE(fixed_up_empty_aliases.empty());
- }
- } // namespace
- } // namespace net
|