Index: net/filter/mock_filter_context.h |
diff --git a/net/filter/mock_filter_context.h b/net/filter/mock_filter_context.h |
index 21e56ac92e17c314ba49d5f1bca21a0348a32816..fb950338a4cb521f8dfa7a546c9c375863e31754 100644 |
--- a/net/filter/mock_filter_context.h |
+++ b/net/filter/mock_filter_context.h |
@@ -12,6 +12,8 @@ |
namespace net { |
+class URLRequestContext; |
+ |
class MockFilterContext : public FilterContext { |
public: |
MockFilterContext(); |
@@ -29,6 +31,9 @@ class MockFilterContext : public FilterContext { |
void SetSdchResponse(bool is_sdch_response) { |
is_sdch_response_ = is_sdch_response; |
} |
+ void SetURLRequestContext(URLRequestContext* context) { |
+ context_ = context; |
+ } |
virtual bool GetMimeType(std::string* mime_type) const OVERRIDE; |
@@ -57,6 +62,9 @@ class MockFilterContext : public FilterContext { |
virtual int GetResponseCode() const OVERRIDE; |
+ // The URLRequestContext associated with the request. |
+ virtual const URLRequestContext* GetURLRequestContext() const OVERRIDE; |
+ |
virtual void RecordPacketStats(StatisticSelector statistic) const OVERRIDE {} |
private: |
@@ -69,6 +77,7 @@ class MockFilterContext : public FilterContext { |
bool is_download_; |
bool is_sdch_response_; |
int response_code_; |
+ URLRequestContext* context_; |
DISALLOW_COPY_AND_ASSIGN(MockFilterContext); |
}; |