OLD | NEW |
| (Empty) |
1 // Copyright (c) 2010 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 "webkit/glue/plugins/ppapi_unittest.h" | |
6 | |
7 #include "webkit/glue/plugins/pepper_resource_tracker.h" | |
8 #include "webkit/glue/plugins/mock_resource.h" | |
9 | |
10 namespace pepper { | |
11 | |
12 namespace { | |
13 | |
14 class TrackedMockResource : public MockResource { | |
15 public: | |
16 static int tracked_objects_alive; | |
17 | |
18 TrackedMockResource(PluginModule* module) : MockResource(module) { | |
19 tracked_objects_alive++; | |
20 } | |
21 ~TrackedMockResource() { | |
22 tracked_objects_alive--; | |
23 } | |
24 }; | |
25 | |
26 int TrackedMockResource::tracked_objects_alive = 0; | |
27 | |
28 } // namespace | |
29 | |
30 class ResourceTrackerTest : public PpapiUnittest { | |
31 public: | |
32 ResourceTrackerTest() { | |
33 } | |
34 | |
35 virtual void SetUp() { | |
36 PpapiUnittest::SetUp(); | |
37 ResourceTracker::SetSingletonOverride(&tracker_); | |
38 } | |
39 virtual void TearDown() { | |
40 ResourceTracker::ClearSingletonOverride(); | |
41 PpapiUnittest::TearDown(); | |
42 } | |
43 | |
44 ResourceTracker& tracker() { return tracker_; } | |
45 | |
46 private: | |
47 ResourceTracker tracker_; | |
48 }; | |
49 | |
50 TEST_F(ResourceTrackerTest, Ref) { | |
51 ASSERT_EQ(0, TrackedMockResource::tracked_objects_alive); | |
52 EXPECT_EQ(0u, tracker().GetLiveObjectsForModule(module())); | |
53 { | |
54 scoped_refptr<TrackedMockResource> new_resource( | |
55 new TrackedMockResource(module())); | |
56 ASSERT_EQ(1, TrackedMockResource::tracked_objects_alive); | |
57 | |
58 // Since we haven't gotten a PP_Resource, it's not associated with the | |
59 // module. | |
60 EXPECT_EQ(0u, tracker().GetLiveObjectsForModule(module())); | |
61 } | |
62 ASSERT_EQ(0, TrackedMockResource::tracked_objects_alive); | |
63 | |
64 // Make a new resource and get it as a PP_Resource. | |
65 PP_Resource resource_id = 0; | |
66 { | |
67 scoped_refptr<TrackedMockResource> new_resource( | |
68 new TrackedMockResource(module())); | |
69 ASSERT_EQ(1, TrackedMockResource::tracked_objects_alive); | |
70 resource_id = new_resource->GetReference(); | |
71 EXPECT_EQ(1u, tracker().GetLiveObjectsForModule(module())); | |
72 | |
73 // Resource IDs should be consistent. | |
74 PP_Resource resource_id_2 = new_resource->GetReference(); | |
75 ASSERT_EQ(resource_id, resource_id_2); | |
76 } | |
77 | |
78 // This time it should not have been deleted since the PP_Resource carries | |
79 // a ref. | |
80 ASSERT_EQ(1, TrackedMockResource::tracked_objects_alive); | |
81 | |
82 // Now we have two refs, derefing twice should delete the object. | |
83 tracker().UnrefResource(resource_id); | |
84 ASSERT_EQ(1, TrackedMockResource::tracked_objects_alive); | |
85 tracker().UnrefResource(resource_id); | |
86 ASSERT_EQ(0, TrackedMockResource::tracked_objects_alive); | |
87 } | |
88 | |
89 TEST_F(ResourceTrackerTest, ForceDelete) { | |
90 // Make two resources. | |
91 scoped_refptr<TrackedMockResource> resource1( | |
92 new TrackedMockResource(module())); | |
93 PP_Resource pp_resource1 = resource1->GetReference(); | |
94 scoped_refptr<TrackedMockResource> resource2( | |
95 new TrackedMockResource(module())); | |
96 PP_Resource pp_resource2 = resource2->GetReference(); | |
97 | |
98 // Keep an "internal" ref to only the first (the PP_Resource also holds a | |
99 // ref to each resource on behalf of the plugin). | |
100 resource2 = NULL; | |
101 | |
102 ASSERT_EQ(2, TrackedMockResource::tracked_objects_alive); | |
103 EXPECT_EQ(2u, tracker().GetLiveObjectsForModule(module())); | |
104 | |
105 // Force delete both refs. | |
106 tracker().ForceDeletePluginResourceRefs(pp_resource1); | |
107 tracker().ForceDeletePluginResourceRefs(pp_resource2); | |
108 EXPECT_EQ(0u, tracker().GetLiveObjectsForModule(module())); | |
109 | |
110 // The resource we have a scoped_refptr to should still be alive. | |
111 ASSERT_EQ(1, TrackedMockResource::tracked_objects_alive); | |
112 resource1 = NULL; | |
113 ASSERT_EQ(0, TrackedMockResource::tracked_objects_alive); | |
114 } | |
115 | |
116 } // namespace pepper | |
OLD | NEW |