src/chrome/browser/task_manager/task_manager_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/task_manager/task_manager.h"
   6
   7 #include <string>
   8
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/strings/utf_string_conversions.h"
  11 #include "chrome/browser/task_manager/resource_provider.h"
  12 #include "grit/chromium_strings.h"
  13 #include "grit/generated_resources.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15 #include "ui/base/l10n/l10n_util.h"
  16 #include "ui/gfx/image/image_skia.h"
  17
  18 namespace {
  19
  20 #if defined(OS_MACOSX)
  21 // From task_manager.cc:
  22 // Activity Monitor shows %cpu with one decimal digit -- be
  23 // consistent with that.
  24 const char* kZeroCPUUsage = "0.0";
  25 #else
  26 const char* kZeroCPUUsage = "0";
  27 #endif
  28
  29 class TestResource : public task_manager::Resource {
  30  public:
  31   TestResource() : refresh_called_(false) {}
  32
  33   virtual string16 GetTitle() const OVERRIDE {
  34     return ASCIIToUTF16("test title");
  35   }
  36   virtual string16 GetProfileName() const OVERRIDE {
  37     return ASCIIToUTF16("test profile");
  38   }
  39   virtual gfx::ImageSkia GetIcon() const OVERRIDE { return gfx::ImageSkia(); }
  40   virtual base::ProcessHandle GetProcess() const OVERRIDE {
  41     return base::GetCurrentProcessHandle();
  42   }
  43   virtual int GetUniqueChildProcessId() const OVERRIDE {
  44     // In reality the unique child process ID is not the actual process ID,
  45     // but for testing purposes it shouldn't make difference.
  46     return static_cast<int>(base::GetCurrentProcId());
  47   }
  48   virtual Type GetType() const OVERRIDE { return RENDERER; }
  49   virtual bool SupportNetworkUsage() const OVERRIDE { return false; }
  50   virtual void SetSupportNetworkUsage() OVERRIDE { NOTREACHED(); }
  51   virtual void Refresh() OVERRIDE { refresh_called_ = true; }
  52   bool refresh_called() const { return refresh_called_; }
  53   void set_refresh_called(bool refresh_called) {
  54     refresh_called_ = refresh_called;
  55   }
  56
  57  private:
  58   bool refresh_called_;
  59 };
  60
  61 }  // namespace
  62
  63 class TaskManagerTest : public testing::Test {
  64 };
  65
  66 TEST_F(TaskManagerTest, Basic) {
  67   TaskManager task_manager;
  68   TaskManagerModel* model = task_manager.model_.get();
  69   EXPECT_EQ(0, model->ResourceCount());
  70 }
  71
  72 TEST_F(TaskManagerTest, Resources) {
  73   TaskManager task_manager;
  74   TaskManagerModel* model = task_manager.model_.get();
  75
  76   TestResource resource1, resource2;
  77
  78   task_manager.AddResource(&resource1);
  79   ASSERT_EQ(1, model->ResourceCount());
  80   EXPECT_TRUE(model->IsResourceFirstInGroup(0));
  81   EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0));
  82   EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(0));
  83   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT),
  84             model->GetResourceNetworkUsage(0));
  85   EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0));
  86
  87   task_manager.AddResource(&resource2);  // Will be in the same group.
  88   ASSERT_EQ(2, model->ResourceCount());
  89   EXPECT_TRUE(model->IsResourceFirstInGroup(0));
  90   EXPECT_FALSE(model->IsResourceFirstInGroup(1));
  91   EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(1));
  92   EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(1));
  93   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT).c_str(),
  94             model->GetResourceNetworkUsage(1));
  95   EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(1));
  96
  97   task_manager.RemoveResource(&resource1);
  98   // Now resource2 will be first in group.
  99   ASSERT_EQ(1, model->ResourceCount());
 100   EXPECT_TRUE(model->IsResourceFirstInGroup(0));
 101   EXPECT_EQ(ASCIIToUTF16("test title"), model->GetResourceTitle(0));
 102   EXPECT_EQ(ASCIIToUTF16("test profile"), model->GetResourceProfileName(0));
 103   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_TASK_MANAGER_NA_CELL_TEXT),
 104             model->GetResourceNetworkUsage(0));
 105   EXPECT_EQ(ASCIIToUTF16(kZeroCPUUsage), model->GetResourceCPUUsage(0));
 106
 107   task_manager.RemoveResource(&resource2);
 108   EXPECT_EQ(0, model->ResourceCount());
 109 }
 110
 111 // Tests that the model is calling Refresh() on its resources.
 112 TEST_F(TaskManagerTest, RefreshCalled) {
 113   base::MessageLoop loop;
 114   TaskManager task_manager;
 115   TaskManagerModel* model = task_manager.model_.get();
 116   TestResource resource;
 117
 118   task_manager.AddResource(&resource);
 119   ASSERT_FALSE(resource.refresh_called());
 120   model->update_state_ = TaskManagerModel::TASK_PENDING;
 121   model->Refresh();
 122   ASSERT_TRUE(resource.refresh_called());
 123   task_manager.RemoveResource(&resource);
 124 }