1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROME_BROWSER_EXTENSIONS_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_ 6 #define CHROME_BROWSER_EXTENSIONS_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_ 7 8 #include <string> 9 10 #include "base/memory/ref_counted.h" 11 #include "base/synchronization/waitable_event.h" 12 13 namespace net { 14 class MockHostResolver; 15 } 16 17 namespace extensions { 18 19 // Used only for testing. Creates a MockHostResolver, respecting threading 20 // constraints. 21 class MockHostResolverCreator 22 : public base::RefCountedThreadSafe<MockHostResolverCreator> { 23 public: 24 static const std::string kHostname; 25 static const std::string kAddress; 26 27 MockHostResolverCreator(); 28 29 net::MockHostResolver* CreateMockHostResolver(); 30 void DeleteMockHostResolver(); 31 32 private: 33 friend class base::RefCountedThreadSafe<MockHostResolverCreator>; 34 virtual ~MockHostResolverCreator(); 35 36 void CreateMockHostResolverOnIOThread(); 37 void DeleteMockHostResolverOnIOThread(); 38 39 base::WaitableEvent resolver_event_; 40 41 // The MockHostResolver asserts that it's used on the same thread on which 42 // it's created, which is actually a stronger rule than its real counterpart. 43 // But that's fine; it's good practice. 44 // 45 // Plain pointer because we have to manage lifetime manually. 46 net::MockHostResolver* mock_host_resolver_; 47 }; 48 49 } // namespace extensions 50 51 #endif // CHROME_BROWSER_EXTENSIONS_API_DNS_MOCK_HOST_RESOLVER_CREATOR_H_ 52