extensions/browser/api/dns/mock_host_resolver_creator.cc

// Copyright 2014 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 "extensions/browser/api/dns/mock_host_resolver_creator.h"

#include "base/logging.h"
#include "content/public/browser/browser_thread.h"
#include "net/dns/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::HostResolver* MockHostResolverCreator::CreateMockHostResolver() {
  DCHECK(!mock_host_resolver_);
  DCHECK_CURRENTLY_ON(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_CURRENTLY_ON(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);
  CHECK(resolver_event_.TimedWait(max_time));
}

void MockHostResolverCreator::DeleteMockHostResolverOnIOThread() {
  delete (mock_host_resolver_);
  mock_host_resolver_ = NULL;
  resolver_event_.Signal();
}

}  // namespace extensions