Allow hostnames in socket.connect().

chrome/browser/extensions/api/dns/mock_host_resolver_creator.cc

+// 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 "chrome/browser/extensions/api/dns/mock_host_resolver_creator.h"
+
+#include "chrome/test/base/in_process_browser_test.h"
+#include "content/public/browser/browser_thread.h"
+#include "net/base/mock_host_resolver.h"
+
+using content::BrowserThread;
+
+namespace extensions {
+
+const std::string MockHostResolverCreator::kHostname = "www.sowbug.com";
+const std::string MockHostResolverCreator::kAddress = "9.8.7.6";
+
+MockHostResolverCreator::MockHostResolverCreator()
+  : resolver_event_(true, false),
+    mock_host_resolver_(NULL) {
+}
+
+MockHostResolverCreator::~MockHostResolverCreator() {
+}
+
+net::MockHostResolver* MockHostResolverCreator::CreateMockHostResolver() {
+  DCHECK(!mock_host_resolver_);
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  bool result = BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&MockHostResolverCreator::CreateMockHostResolverOnIOThread,
+                 this));
+  DCHECK(result);
+
+  base::TimeDelta max_time = base::TimeDelta::FromSeconds(5);
+  resolver_event_.TimedWait(max_time);
+
+  return mock_host_resolver_;
+}
+
+void MockHostResolverCreator::CreateMockHostResolverOnIOThread() {
+  mock_host_resolver_ = new net::MockHostResolver();
+  mock_host_resolver_->rules()->AddRule(kHostname, kAddress);
+  mock_host_resolver_->rules()->AddSimulatedFailure("this.hostname.is.bogus");
+  resolver_event_.Signal();
+}
+
+void MockHostResolverCreator::DeleteMockHostResolver() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  if (!mock_host_resolver_)
+    return;
+  resolver_event_.Reset();
+  bool result = BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&MockHostResolverCreator::DeleteMockHostResolverOnIOThread,
+                 this));
+  DCHECK(result);
+
+  base::TimeDelta max_time = base::TimeDelta::FromSeconds(5);
+  ASSERT_TRUE(resolver_event_.TimedWait(max_time));
+}
+
+void MockHostResolverCreator::DeleteMockHostResolverOnIOThread() {
+  delete(mock_host_resolver_);
+  mock_host_resolver_ = NULL;
+  resolver_event_.Signal();
+}
+
+}  // namespace extensions