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
3 // found in the LICENSE file.
4
5 #include "base/memory/ref_counted_memory.h"
6 #include "chrome/browser/profiles/profile.h"
7 #include "chrome/browser/ui/webui/theme_source.h"
8 #include "chrome/common/url_constants.h"
9 #include "chrome/test/testing_profile.h"
10 #include "content/browser/browser_thread.h"
11 #include "grit/theme_resources.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13
14 // A mock ThemeSource (so we can override SendResponse to get at its data).
15 class MockThemeSource : public ThemeSource {
16 public:
MockThemeSource(Profile * profile)17 explicit MockThemeSource(Profile* profile)
18 : ThemeSource(profile),
19 result_request_id_(-1),
20 result_data_size_(0) {
21 }
22
SendResponse(int request_id,RefCountedMemory * data)23 virtual void SendResponse(int request_id, RefCountedMemory* data) {
24 result_data_size_ = data ? data->size() : 0;
25 result_request_id_ = request_id;
26 }
27
28 int result_request_id_;
29 size_t result_data_size_;
30
31 private:
~MockThemeSource()32 ~MockThemeSource() {}
33 };
34
35 class WebUISourcesTest : public testing::Test {
36 public:
WebUISourcesTest()37 WebUISourcesTest() : ui_thread_(BrowserThread::UI, MessageLoop::current()) {}
38
profile() const39 TestingProfile* profile() const { return profile_.get(); }
theme_source() const40 MockThemeSource* theme_source() const { return theme_source_.get(); }
41 private:
SetUp()42 virtual void SetUp() {
43 profile_.reset(new TestingProfile());
44 theme_source_ = new MockThemeSource(profile_.get());
45 }
46
TearDown()47 virtual void TearDown() {
48 theme_source_ = NULL;
49 profile_.reset(NULL);
50 }
51
52 MessageLoop loop_;
53 BrowserThread ui_thread_;
54
55 scoped_ptr<TestingProfile> profile_;
56 scoped_refptr<MockThemeSource> theme_source_;
57 };
58
TEST_F(WebUISourcesTest,ThemeSourceMimeTypes)59 TEST_F(WebUISourcesTest, ThemeSourceMimeTypes) {
60 EXPECT_EQ(theme_source()->GetMimeType("css/newtab.css"), "text/css");
61 EXPECT_EQ(theme_source()->GetMimeType("css/newtab.css?foo"), "text/css");
62 EXPECT_EQ(theme_source()->GetMimeType("WRONGURL"), "image/png");
63 }
64
TEST_F(WebUISourcesTest,ThemeSourceImages)65 TEST_F(WebUISourcesTest, ThemeSourceImages) {
66 // We used to PNGEncode the images ourselves, but encoder differences
67 // invalidated that. We now just check that the image exists.
68 theme_source()->StartDataRequest("IDR_THEME_FRAME_INCOGNITO", true, 1);
69 size_t min = 0;
70 EXPECT_EQ(theme_source()->result_request_id_, 1);
71 EXPECT_GT(theme_source()->result_data_size_, min);
72
73 theme_source()->StartDataRequest("IDR_THEME_TOOLBAR", true, 2);
74 EXPECT_EQ(theme_source()->result_request_id_, 2);
75 EXPECT_GT(theme_source()->result_data_size_, min);
76 }
77
TEST_F(WebUISourcesTest,ThemeSourceCSS)78 TEST_F(WebUISourcesTest, ThemeSourceCSS) {
79 BrowserThread io_thread(BrowserThread::IO, MessageLoop::current());
80 // Generating the test data for the NTP CSS would just involve copying the
81 // method, or being super brittle and hard-coding the result (requiring
82 // an update to the unittest every time the CSS template changes), so we
83 // just check for a successful request and data that is non-null.
84 size_t empty_size = 0;
85
86 theme_source()->StartDataRequest("css/newtab.css", false, 1);
87 EXPECT_EQ(theme_source()->result_request_id_, 1);
88 EXPECT_NE(theme_source()->result_data_size_, empty_size);
89
90 theme_source()->StartDataRequest("css/newtab.css?pie", false, 3);
91 EXPECT_EQ(theme_source()->result_request_id_, 3);
92 EXPECT_NE(theme_source()->result_data_size_, empty_size);
93
94 // Check that we send NULL back when we can't find what we're looking for.
95 theme_source()->StartDataRequest("css/WRONGURL", false, 7);
96 EXPECT_EQ(theme_source()->result_request_id_, 7);
97 EXPECT_EQ(theme_source()->result_data_size_, empty_size);
98 }
99