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 |