OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2010 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_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
| 6 #define CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
| 7 |
| 8 #include "base/ref_counted.h" |
| 9 #include "chrome/browser/browser_thread.h" |
| 10 #include "chrome/common/net/url_request_context_getter.h" |
| 11 #include "net/url_request/url_request_unittest.h" |
| 12 |
| 13 // Used to return a dummy context (normally the context is on the IO thread). |
| 14 // The one here can be run on the main test thread. Note that this can lead to |
| 15 // a leak if your test does not have a BrowserThread::IO in it because |
| 16 // URLRequestContextGetter is defined as a ReferenceCounted object with a |
| 17 // special trait that deletes it on the IO thread. |
| 18 class TestURLRequestContextGetter : public URLRequestContextGetter { |
| 19 public: |
| 20 virtual URLRequestContext* GetURLRequestContext() { |
| 21 if (!context_) |
| 22 context_ = new TestURLRequestContext(); |
| 23 return context_.get(); |
| 24 } |
| 25 virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy() const { |
| 26 return BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO); |
| 27 } |
| 28 |
| 29 private: |
| 30 scoped_refptr<URLRequestContext> context_; |
| 31 }; |
| 32 |
| 33 #endif // CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
OLD | NEW |