OLD | NEW |
(Empty) | |
| 1 // Copyright 2016 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 <vector> |
| 6 |
| 7 #include "base/bind.h" |
| 8 #include "cc/tiles/decoded_image_tracker.h" |
| 9 #include "cc/tiles/image_controller.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 |
| 12 namespace cc { |
| 13 |
| 14 class TestImageController : public ImageController { |
| 15 public: |
| 16 TestImageController() : ImageController(nullptr, nullptr) {} |
| 17 |
| 18 void UnlockImageDecode(ImageDecodeRequestId id) override { |
| 19 auto it = std::find(locked_ids_.begin(), locked_ids_.end(), id); |
| 20 ASSERT_FALSE(it == locked_ids_.end()); |
| 21 locked_ids_.erase(it); |
| 22 } |
| 23 |
| 24 ImageDecodeRequestId QueueImageDecode( |
| 25 sk_sp<const SkImage> image, |
| 26 const ImageDecodedCallback& callback) override { |
| 27 auto id = next_id_++; |
| 28 locked_ids_.push_back(id); |
| 29 callback.Run(id); |
| 30 return id; |
| 31 } |
| 32 |
| 33 size_t num_locked_images() { return locked_ids_.size(); } |
| 34 |
| 35 private: |
| 36 ImageDecodeRequestId next_id_ = 1; |
| 37 std::vector<ImageDecodeRequestId> locked_ids_; |
| 38 }; |
| 39 |
| 40 class DecodedImageTrackerTest : public testing::Test { |
| 41 public: |
| 42 void SetUp() override { |
| 43 decoded_image_tracker_.set_image_controller(image_controller()); |
| 44 } |
| 45 |
| 46 TestImageController* image_controller() { return &image_controller_; } |
| 47 DecodedImageTracker* decoded_image_tracker() { |
| 48 return &decoded_image_tracker_; |
| 49 } |
| 50 |
| 51 private: |
| 52 TestImageController image_controller_; |
| 53 DecodedImageTracker decoded_image_tracker_; |
| 54 }; |
| 55 |
| 56 TEST_F(DecodedImageTrackerTest, QueueImageLocksImages) { |
| 57 bool locked = false; |
| 58 decoded_image_tracker()->QueueImageDecode( |
| 59 nullptr, base::Bind([](bool* locked) { *locked = true; }, |
| 60 base::Unretained(&locked))); |
| 61 EXPECT_TRUE(locked); |
| 62 EXPECT_EQ(1u, image_controller()->num_locked_images()); |
| 63 } |
| 64 |
| 65 TEST_F(DecodedImageTrackerTest, NotifyFrameFinishedUnlocksImages) { |
| 66 bool locked = false; |
| 67 decoded_image_tracker()->QueueImageDecode( |
| 68 nullptr, base::Bind([](bool* locked) { *locked = true; }, |
| 69 base::Unretained(&locked))); |
| 70 EXPECT_TRUE(locked); |
| 71 EXPECT_EQ(1u, image_controller()->num_locked_images()); |
| 72 |
| 73 decoded_image_tracker()->NotifyFrameFinished(); |
| 74 EXPECT_EQ(1u, image_controller()->num_locked_images()); |
| 75 |
| 76 locked = false; |
| 77 decoded_image_tracker()->QueueImageDecode( |
| 78 nullptr, base::Bind([](bool* locked) { *locked = true; }, |
| 79 base::Unretained(&locked))); |
| 80 EXPECT_TRUE(locked); |
| 81 EXPECT_EQ(2u, image_controller()->num_locked_images()); |
| 82 |
| 83 decoded_image_tracker()->NotifyFrameFinished(); |
| 84 EXPECT_EQ(1u, image_controller()->num_locked_images()); |
| 85 |
| 86 decoded_image_tracker()->NotifyFrameFinished(); |
| 87 EXPECT_EQ(0u, image_controller()->num_locked_images()); |
| 88 } |
| 89 |
| 90 } // namespace cc |
OLD | NEW |