OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/bind.h" | |
6 #include "base/memory/ref_counted_memory.h" | |
7 #include "base/utf_string_conversions.h" | |
8 #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h" | |
9 #include "grit/generated_resources.h" | |
10 #include "grit/webui_resources.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 class MockChromeWebUIDataSource : public ChromeWebUIDataSource { | |
14 public: | |
15 MockChromeWebUIDataSource() | |
16 : ChromeWebUIDataSource("host"), | |
17 result_data_(NULL) { | |
18 } | |
19 | |
20 void StartDataRequest(const std::string& path) { | |
21 ChromeWebUIDataSource::StartDataRequest( | |
22 path, | |
23 false, | |
24 base::Bind(&MockChromeWebUIDataSource::SendResult, this)); | |
25 } | |
26 | |
27 std::string GetMimeType(const std::string& path) const { | |
28 return ChromeWebUIDataSource::GetMimeType(path); | |
29 } | |
30 | |
31 scoped_refptr<base::RefCountedMemory> result_data_; | |
32 | |
33 private: | |
34 // Store response for later comparisons. | |
35 void SendResult(base::RefCountedMemory* data) { | |
36 result_data_ = data; | |
37 } | |
38 | |
39 virtual ~MockChromeWebUIDataSource() {} | |
40 }; | |
41 | |
42 class ChromeWebUIDataSourceTest : public testing::Test { | |
43 public: | |
44 ChromeWebUIDataSourceTest() {} | |
45 virtual ~ChromeWebUIDataSourceTest() {} | |
46 MockChromeWebUIDataSource* source() { return source_.get(); } | |
47 | |
48 private: | |
49 virtual void SetUp() { | |
50 source_ = make_scoped_refptr(new MockChromeWebUIDataSource()); | |
51 } | |
52 | |
53 virtual void TearDown() { | |
54 } | |
55 | |
56 scoped_refptr<MockChromeWebUIDataSource> source_; | |
57 }; | |
58 | |
59 TEST_F(ChromeWebUIDataSourceTest, EmptyStrings) { | |
60 source()->SetJsonPath("strings.js"); | |
61 source()->StartDataRequest("strings.js"); | |
62 std::string result(reinterpret_cast<const char*>( | |
63 source()->result_data_->front()), source()->result_data_->size()); | |
64 EXPECT_NE(result.find("var templateData = {"), std::string::npos); | |
65 EXPECT_NE(result.find("};"), std::string::npos); | |
66 } | |
67 | |
68 TEST_F(ChromeWebUIDataSourceTest, SomeStrings) { | |
69 source()->SetJsonPath("strings.js"); | |
70 source()->AddString("planet", ASCIIToUTF16("pluto")); | |
71 source()->AddLocalizedString("button", IDS_OK); | |
72 source()->StartDataRequest("strings.js"); | |
73 std::string result(reinterpret_cast<const char*>( | |
74 source()->result_data_->front()), source()->result_data_->size()); | |
75 EXPECT_NE(result.find("\"planet\":\"pluto\""), std::string::npos); | |
76 EXPECT_NE(result.find("\"button\":\"OK\""), std::string::npos); | |
77 } | |
78 | |
79 TEST_F(ChromeWebUIDataSourceTest, DefaultResource) { | |
80 source()->SetDefaultResource(IDR_WEBUI_I18N_PROCESS_JS); | |
81 source()->StartDataRequest("foobar" ); | |
82 std::string result( | |
83 reinterpret_cast<const char*>(source()->result_data_->front()), | |
84 source()->result_data_->size()); | |
85 EXPECT_NE(result.find("i18nTemplate.process"), std::string::npos); | |
86 source()->StartDataRequest("strings.js"); | |
87 result = std::string( | |
88 reinterpret_cast<const char*>(source()->result_data_->front()), | |
89 source()->result_data_->size()); | |
90 EXPECT_NE(result.find("i18nTemplate.process"), std::string::npos); | |
91 } | |
92 | |
93 TEST_F(ChromeWebUIDataSourceTest, NamedResource) { | |
94 source()->SetDefaultResource(IDR_WEBUI_I18N_PROCESS_JS); | |
95 source()->AddResourcePath("foobar", IDR_WEBUI_I18N_TEMPLATE_JS); | |
96 source()->StartDataRequest("foobar"); | |
97 std::string result( | |
98 reinterpret_cast<const char*>(source()->result_data_->front()), | |
99 source()->result_data_->size()); | |
100 EXPECT_NE(result.find("var i18nTemplate"), std::string::npos); | |
101 source()->StartDataRequest("strings.js"); | |
102 result = std::string( | |
103 reinterpret_cast<const char*>(source()->result_data_->front()), | |
104 source()->result_data_->size()); | |
105 EXPECT_NE(result.find("i18nTemplate.process"), std::string::npos); | |
106 } | |
107 | |
108 TEST_F(ChromeWebUIDataSourceTest, MimeType) { | |
109 const char* html = "text/html"; | |
110 const char* js = "application/javascript"; | |
111 EXPECT_EQ(source()->GetMimeType(""), html); | |
112 EXPECT_EQ(source()->GetMimeType("foo"), html); | |
113 EXPECT_EQ(source()->GetMimeType("foo.html"), html); | |
114 EXPECT_EQ(source()->GetMimeType(".js"), js); | |
115 EXPECT_EQ(source()->GetMimeType("foo.js"), js); | |
116 EXPECT_EQ(source()->GetMimeType("js"), html); | |
117 EXPECT_EQ(source()->GetMimeType("foojs"), html); | |
118 EXPECT_EQ(source()->GetMimeType("foo.jsp"), html); | |
119 } | |
OLD | NEW |