OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ | 5 #ifndef CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
6 #define CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ | 6 #define CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include "base/ref_counted.h" | 9 #include "base/ref_counted.h" |
10 #include "chrome/browser/browser_thread.h" | 10 #include "chrome/browser/browser_thread.h" |
11 #include "chrome/common/net/url_request_context_getter.h" | 11 #include "chrome/common/net/url_request_context_getter.h" |
12 #include "net/url_request/url_request_unittest.h" | 12 #include "net/url_request/url_request_test_util.h" |
13 | 13 |
14 // Used to return a dummy context (normally the context is on the IO thread). | 14 // Used to return a dummy context (normally the context is on the IO thread). |
15 // The one here can be run on the main test thread. Note that this can lead to | 15 // The one here can be run on the main test thread. Note that this can lead to |
16 // a leak if your test does not have a BrowserThread::IO in it because | 16 // a leak if your test does not have a BrowserThread::IO in it because |
17 // URLRequestContextGetter is defined as a ReferenceCounted object with a | 17 // URLRequestContextGetter is defined as a ReferenceCounted object with a |
18 // special trait that deletes it on the IO thread. | 18 // special trait that deletes it on the IO thread. |
19 class TestURLRequestContextGetter : public URLRequestContextGetter { | 19 class TestURLRequestContextGetter : public URLRequestContextGetter { |
20 public: | 20 public: |
21 virtual net::URLRequestContext* GetURLRequestContext() { | 21 virtual net::URLRequestContext* GetURLRequestContext() { |
22 if (!context_) | 22 if (!context_) |
23 context_ = new TestURLRequestContext(); | 23 context_ = new TestURLRequestContext(); |
24 return context_.get(); | 24 return context_.get(); |
25 } | 25 } |
26 virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy() const { | 26 virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy() const { |
27 return BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO); | 27 return BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO); |
28 } | 28 } |
29 | 29 |
30 private: | 30 private: |
31 scoped_refptr<net::URLRequestContext> context_; | 31 scoped_refptr<net::URLRequestContext> context_; |
32 }; | 32 }; |
33 | 33 |
34 #endif // CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ | 34 #endif // CHROME_TEST_TEST_URL_REQUEST_CONTEXT_GETTER_H_ |
OLD | NEW |