From 471822cad3c7f41002c28fd9a32243abe1bbb2a2 Mon Sep 17 00:00:00 2001 From: "phajdan.jr@chromium.org" Date: Thu, 29 Jan 2009 11:32:26 +0000 Subject: Add more unit tests for net/base/host_resolver. BUG=6661 Review URL: http://codereview.chromium.org/18775 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@8872 0039d316-1c4b-4281-b951-d872f2087c98 --- net/base/host_resolver.h | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'net/base/host_resolver.h') diff --git a/net/base/host_resolver.h b/net/base/host_resolver.h index 1f649f8..bfae53c 100644 --- a/net/base/host_resolver.h +++ b/net/base/host_resolver.h @@ -57,10 +57,24 @@ class HostResolver { // A helper class used in unit tests to alter hostname mappings. See // SetHostMapper for details. -class HostMapper { +class HostMapper : public base::RefCountedThreadSafe { public: virtual ~HostMapper() {} virtual std::string Map(const std::string& host) = 0; + + protected: + // Ask previous host mapper (if set) for mapping of given host. + std::string MapUsingPrevious(const std::string& host); + + private: + friend class ScopedHostMapper; + + // Set mapper to ask when this mapper doesn't want to modify the result. + void set_previous_mapper(HostMapper* mapper) { + previous_mapper_ = mapper; + } + + scoped_refptr previous_mapper_; }; #ifdef UNIT_TEST -- cgit v1.1