Implement OnMemoryStateChange for Various CC Classes

cc/tiles/gpu_image_decode_controller_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "cc/tiles/gpu_image_decode_controller.h"
 
 #include "cc/playback/draw_image.h"
 #include "cc/test/test_context_provider.h"
 #include "cc/test/test_tile_task_runner.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 619 matching lines @@
   EXPECT_FALSE(controller.DiscardableIsLockedForTesting(draw_image));
 }
 
 TEST(GpuImageDecodeControllerTest, GetDecodedImageForDrawAtRasterDecode) {
   auto context_provider = TestContextProvider::Create();
   context_provider->BindToCurrentThread();
   TestGpuImageDecodeController controller(context_provider.get());
   bool is_decomposable = true;
   SkFilterQuality quality = kHigh_SkFilterQuality;
 
-  controller.SetCachedItemLimitForTesting(0);
   controller.SetCachedBytesLimitForTesting(0);
 
   sk_sp<SkImage> image = CreateImage(100, 100);
   DrawImage draw_image(image, SkIRect::MakeWH(image->width(), image->height()),
                        quality,
                        CreateMatrix(SkSize::Make(1.0f, 1.0f), is_decomposable));
 
   scoped_refptr<TileTask> task;
   bool need_unref = controller.GetTaskForImageAndRef(
       draw_image, ImageDecodeController::TracingInfo(), &task);
@@ skipping 203 matching lines @@
   EXPECT_FALSE(controller.DiscardableIsLockedForTesting(draw_image));
 }
 
 TEST(GpuImageDecodeControllerTest, AtRasterUsedDirectlyIfSpaceAllows) {
   auto context_provider = TestContextProvider::Create();
   context_provider->BindToCurrentThread();
   TestGpuImageDecodeController controller(context_provider.get());
   bool is_decomposable = true;
   SkFilterQuality quality = kHigh_SkFilterQuality;
 
-  controller.SetCachedItemLimitForTesting(0);
   controller.SetCachedBytesLimitForTesting(0);
 
   sk_sp<SkImage> image = CreateImage(100, 100);
   DrawImage draw_image(image, SkIRect::MakeWH(image->width(), image->height()),
                        quality,
                        CreateMatrix(SkSize::Make(0.5f, 0.5f), is_decomposable));
 
   scoped_refptr<TileTask> task;
   bool need_unref = controller.GetTaskForImageAndRef(
       draw_image, ImageDecodeController::TracingInfo(), &task);
   EXPECT_FALSE(need_unref);
   EXPECT_FALSE(task);
 
   // Must hold context lock before calling GetDecodedImageForDraw /
   // DrawWithImageFinished.
   ContextProvider::ScopedContextLock context_lock(context_provider.get());
   DecodedDrawImage decoded_draw_image =
       controller.GetDecodedImageForDraw(draw_image);
   EXPECT_TRUE(decoded_draw_image.image());
   EXPECT_TRUE(decoded_draw_image.image()->isTextureBacked());
   EXPECT_TRUE(decoded_draw_image.is_at_raster_decode());
   EXPECT_FALSE(controller.DiscardableIsLockedForTesting(draw_image));
 
-  controller.SetCachedItemLimitForTesting(1000);
   controller.SetCachedBytesLimitForTesting(96 * 1024 * 1024);
 
   // Finish our draw after increasing the memory limit, image should be added to
   // cache.
   controller.DrawWithImageFinished(draw_image, decoded_draw_image);
 
   scoped_refptr<TileTask> another_task;
   bool another_task_needs_unref = controller.GetTaskForImageAndRef(
       draw_image, ImageDecodeController::TracingInfo(), &task);
   EXPECT_TRUE(another_task_needs_unref);
   EXPECT_FALSE(another_task);
   controller.UnrefImage(draw_image);
 }
 
 TEST(GpuImageDecodeControllerTest,
      GetDecodedImageForDrawAtRasterDecodeMultipleTimes) {
   auto context_provider = TestContextProvider::Create();
   context_provider->BindToCurrentThread();
   TestGpuImageDecodeController controller(context_provider.get());
   bool is_decomposable = true;
   SkFilterQuality quality = kHigh_SkFilterQuality;
 
-  controller.SetCachedItemLimitForTesting(0);
   controller.SetCachedBytesLimitForTesting(0);
 
   sk_sp<SkImage> image = CreateImage(100, 100);
   DrawImage draw_image(image, SkIRect::MakeWH(image->width(), image->height()),
                        quality,
                        CreateMatrix(SkSize::Make(0.5f, 0.5f), is_decomposable));
 
   // Must hold context lock before calling GetDecodedImageForDraw /
   // DrawWithImageFinished.
   ContextProvider::ScopedContextLock context_lock(context_provider.get());
@@ skipping 378 matching lines @@
 
   // Do an at-raster decode of the above image that *does* require mips.
   DrawImage draw_image_mips(
       image, SkIRect::MakeWH(image->width(), image->height()), quality,
       CreateMatrix(SkSize::Make(0.6f, 0.6f), is_decomposable));
   DecodedDrawImage decoded_draw_image =
       controller.GetDecodedImageForDraw(draw_image_mips);
   controller.DrawWithImageFinished(draw_image_mips, decoded_draw_image);
 }
 
