Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(419)

Unified Diff: net/base/mapped_host_resolver_unittest.cc

Issue 12518036: net: move host_resolver files from net/base to net/dns (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 7 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: net/base/mapped_host_resolver_unittest.cc
diff --git a/net/base/mapped_host_resolver_unittest.cc b/net/base/mapped_host_resolver_unittest.cc
deleted file mode 100644
index 35fe011ebd038dd154e8a5266582fa34c47a1a39..0000000000000000000000000000000000000000
--- a/net/base/mapped_host_resolver_unittest.cc
+++ /dev/null
@@ -1,219 +0,0 @@
-// Copyright (c) 2012 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/base/mapped_host_resolver.h"
-
-#include "net/base/address_list.h"
-#include "net/base/mock_host_resolver.h"
-#include "net/base/net_errors.h"
-#include "net/base/net_log.h"
-#include "net/base/net_util.h"
-#include "net/base/test_completion_callback.h"
-#include "testing/gtest/include/gtest/gtest.h"
-
-namespace net {
-
-namespace {
-
-std::string FirstAddress(const AddressList& address_list) {
- if (address_list.empty())
- return "";
- return address_list.front().ToString();
-}
-
-TEST(MappedHostResolverTest, Inclusion) {
- // Create a mock host resolver, with specific hostname to IP mappings.
- scoped_ptr<MockHostResolver> resolver_impl(new MockHostResolver());
- resolver_impl->rules()->AddSimulatedFailure("*google.com");
- resolver_impl->rules()->AddRule("baz.com", "192.168.1.5");
- resolver_impl->rules()->AddRule("foo.com", "192.168.1.8");
- resolver_impl->rules()->AddRule("proxy", "192.168.1.11");
-
- // Create a remapped resolver that uses |resolver_impl|.
- scoped_ptr<MappedHostResolver> resolver(
- new MappedHostResolver(resolver_impl.PassAs<HostResolver>()));
-
- int rv;
- AddressList address_list;
-
- // Try resolving "www.google.com:80". There are no mappings yet, so this
- // hits |resolver_impl| and fails.
- TestCompletionCallback callback;
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.google.com", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(ERR_NAME_NOT_RESOLVED, rv);
-
- // Remap *.google.com to baz.com.
- EXPECT_TRUE(resolver->AddRuleFromString("map *.google.com baz.com"));
-
- // Try resolving "www.google.com:80". Should be remapped to "baz.com:80".
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.google.com", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.5:80", FirstAddress(address_list));
-
- // Try resolving "foo.com:77". This will NOT be remapped, so result
- // is "foo.com:77".
- rv = resolver->Resolve(HostResolver::RequestInfo(HostPortPair("foo.com", 77)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.8:77", FirstAddress(address_list));
-
- // Remap "*.org" to "proxy:99".
- EXPECT_TRUE(resolver->AddRuleFromString("Map *.org proxy:99"));
-
- // Try resolving "chromium.org:61". Should be remapped to "proxy:99".
- rv = resolver->Resolve(HostResolver::RequestInfo
- (HostPortPair("chromium.org", 61)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.11:99", FirstAddress(address_list));
-}
-
-// Tests that exclusions are respected.
-TEST(MappedHostResolverTest, Exclusion) {
- // Create a mock host resolver, with specific hostname to IP mappings.
- scoped_ptr<MockHostResolver> resolver_impl(new MockHostResolver());
- resolver_impl->rules()->AddRule("baz", "192.168.1.5");
- resolver_impl->rules()->AddRule("www.google.com", "192.168.1.3");
-
- // Create a remapped resolver that uses |resolver_impl|.
- scoped_ptr<MappedHostResolver> resolver(
- new MappedHostResolver(resolver_impl.PassAs<HostResolver>()));
-
- int rv;
- AddressList address_list;
- TestCompletionCallback callback;
-
- // Remap "*.com" to "baz".
- EXPECT_TRUE(resolver->AddRuleFromString("map *.com baz"));
-
- // Add an exclusion for "*.google.com".
- EXPECT_TRUE(resolver->AddRuleFromString("EXCLUDE *.google.com"));
-
- // Try resolving "www.google.com". Should not be remapped due to exclusion).
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.google.com", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.3:80", FirstAddress(address_list));
-
- // Try resolving "chrome.com:80". Should be remapped to "baz:80".
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("chrome.com", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.5:80", FirstAddress(address_list));
-}
-
-TEST(MappedHostResolverTest, SetRulesFromString) {
- // Create a mock host resolver, with specific hostname to IP mappings.
- scoped_ptr<MockHostResolver> resolver_impl(new MockHostResolver());
- resolver_impl->rules()->AddRule("baz", "192.168.1.7");
- resolver_impl->rules()->AddRule("bar", "192.168.1.9");
-
- // Create a remapped resolver that uses |resolver_impl|.
- scoped_ptr<MappedHostResolver> resolver(
- new MappedHostResolver(resolver_impl.PassAs<HostResolver>()));
-
- int rv;
- AddressList address_list;
- TestCompletionCallback callback;
-
- // Remap "*.com" to "baz", and *.net to "bar:60".
- resolver->SetRulesFromString("map *.com baz , map *.net bar:60");
-
- // Try resolving "www.google.com". Should be remapped to "baz".
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.google.com", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.7:80", FirstAddress(address_list));
-
- // Try resolving "chrome.net:80". Should be remapped to "bar:60".
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("chrome.net", 80)),
- &address_list, callback.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.9:60", FirstAddress(address_list));
-}
-
-// Parsing bad rules should silently discard the rule (and never crash).
-TEST(MappedHostResolverTest, ParseInvalidRules) {
- scoped_ptr<MappedHostResolver> resolver(
- new MappedHostResolver(scoped_ptr<HostResolver>()));
-
- EXPECT_FALSE(resolver->AddRuleFromString("xyz"));
- EXPECT_FALSE(resolver->AddRuleFromString(""));
- EXPECT_FALSE(resolver->AddRuleFromString(" "));
- EXPECT_FALSE(resolver->AddRuleFromString("EXCLUDE"));
- EXPECT_FALSE(resolver->AddRuleFromString("EXCLUDE foo bar"));
- EXPECT_FALSE(resolver->AddRuleFromString("INCLUDE"));
- EXPECT_FALSE(resolver->AddRuleFromString("INCLUDE x"));
- EXPECT_FALSE(resolver->AddRuleFromString("INCLUDE x :10"));
-}
-
-// Test mapping hostnames to resolving failures.
-TEST(MappedHostResolverTest, MapToError) {
- scoped_ptr<MockHostResolver> resolver_impl(new MockHostResolver());
- resolver_impl->rules()->AddRule("*", "192.168.1.5");
-
- scoped_ptr<MappedHostResolver> resolver(
- new MappedHostResolver(resolver_impl.PassAs<HostResolver>()));
-
- int rv;
- AddressList address_list;
-
- // Remap *.google.com to resolving failures.
- EXPECT_TRUE(resolver->AddRuleFromString("MAP *.google.com ~NOTFOUND"));
-
- // Try resolving www.google.com --> Should give an error.
- TestCompletionCallback callback1;
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.google.com", 80)),
- &address_list, callback1.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_NAME_NOT_RESOLVED, rv);
-
- // Try resolving www.foo.com --> Should succeed.
- TestCompletionCallback callback2;
- rv = resolver->Resolve(HostResolver::RequestInfo(
- HostPortPair("www.foo.com", 80)),
- &address_list, callback2.callback(), NULL,
- BoundNetLog());
- EXPECT_EQ(ERR_IO_PENDING, rv);
- rv = callback2.WaitForResult();
- EXPECT_EQ(OK, rv);
- EXPECT_EQ("192.168.1.5:80", FirstAddress(address_list));
-}
-
-} // namespace
-
-} // namespace net
« no previous file with comments | « net/base/mapped_host_resolver.cc ('k') | net/base/mock_host_resolver.h » ('j') | webkit/DEPS » ('J')

Powered by Google App Engine
This is Rietveld 408576698