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 "chrome/browser/task_manager/task_manager.h" | 5 #include "chrome/browser/task_manager/task_manager.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
10 #include "base/process_util.h" | 10 #include "base/process_util.h" |
(...skipping 15 matching lines...) Expand all Loading... |
26 const char* kZeroCPUUsage = "0"; | 26 const char* kZeroCPUUsage = "0"; |
27 #endif | 27 #endif |
28 | 28 |
29 class TestResource : public TaskManager::Resource { | 29 class TestResource : public TaskManager::Resource { |
30 public: | 30 public: |
31 TestResource() : refresh_called_(false) {} | 31 TestResource() : refresh_called_(false) {} |
32 | 32 |
33 virtual string16 GetTitle() const OVERRIDE { | 33 virtual string16 GetTitle() const OVERRIDE { |
34 return ASCIIToUTF16("test title"); | 34 return ASCIIToUTF16("test title"); |
35 } | 35 } |
| 36 virtual string16 GetProfileName() const OVERRIDE { |
| 37 return ASCIIToUTF16("test profile"); |
| 38 } |
36 virtual SkBitmap GetIcon() const { return SkBitmap(); } | 39 virtual SkBitmap GetIcon() const { return SkBitmap(); } |
37 virtual base::ProcessHandle GetProcess() const { | 40 virtual base::ProcessHandle GetProcess() const { |
38 return base::GetCurrentProcessHandle(); | 41 return base::GetCurrentProcessHandle(); |
39 } | 42 } |
40 virtual Type GetType() const { return RENDERER; } | 43 virtual Type GetType() const { return RENDERER; } |
41 virtual bool SupportNetworkUsage() const { return false; } | 44 virtual bool SupportNetworkUsage() const { return false; } |
42 virtual void SetSupportNetworkUsage() { NOTREACHED(); } | 45 virtual void SetSupportNetworkUsage() { NOTREACHED(); } |
43 virtual void Refresh() { refresh_called_ = true; } | 46 virtual void Refresh() { refresh_called_ = true; } |
44 bool refresh_called() const { return refresh_called_; } | 47 bool refresh_called() const { return refresh_called_; } |
45 void set_refresh_called(bool refresh_called) { | 48 void set_refresh_called(bool refresh_called) { |
(...skipping 18 matching lines...) Expand all Loading... |
64 TEST_F(TaskManagerTest, Resources) { | 67 TEST_F(TaskManagerTest, Resources) { |
65 TaskManager task_manager; | 68 TaskManager task_manager; |
66 TaskManagerModel* model = task_manager.model_; | 69 TaskManagerModel* model = task_manager.model_; |
67 | 70 |
68 TestResource resource1, resource2; | 71 TestResource resource1, resource2; |
69 | 72 |
70 task_manager.AddResource(&resource1); | 73 task_manager.AddResource(&resource1); |
71 ASSERT_EQ(1, model->ResourceCount()); | 74 ASSERT_EQ(1, model->ResourceCount()); |
72 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); | 75 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); |
73 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0)); | 76 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0)); |
| 77 EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(0)); |
74 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT), | 78 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT), |
75 model->GetResourceNetworkUsage(0)); | 79 model->GetResourceNetworkUsage(0)); |
76 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0)); | 80 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0)); |
77 | 81 |
78 task_manager.AddResource(&resource2); // Will be in the same group. | 82 task_manager.AddResource(&resource2); // Will be in the same group. |
79 ASSERT_EQ(2, model->ResourceCount()); | 83 ASSERT_EQ(2, model->ResourceCount()); |
80 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); | 84 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); |
81 EXPECT_FALSE(model->IsResourceFirstInGroup(1)); | 85 EXPECT_FALSE(model->IsResourceFirstInGroup(1)); |
82 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(1)); | 86 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(1)); |
| 87 EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(1)); |
83 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT).c_str(), | 88 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT).c_str(), |
84 model->GetResourceNetworkUsage(1)); | 89 model->GetResourceNetworkUsage(1)); |
85 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(1)); | 90 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(1)); |
86 | 91 |
87 task_manager.RemoveResource(&resource1); | 92 task_manager.RemoveResource(&resource1); |
88 // Now resource2 will be first in group. | 93 // Now resource2 will be first in group. |
89 ASSERT_EQ(1, model->ResourceCount()); | 94 ASSERT_EQ(1, model->ResourceCount()); |
90 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); | 95 EXPECT_TRUE(model->IsResourceFirstInGroup(0)); |
91 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0)); | 96 EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0)); |
| 97 EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(0)); |
92 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT), | 98 EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT), |
93 model->GetResourceNetworkUsage(0)); | 99 model->GetResourceNetworkUsage(0)); |
94 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0)); | 100 EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0)); |
95 | 101 |
96 task_manager.RemoveResource(&resource2); | 102 task_manager.RemoveResource(&resource2); |
97 EXPECT_EQ(0, model->ResourceCount()); | 103 EXPECT_EQ(0, model->ResourceCount()); |
98 } | 104 } |
99 | 105 |
100 // Tests that the model is calling Refresh() on its resources. | 106 // Tests that the model is calling Refresh() on its resources. |
101 TEST_F(TaskManagerTest, RefreshCalled) { | 107 TEST_F(TaskManagerTest, RefreshCalled) { |
102 MessageLoop loop; | 108 MessageLoop loop; |
103 TaskManager task_manager; | 109 TaskManager task_manager; |
104 TaskManagerModel* model = task_manager.model_; | 110 TaskManagerModel* model = task_manager.model_; |
105 TestResource resource; | 111 TestResource resource; |
106 | 112 |
107 task_manager.AddResource(&resource); | 113 task_manager.AddResource(&resource); |
108 ASSERT_FALSE(resource.refresh_called()); | 114 ASSERT_FALSE(resource.refresh_called()); |
109 model->update_state_ = TaskManagerModel::TASK_PENDING; | 115 model->update_state_ = TaskManagerModel::TASK_PENDING; |
110 model->Refresh(); | 116 model->Refresh(); |
111 ASSERT_TRUE(resource.refresh_called()); | 117 ASSERT_TRUE(resource.refresh_called()); |
112 task_manager.RemoveResource(&resource); | 118 task_manager.RemoveResource(&resource); |
113 } | 119 } |
OLD | NEW |