| 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() { | 21 virtual ~MockChromeWebUIDataSource() { |
| 22 result_data_ = NULL; | |
| 23 } | 22 } |
| 24 | 23 |
| 25 // Subvert protected methods. | 24 // Subvert protected methods. |
| 26 virtual void StartDataRequest( | 25 virtual void StartDataRequest( |
| 27 const std::string& path, bool is_incognito, int request_id) OVERRIDE { | 26 const std::string& path, bool is_incognito, int request_id) OVERRIDE { |
| 28 ChromeWebUIDataSource::StartDataRequest(path, is_incognito, request_id); | 27 ChromeWebUIDataSource::StartDataRequest(path, is_incognito, request_id); |
| 29 } | 28 } |
| 30 | 29 |
| 31 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { | 30 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { |
| 32 return ChromeWebUIDataSource::GetMimeType(path); | 31 return ChromeWebUIDataSource::GetMimeType(path); |
| 33 } | 32 } |
| 34 | 33 |
| 35 // Store response for later comparisons. | 34 // Store response for later comparisons. |
| 36 virtual void SendResponse(int request_id, RefCountedMemory* data) { | 35 virtual void SendResponse(int request_id, RefCountedMemory* data) { |
| 37 result_request_id_ = request_id; | 36 result_request_id_ = request_id; |
| 38 result_data_ = data; | 37 result_data_ = data; |
| 39 } | 38 } |
| 40 | 39 |
| 41 int result_request_id_; | 40 int result_request_id_; |
| 42 scoped_refptr<RefCountedMemory> result_data_; | 41 scoped_refptr<RefCountedMemory> result_data_; |
| 43 }; | 42 }; |
| 44 | 43 |
| 45 class ChromeWebUIDataSourceTest : public testing::Test { | 44 class ChromeWebUIDataSourceTest : public testing::Test { |
| 46 public: | 45 public: |
| 47 ChromeWebUIDataSourceTest() {} | 46 ChromeWebUIDataSourceTest() {} |
| 48 MockChromeWebUIDataSource* source() { return source_; } | 47 virtual ~ChromeWebUIDataSourceTest() {} |
| 48 MockChromeWebUIDataSource* source() { return source_.get(); } |
| 49 | 49 |
| 50 private: | 50 private: |
| 51 virtual void SetUp() { | 51 virtual void SetUp() { |
| 52 source_ = new MockChromeWebUIDataSource(); | 52 source_ = make_scoped_refptr(new MockChromeWebUIDataSource()); |
| 53 } | 53 } |
| 54 | 54 |
| 55 virtual void TearDown() { | 55 virtual void TearDown() { |
| 56 } | 56 } |
| 57 | 57 |
| 58 MockChromeWebUIDataSource* source_; | 58 scoped_refptr<MockChromeWebUIDataSource> source_; |
| 59 }; | 59 }; |
| 60 | 60 |
| 61 TEST_F(ChromeWebUIDataSourceTest, EmptyStrings) { | 61 TEST_F(ChromeWebUIDataSourceTest, EmptyStrings) { |
| 62 source()->set_json_path("strings.js"); | 62 source()->set_json_path("strings.js"); |
| 63 source()->StartDataRequest("strings.js" , false, 1); | 63 source()->StartDataRequest("strings.js" , false, 1); |
| 64 EXPECT_EQ(source()->result_request_id_, 1); | 64 EXPECT_EQ(source()->result_request_id_, 1); |
| 65 std::string result(reinterpret_cast<const char*>( | 65 std::string result(reinterpret_cast<const char*>( |
| 66 source()->result_data_->front()), source()->result_data_->size()); | 66 source()->result_data_->front()), source()->result_data_->size()); |
| 67 EXPECT_NE(result.find("var templateData = {"), std::string::npos); | 67 EXPECT_NE(result.find("var templateData = {"), std::string::npos); |
| 68 EXPECT_NE(result.find("};"), std::string::npos); | 68 EXPECT_NE(result.find("};"), std::string::npos); |
| (...skipping 49 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 |