1 files changed, 52 insertions, 0 deletions
diff --git a/extensions/browser/api/dns/mock_host_resolver_creator.h b/extensions/browser/api/dns/mock_host_resolver_creator.h
new file mode 100644
index 0000000..f57fdb8
--- /dev/null
+++ b/extensions/browser/api/dns/mock_host_resolver_creator.h
@@ -0,0 +1,52 @@
+// 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.
+
+#ifndef EXTENSIONS_BROWSER_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_
+#define EXTENSIONS_BROWSER_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_
+
+#include <string>
+
+#include "base/memory/ref_counted.h"
+#include "base/synchronization/waitable_event.h"
+
+namespace net {
+class HostResolver;
+class MockHostResolver;
+}
+
+namespace extensions {
+
+// Used only for testing. Creates a MockHostResolver, respecting threading
+// constraints.
+class MockHostResolverCreator
+    : public base::RefCountedThreadSafe<MockHostResolverCreator> {
+ public:
+  static const std::string kHostname;
+  static const std::string kAddress;
+
+  MockHostResolverCreator();
+
+  net::HostResolver* CreateMockHostResolver();
+  void DeleteMockHostResolver();
+
+ private:
+  friend class base::RefCountedThreadSafe<MockHostResolverCreator>;
+  virtual ~MockHostResolverCreator();
+
+  void CreateMockHostResolverOnIOThread();
+  void DeleteMockHostResolverOnIOThread();
+
+  base::WaitableEvent resolver_event_;
+
+  // The MockHostResolver asserts that it's used on the same thread on which
+  // it's created, which is actually a stronger rule than its real counterpart.
+  // But that's fine; it's good practice.
+  //
+  // Plain pointer because we have to manage lifetime manually.
+  net::MockHostResolver* mock_host_resolver_;
+};
+
+}  // namespace extensions
+
+#endif  // EXTENSIONS_BROWSER_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_