content/browser/android/system_ui_resource_manager_impl_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 "cc/resources/ui_resource_bitmap.h"
   6 #include "content/browser/android/system_ui_resource_manager_impl.h"
   7 #include "content/public/browser/android/ui_resource_client_android.h"
   8 #include "content/public/browser/android/ui_resource_provider.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10 #include "third_party/skia/include/core/SkBitmap.h"
  11 #include "third_party/skia/include/core/SkCanvas.h"
  12
  13 namespace content {
  14
  15 class TestSystemUIResourceManagerImpl
  16     : public content::SystemUIResourceManagerImpl {
  17  public:
  18   TestSystemUIResourceManagerImpl(content::UIResourceProvider* provider)
  19       : SystemUIResourceManagerImpl(provider) {}
  20
  21   virtual ~TestSystemUIResourceManagerImpl() {}
  22
  23   virtual void BuildResource(
  24       ui::SystemUIResourceManager::ResourceType type) override {}
  25
  26   void SetResourceAsLoaded(ui::SystemUIResourceManager::ResourceType type) {
  27     SkBitmap small_bitmap;
  28     SkCanvas canvas(small_bitmap);
  29     small_bitmap.allocPixels(
  30         SkImageInfo::Make(1, 1, kRGBA_8888_SkColorType, kOpaque_SkAlphaType));
  31     canvas.drawColor(SK_ColorWHITE);
  32     small_bitmap.setImmutable();
  33     OnFinishedLoadBitmap(type, &small_bitmap);
  34   }
  35 };
  36
  37 namespace {
  38
  39 const ui::SystemUIResourceManager::ResourceType TEST_RESOURCE_TYPE =
  40     ui::SystemUIResourceManager::OVERSCROLL_GLOW;
  41
  42 class MockUIResourceProvider : public content::UIResourceProvider {
  43  public:
  44   MockUIResourceProvider()
  45       : next_ui_resource_id_(1),
  46         has_layer_tree_host_(true),
  47         system_ui_resource_manager_(this) {}
  48
  49   virtual ~MockUIResourceProvider() {}
  50
  51   virtual cc::UIResourceId CreateUIResource(
  52       content::UIResourceClientAndroid* client) override {
  53     if (!has_layer_tree_host_)
  54       return 0;
  55     cc::UIResourceId id = next_ui_resource_id_++;
  56     client->GetBitmap(id, false);
  57     ui_resource_client_map_[id] = client;
  58     return id;
  59   }
  60
  61   virtual void DeleteUIResource(cc::UIResourceId id) override {
  62     CHECK(has_layer_tree_host_);
  63     ui_resource_client_map_.erase(id);
  64   }
  65
  66   virtual bool SupportsETC1NonPowerOfTwo() const override { return true; }
  67
  68   void LayerTreeHostCleared() {
  69     has_layer_tree_host_ = false;
  70     UIResourceClientMap client_map = ui_resource_client_map_;
  71     ui_resource_client_map_.clear();
  72     for (UIResourceClientMap::iterator iter = client_map.begin();
  73          iter != client_map.end();
  74          iter++) {
  75       iter->second->UIResourceIsInvalid();
  76     }
  77   }
  78
  79   void LayerTreeHostReturned() { has_layer_tree_host_ = true; }
  80
  81   TestSystemUIResourceManagerImpl& GetSystemUIResourceManager() {
  82     return system_ui_resource_manager_;
  83   }
  84
  85   cc::UIResourceId next_ui_resource_id() const { return next_ui_resource_id_; }
  86
  87  private:
  88   typedef base::hash_map<cc::UIResourceId, content::UIResourceClientAndroid*>
  89       UIResourceClientMap;
  90
  91   cc::UIResourceId next_ui_resource_id_;
  92   UIResourceClientMap ui_resource_client_map_;
  93   bool has_layer_tree_host_;
  94
  95   // The UIResourceProvider owns the SystemUIResourceManager.
  96   TestSystemUIResourceManagerImpl system_ui_resource_manager_;
  97 };
  98
  99 }  // namespace
 100
 101 class SystemUIResourceManagerImplTest : public testing::Test {
 102  public:
 103   void PreloadResource(ui::SystemUIResourceManager::ResourceType type) {
 104     ui_resource_provider_.GetSystemUIResourceManager().PreloadResource(type);
 105   }
 106
 107   cc::UIResourceId GetUIResourceId(
 108       ui::SystemUIResourceManager::ResourceType type) {
 109     return ui_resource_provider_.GetSystemUIResourceManager().GetUIResourceId(
 110         type);
 111   }
 112
 113   void LayerTreeHostCleared() { ui_resource_provider_.LayerTreeHostCleared(); }
 114
 115   void LayerTreeHostReturned() {
 116     ui_resource_provider_.LayerTreeHostReturned();
 117   }
 118
 119   void SetResourceAsLoaded(ui::SystemUIResourceManager::ResourceType type) {
 120     ui_resource_provider_.GetSystemUIResourceManager().SetResourceAsLoaded(
 121         type);
 122   }
 123
 124   cc::UIResourceId GetNextUIResourceId() const {
 125     return ui_resource_provider_.next_ui_resource_id();
 126   }
 127
 128  private:
 129   MockUIResourceProvider ui_resource_provider_;
 130 };
 131
 132 TEST_F(SystemUIResourceManagerImplTest, GetResourceAfterBitmapLoaded) {
 133   SetResourceAsLoaded(TEST_RESOURCE_TYPE);
 134   EXPECT_NE(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 135 }
 136
 137 TEST_F(SystemUIResourceManagerImplTest, GetResourceBeforeLoadBitmap) {
 138   EXPECT_EQ(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 139   SetResourceAsLoaded(TEST_RESOURCE_TYPE);
 140   EXPECT_NE(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 141 }
 142
 143 TEST_F(SystemUIResourceManagerImplTest, PreloadEnsureResource) {
 144   // Preloading the resource should trigger bitmap loading, but the actual
 145   // resource id will not be generated until it is explicitly requested.
 146   cc::UIResourceId first_resource_id = GetNextUIResourceId();
 147   PreloadResource(TEST_RESOURCE_TYPE);
 148   SetResourceAsLoaded(TEST_RESOURCE_TYPE);
 149   EXPECT_EQ(first_resource_id, GetNextUIResourceId());
 150   EXPECT_NE(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 151   EXPECT_NE(first_resource_id, GetNextUIResourceId());
 152 }
 153
 154 TEST_F(SystemUIResourceManagerImplTest, ResetLayerTreeHost) {
 155   EXPECT_EQ(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 156   LayerTreeHostCleared();
 157   EXPECT_EQ(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 158   LayerTreeHostReturned();
 159   EXPECT_EQ(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 160
 161   SetResourceAsLoaded(TEST_RESOURCE_TYPE);
 162   EXPECT_NE(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 163   LayerTreeHostCleared();
 164   EXPECT_EQ(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 165   LayerTreeHostReturned();
 166   EXPECT_NE(0, GetUIResourceId(TEST_RESOURCE_TYPE));
 167 }
 168
 169 }  // namespace content