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