| 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 |