dns_alias_utility_unittest.cc 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "net/dns/dns_alias_utility.h"
  5. #include <string>
  6. #include <vector>
  7. #include "net/dns/public/dns_protocol.h"
  8. #include "testing/gmock/include/gmock/gmock.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace net {
  11. namespace {
  12. TEST(DnsAliasUtilityTest, FixUpDnsAliases) {
  13. const std::set<std::string> kAliases = {
  14. "localhost", "1.2.3.4", "a.com", "", "test",
  15. "0", "[::1]", "::1", "-www.e.com", "alias.com",
  16. "s .de", "www-1", "2a", "a-", "b..net",
  17. "a.com", "b_o.org", "alias.com", "1..3.2", "1.2.3.09",
  18. "foo.4", "a,b,c", "f/g", "www?", "[3a2:401f::1]",
  19. "0.0.1.2", "a.b.com", "c.org", "123.tld", "d-e.net",
  20. "f__g", "h"};
  21. const std::set<std::string> kExpectedFixedUpAliases = {
  22. "a.com", "test", "-www.e.com", "alias.com", "s%20.de", "www-1",
  23. "2a", "a-", "b..net", "b_o.org", "a%2Cb%2Cc", "a.b.com",
  24. "c.org", "123.tld", "d-e.net", "f__g", "h"};
  25. std::set<std::string> fixed_up_aliases =
  26. dns_alias_utility::FixUpDnsAliases(kAliases);
  27. EXPECT_EQ(kExpectedFixedUpAliases, fixed_up_aliases);
  28. std::string long_unqualified_alias(dns_protocol::kMaxCharNameLength + 1, 'x');
  29. std::string long_qualified_alias(dns_protocol::kMaxCharNameLength, 'x');
  30. long_qualified_alias += ".";
  31. std::set<std::string> set_with_long_aliases(
  32. {long_unqualified_alias, long_qualified_alias});
  33. std::set<std::string> fixed_up_set_with_long_aliases =
  34. dns_alias_utility::FixUpDnsAliases(set_with_long_aliases);
  35. EXPECT_THAT(fixed_up_set_with_long_aliases,
  36. testing::ElementsAre(long_qualified_alias));
  37. std::set<std::string> empty_aliases;
  38. std::set<std::string> fixed_up_empty_aliases =
  39. dns_alias_utility::FixUpDnsAliases(empty_aliases);
  40. EXPECT_TRUE(fixed_up_empty_aliases.empty());
  41. }
  42. } // namespace
  43. } // namespace net