+TEST(GpuImageDecodeControllerTest, MemoryStateSuspended) {
+  auto context_provider = TestContextProvider::Create();
+  context_provider->BindToCurrentThread();
+  TestGpuImageDecodeController controller(context_provider.get());
+
+  // First Insert an image into our cache.
+  sk_sp<SkImage> image = CreateImage(1, 1);
+  bool is_decomposable = true;
+  SkMatrix matrix = CreateMatrix(SkSize::Make(1.0f, 1.0f), is_decomposable);
+  DrawImage draw_image(image, SkIRect::MakeWH(image->width(), image->height()),
+                       kLow_SkFilterQuality, matrix);
+  scoped_refptr<TileTask> task;
+  bool need_unref = controller.GetTaskForImageAndRef(
+      draw_image, ImageDecodeController::TracingInfo(), &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  TestTileTaskRunner::ProcessTask(task->dependencies()[0].get());
+  TestTileTaskRunner::ProcessTask(task.get());
+  controller.UnrefImage(draw_image);
+
+  // The image should be cached.
+  DCHECK_GT(controller.GetBytesUsedForTesting(), 0u);
+  DCHECK_EQ(controller.GetNumCacheEntriesForTesting(), 1u);
+
+  // Set us to the not visible state (prerequisite for SUSPENDED).
+  controller.SetShouldAggressivelyFreeResources(true);
+
+  // Image should be cached, but not using memory budget.
+  DCHECK_EQ(controller.GetBytesUsedForTesting(), 0u);
+  DCHECK_EQ(controller.GetNumCacheEntriesForTesting(), 1u);
+
+  // Set us to the SUSPENDED state.
+  controller.OnMemoryStateChange(base::MemoryState::SUSPENDED);
+
+  // Nothing should be cached.
+  DCHECK_EQ(controller.GetBytesUsedForTesting(), 0u);
+  DCHECK_EQ(controller.GetNumCacheEntriesForTesting(), 0u);
+
+  // Attempts to get a task for the image should fail, as we have no space (at
+  // raster only).
+  need_unref = controller.GetTaskForImageAndRef(
+      draw_image, ImageDecodeController::TracingInfo(), &task);
+  EXPECT_FALSE(need_unref);
+  EXPECT_FALSE(task);
+
+  // Restore us to visible and NORMAL memory state.
+  controller.OnMemoryStateChange(base::MemoryState::NORMAL);
+  controller.SetShouldAggressivelyFreeResources(false);
+
+  // We should now be able to create a task again (space available).
+  need_unref = controller.GetTaskForImageAndRef(
+      draw_image, ImageDecodeController::TracingInfo(), &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  TestTileTaskRunner::ProcessTask(task->dependencies()[0].get());
+  TestTileTaskRunner::ProcessTask(task.get());
+  controller.UnrefImage(draw_image);
+}
+
 }  // namespace
 }  // namespace cc