ui/app_list/folder_image_unittest.cc

   1 // Copyright 2014 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 <string>
   6
   7 #include "ui/app_list/folder_image.h"
   8
   9 #include "base/macros.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #include "third_party/skia/include/core/SkBitmap.h"
  12 #include "third_party/skia/include/core/SkColor.h"
  13 #include "ui/app_list/app_list_constants.h"
  14 #include "ui/app_list/app_list_item.h"
  15 #include "ui/app_list/app_list_item_list.h"
  16 #include "ui/app_list/app_list_model.h"
  17 #include "ui/gfx/skia_util.h"
  18
  19 namespace app_list {
  20
  21 namespace {
  22
  23 gfx::ImageSkia CreateSquareBitmapWithColor(int size, SkColor color) {
  24   SkBitmap bitmap;
  25   bitmap.allocN32Pixels(size, size);
  26   bitmap.eraseColor(color);
  27   return gfx::ImageSkia::CreateFrom1xBitmap(bitmap);
  28 }
  29
  30 bool ImagesAreEqual(const gfx::ImageSkia& image1,
  31                     const gfx::ImageSkia& image2) {
  32   return gfx::BitmapsAreEqual(*image1.bitmap(), *image2.bitmap());
  33 }
  34
  35 // Listens for OnFolderImageUpdated and sets a flag upon receiving the signal.
  36 class TestFolderImageObserver : public FolderImageObserver {
  37  public:
  38   TestFolderImageObserver() : updated_flag_(false) {}
  39
  40   bool updated() const { return updated_flag_; }
  41
  42   void Reset() { updated_flag_ = false; }
  43
  44   // FolderImageObserver overrides:
  45   void OnFolderImageUpdated() override { updated_flag_ = true; }
  46
  47  private:
  48   bool updated_flag_;
  49
  50   DISALLOW_COPY_AND_ASSIGN(TestFolderImageObserver);
  51 };
  52
  53 }  // namespace
  54
  55 class FolderImageTest : public testing::Test {
  56  public:
  57   FolderImageTest() : folder_image_(app_list_model_.top_level_item_list()) {}
  58
  59   ~FolderImageTest() override {}
  60
  61   void SetUp() override {
  62     // Populate the AppListModel with three items (to test that the FolderImage
  63     // correctly supports having fewer than four icons).
  64     AddAppWithColoredIcon("app1", SK_ColorRED);
  65     AddAppWithColoredIcon("app2", SK_ColorGREEN);
  66     AddAppWithColoredIcon("app3", SK_ColorBLUE);
  67
  68     observer_.Reset();
  69     folder_image_.AddObserver(&observer_);
  70   }
  71
  72   void TearDown() override { folder_image_.RemoveObserver(&observer_); }
  73
  74  protected:
  75   void AddAppWithColoredIcon(const std::string& id, SkColor icon_color) {
  76     scoped_ptr<AppListItem> item(new AppListItem(id));
  77     item->SetIcon(CreateSquareBitmapWithColor(kListIconSize, icon_color));
  78     app_list_model_.AddItem(item.Pass());
  79   }
  80
  81   AppListModel app_list_model_;
  82
  83   FolderImage folder_image_;
  84
  85   TestFolderImageObserver observer_;
  86
  87  private:
  88   DISALLOW_COPY_AND_ASSIGN(FolderImageTest);
  89 };
  90
  91 TEST_F(FolderImageTest, UpdateListTest) {
  92   gfx::ImageSkia icon1 = folder_image_.icon();
  93
  94   // Call UpdateIcon and ensure that the observer event fired.
  95   folder_image_.UpdateIcon();
  96   EXPECT_TRUE(observer_.updated());
  97   observer_.Reset();
  98   // The icon should not have changed.
  99   EXPECT_TRUE(ImagesAreEqual(icon1, folder_image_.icon()));
 100
 101   // Swap two items. Ensure that the observer fired and the icon changed.
 102   app_list_model_.top_level_item_list()->MoveItem(2, 1);
 103   EXPECT_TRUE(observer_.updated());
 104   observer_.Reset();
 105   gfx::ImageSkia icon2 = folder_image_.icon();
 106   EXPECT_FALSE(ImagesAreEqual(icon1, icon2));
 107
 108   // Swap back items. Ensure that the observer fired and the icon changed back.
 109   app_list_model_.top_level_item_list()->MoveItem(2, 1);
 110   EXPECT_TRUE(observer_.updated());
 111   observer_.Reset();
 112   EXPECT_TRUE(ImagesAreEqual(icon1, folder_image_.icon()));
 113
 114   // Add a new item. Ensure that the observer fired and the icon changed.
 115   AddAppWithColoredIcon("app4", SK_ColorYELLOW);
 116   EXPECT_TRUE(observer_.updated());
 117   observer_.Reset();
 118   gfx::ImageSkia icon3 = folder_image_.icon();
 119   EXPECT_FALSE(ImagesAreEqual(icon1, icon3));
 120
 121   // Add a new item. The observer should not fire, nor should the icon change
 122   // (because it does not affect the first four icons).
 123   AddAppWithColoredIcon("app5", SK_ColorCYAN);
 124   EXPECT_FALSE(observer_.updated());
 125   observer_.Reset();
 126   EXPECT_TRUE(ImagesAreEqual(icon3, folder_image_.icon()));
 127
 128   // Delete an item. Ensure that the observer fired and the icon changed.
 129   app_list_model_.DeleteItem("app2");
 130   EXPECT_TRUE(observer_.updated());
 131   observer_.Reset();
 132   gfx::ImageSkia icon4 = folder_image_.icon();
 133   EXPECT_FALSE(ImagesAreEqual(icon3, icon4));
 134 }
 135
 136 TEST_F(FolderImageTest, UpdateItemTest) {
 137   gfx::ImageSkia icon1 = folder_image_.icon();
 138
 139   // Change an item's icon. Ensure that the observer fired and the icon changed.
 140   app_list_model_.FindItem("app2")
 141       ->SetIcon(CreateSquareBitmapWithColor(kListIconSize, SK_ColorMAGENTA));
 142   EXPECT_TRUE(observer_.updated());
 143   observer_.Reset();
 144   EXPECT_FALSE(ImagesAreEqual(icon1, folder_image_.icon()));
 145 }
 146
 147 }  // namespace app_list