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