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 "chrome/browser/profiles/profile.h" | 6 #include "chrome/browser/profiles/profile.h" |
7 #include "chrome/browser/ui/webui/theme_source.h" | 7 #include "chrome/browser/ui/webui/theme_source.h" |
8 #include "chrome/common/url_constants.h" | 8 #include "chrome/common/url_constants.h" |
9 #include "chrome/test/base/testing_browser_process_test.h" | |
10 #include "chrome/test/base/testing_profile.h" | 9 #include "chrome/test/base/testing_profile.h" |
11 #include "content/browser/browser_thread.h" | 10 #include "content/browser/browser_thread.h" |
12 #include "grit/theme_resources.h" | 11 #include "grit/theme_resources.h" |
13 #include "grit/theme_resources_standard.h" | 12 #include "grit/theme_resources_standard.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
15 | 14 |
16 // A mock ThemeSource (so we can override SendResponse to get at its data). | 15 // A mock ThemeSource (so we can override SendResponse to get at its data). |
17 class MockThemeSource : public ThemeSource { | 16 class MockThemeSource : public ThemeSource { |
18 public: | 17 public: |
19 explicit MockThemeSource(Profile* profile) | 18 explicit MockThemeSource(Profile* profile) |
20 : ThemeSource(profile), | 19 : ThemeSource(profile), |
21 result_request_id_(-1), | 20 result_request_id_(-1), |
22 result_data_size_(0) { | 21 result_data_size_(0) { |
23 } | 22 } |
24 | 23 |
25 virtual void SendResponse(int request_id, RefCountedMemory* data) { | 24 virtual void SendResponse(int request_id, RefCountedMemory* data) { |
26 result_data_size_ = data ? data->size() : 0; | 25 result_data_size_ = data ? data->size() : 0; |
27 result_request_id_ = request_id; | 26 result_request_id_ = request_id; |
28 } | 27 } |
29 | 28 |
30 int result_request_id_; | 29 int result_request_id_; |
31 size_t result_data_size_; | 30 size_t result_data_size_; |
32 | 31 |
33 private: | 32 private: |
34 ~MockThemeSource() {} | 33 ~MockThemeSource() {} |
35 }; | 34 }; |
36 | 35 |
37 class WebUISourcesTest : public TestingBrowserProcessTest { | 36 class WebUISourcesTest : public testing::Test { |
38 public: | 37 public: |
39 WebUISourcesTest() : ui_thread_(BrowserThread::UI, MessageLoop::current()) {} | 38 WebUISourcesTest() : ui_thread_(BrowserThread::UI, MessageLoop::current()) {} |
40 | 39 |
41 TestingProfile* profile() const { return profile_.get(); } | 40 TestingProfile* profile() const { return profile_.get(); } |
42 MockThemeSource* theme_source() const { return theme_source_.get(); } | 41 MockThemeSource* theme_source() const { return theme_source_.get(); } |
43 private: | 42 private: |
44 virtual void SetUp() { | 43 virtual void SetUp() { |
45 profile_.reset(new TestingProfile()); | 44 profile_.reset(new TestingProfile()); |
46 theme_source_ = new MockThemeSource(profile_.get()); | 45 theme_source_ = new MockThemeSource(profile_.get()); |
47 } | 46 } |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 | 90 |
92 theme_source()->StartDataRequest("css/newtab.css?pie", false, 3); | 91 theme_source()->StartDataRequest("css/newtab.css?pie", false, 3); |
93 EXPECT_EQ(theme_source()->result_request_id_, 3); | 92 EXPECT_EQ(theme_source()->result_request_id_, 3); |
94 EXPECT_NE(theme_source()->result_data_size_, empty_size); | 93 EXPECT_NE(theme_source()->result_data_size_, empty_size); |
95 | 94 |
96 // Check that we send NULL back when we can't find what we're looking for. | 95 // Check that we send NULL back when we can't find what we're looking for. |
97 theme_source()->StartDataRequest("css/WRONGURL", false, 7); | 96 theme_source()->StartDataRequest("css/WRONGURL", false, 7); |
98 EXPECT_EQ(theme_source()->result_request_id_, 7); | 97 EXPECT_EQ(theme_source()->result_request_id_, 7); |
99 EXPECT_EQ(theme_source()->result_data_size_, empty_size); | 98 EXPECT_EQ(theme_source()->result_data_size_, empty_size); |
100 } | 99 } |
OLD | NEW |