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 #include "base/memory/ref_counted_memory.h" | 5 #include "base/memory/ref_counted_memory.h" |
6 #include "base/utf_string_conversions.h" | 6 #include "base/utf_string_conversions.h" |
7 #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h" | 7 #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h" |
8 #include "grit/common_resources.h" | 8 #include "grit/common_resources.h" |
9 #include "grit/generated_resources.h" | 9 #include "grit/generated_resources.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
11 | 11 |
12 // A mock data source (so we can override SendResponse to get at its data). | 12 // A mock data source (so we can override SendResponse to get at its data). |
13 class MockChromeWebUIDataSource : public ChromeWebUIDataSource { | 13 class MockChromeWebUIDataSource : public ChromeWebUIDataSource { |
14 public: | 14 public: |
15 MockChromeWebUIDataSource() | 15 MockChromeWebUIDataSource() |
16 : ChromeWebUIDataSource("host"), | 16 : ChromeWebUIDataSource("host"), |
17 result_request_id_(-1), | 17 result_request_id_(-1), |
18 result_data_(NULL) { | 18 result_data_(NULL) { |
19 } | 19 } |
20 | 20 |
21 virtual ~MockChromeWebUIDataSource() { | |
22 } | |
23 | |
24 // Subvert protected methods. | 21 // Subvert protected methods. |
25 virtual void StartDataRequest( | 22 virtual void StartDataRequest( |
26 const std::string& path, bool is_incognito, int request_id) OVERRIDE { | 23 const std::string& path, bool is_incognito, int request_id) OVERRIDE { |
27 ChromeWebUIDataSource::StartDataRequest(path, is_incognito, request_id); | 24 ChromeWebUIDataSource::StartDataRequest(path, is_incognito, request_id); |
28 } | 25 } |
29 | 26 |
30 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { | 27 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { |
31 return ChromeWebUIDataSource::GetMimeType(path); | 28 return ChromeWebUIDataSource::GetMimeType(path); |
32 } | 29 } |
33 | 30 |
34 // Store response for later comparisons. | 31 // Store response for later comparisons. |
35 virtual void SendResponse(int request_id, RefCountedMemory* data) { | 32 virtual void SendResponse(int request_id, RefCountedMemory* data) { |
36 result_request_id_ = request_id; | 33 result_request_id_ = request_id; |
37 result_data_ = data; | 34 result_data_ = data; |
38 } | 35 } |
39 | 36 |
40 int result_request_id_; | 37 int result_request_id_; |
41 scoped_refptr<RefCountedMemory> result_data_; | 38 scoped_refptr<RefCountedMemory> result_data_; |
| 39 |
| 40 private: |
| 41 virtual ~MockChromeWebUIDataSource() {} |
42 }; | 42 }; |
43 | 43 |
44 class ChromeWebUIDataSourceTest : public testing::Test { | 44 class ChromeWebUIDataSourceTest : public testing::Test { |
45 public: | 45 public: |
46 ChromeWebUIDataSourceTest() {} | 46 ChromeWebUIDataSourceTest() {} |
47 virtual ~ChromeWebUIDataSourceTest() {} | 47 virtual ~ChromeWebUIDataSourceTest() {} |
48 MockChromeWebUIDataSource* source() { return source_.get(); } | 48 MockChromeWebUIDataSource* source() { return source_.get(); } |
49 | 49 |
50 private: | 50 private: |
51 virtual void SetUp() { | 51 virtual void SetUp() { |
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
118 const char* js = "application/javascript"; | 118 const char* js = "application/javascript"; |
119 EXPECT_EQ(source()->GetMimeType(""), html); | 119 EXPECT_EQ(source()->GetMimeType(""), html); |
120 EXPECT_EQ(source()->GetMimeType("foo"), html); | 120 EXPECT_EQ(source()->GetMimeType("foo"), html); |
121 EXPECT_EQ(source()->GetMimeType("foo.html"), html); | 121 EXPECT_EQ(source()->GetMimeType("foo.html"), html); |
122 EXPECT_EQ(source()->GetMimeType(".js"), js); | 122 EXPECT_EQ(source()->GetMimeType(".js"), js); |
123 EXPECT_EQ(source()->GetMimeType("foo.js"), js); | 123 EXPECT_EQ(source()->GetMimeType("foo.js"), js); |
124 EXPECT_EQ(source()->GetMimeType("js"), html); | 124 EXPECT_EQ(source()->GetMimeType("js"), html); |
125 EXPECT_EQ(source()->GetMimeType("foojs"), html); | 125 EXPECT_EQ(source()->GetMimeType("foojs"), html); |
126 EXPECT_EQ(source()->GetMimeType("foo.jsp"), html); | 126 EXPECT_EQ(source()->GetMimeType("foo.jsp"), html); |
127 } | 127 } |
OLD | NEW |