| 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 |