// 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/image_decode_controller.h"
#include "cc/tiles/image_manager.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace cc {

class TestableController : public ImageDecodeController {
 public:
  bool GetTaskForImageAndRef(const DrawImage& image,
                             const TracingInfo& tracing_info,
                             scoped_refptr<TileTask>* task) override {
    *task = nullptr;
    ++number_of_refs_;
    return true;
  }

  void UnrefImage(const DrawImage& image) override {
    ASSERT_GT(number_of_refs_, 0);
    --number_of_refs_;
  }
  DecodedDrawImage GetDecodedImageForDraw(const DrawImage& image) override {
    return DecodedDrawImage(nullptr, kNone_SkFilterQuality);
  }
  void DrawWithImageFinished(const DrawImage& image,
                             const DecodedDrawImage& decoded_image) override {}
  void ReduceCacheUsage() override {}
  void SetShouldAggressivelyFreeResources(
      bool aggressively_free_resources) override {}

  int number_of_refs() const { return number_of_refs_; }

 private:
  int number_of_refs_ = 0;
};

TEST(ImageManagerTest, NullControllerUnrefsImages) {
  TestableController controller;
  ImageManager manager;
  manager.SetImageDecodeController(&controller);

  std::vector<DrawImage> images(10);
  ImageDecodeController::TracingInfo tracing_info;

  ASSERT_EQ(10u, images.size());
  auto tasks = manager.SetPredecodeImages(std::move(images), tracing_info);
  EXPECT_EQ(0u, tasks.size());
  EXPECT_EQ(10, controller.number_of_refs());

  manager.SetImageDecodeController(nullptr);
  EXPECT_EQ(0, controller.number_of_refs());
}

}  // namespace cc