[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 1 | // Copyright (c) 2011 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 | |
dcheng | ced9224 | 2016-04-07 00:00:12 | [diff] [blame] | 5 | #include "ppapi/proxy/plugin_resource_tracker.h" |
| 6 | |
[email protected] | 1575659 | 2013-07-25 14:17:53 | [diff] [blame] | 7 | #include "base/process/process.h" |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 8 | #include "ppapi/proxy/mock_resource.h" |
| 9 | #include "ppapi/proxy/plugin_dispatcher.h" |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 10 | #include "ppapi/proxy/ppapi_messages.h" |
| 11 | #include "ppapi/proxy/ppapi_proxy_test.h" |
[email protected] | b73c659 | 2013-03-30 17:08:13 | [diff] [blame] | 12 | #include "ppapi/shared_impl/proxy_lock.h" |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 13 | |
[email protected] | 4d2efd2 | 2011-08-18 21:58:02 | [diff] [blame] | 14 | namespace ppapi { |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 15 | namespace proxy { |
| 16 | |
| 17 | namespace { |
| 18 | |
| 19 | // Object so we know when a resource has been released. |
| 20 | class TrackedMockResource : public MockResource { |
| 21 | public: |
| 22 | TrackedMockResource(const HostResource& serialized) |
| 23 | : MockResource(serialized) { |
| 24 | tracked_alive_count++; |
| 25 | } |
| 26 | ~TrackedMockResource() { |
| 27 | tracked_alive_count--; |
| 28 | } |
| 29 | |
| 30 | static int tracked_alive_count; |
| 31 | }; |
| 32 | |
| 33 | int TrackedMockResource::tracked_alive_count = 0; |
| 34 | |
| 35 | } // namespace |
| 36 | |
| 37 | class PluginResourceTrackerTest : public PluginProxyTest { |
| 38 | public: |
| 39 | PluginResourceTrackerTest() {} |
| 40 | ~PluginResourceTrackerTest() {} |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 41 | }; |
| 42 | |
| 43 | TEST_F(PluginResourceTrackerTest, PluginResourceForHostResource) { |
[email protected] | b73c659 | 2013-03-30 17:08:13 | [diff] [blame] | 44 | ProxyAutoLock lock; |
| 45 | |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 46 | PP_Resource host_resource = 0x5678; |
| 47 | |
| 48 | HostResource serialized; |
[email protected] | 465faa2 | 2011-02-08 16:31:46 | [diff] [blame] | 49 | serialized.SetHostResource(pp_instance(), host_resource); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 50 | |
| 51 | // When we haven't added an object, the return value should be 0. |
[email protected] | 7f8b26b | 2011-08-18 15:41:01 | [diff] [blame] | 52 | EXPECT_EQ(0, resource_tracker().PluginResourceForHostResource(serialized)); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 53 | |
| 54 | EXPECT_EQ(0, TrackedMockResource::tracked_alive_count); |
| 55 | TrackedMockResource* object = new TrackedMockResource(serialized); |
| 56 | EXPECT_EQ(1, TrackedMockResource::tracked_alive_count); |
[email protected] | 7f8b26b | 2011-08-18 15:41:01 | [diff] [blame] | 57 | PP_Resource plugin_resource = object->GetReference(); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 58 | |
| 59 | // Now that the object has been added, the return value should be the plugin |
| 60 | // resource ID we already got. |
| 61 | EXPECT_EQ(plugin_resource, |
[email protected] | 7f8b26b | 2011-08-18 15:41:01 | [diff] [blame] | 62 | resource_tracker().PluginResourceForHostResource(serialized)); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 63 | |
| 64 | // Releasing the resource should have freed it. |
[email protected] | 7f8b26b | 2011-08-18 15:41:01 | [diff] [blame] | 65 | resource_tracker().ReleaseResource(plugin_resource); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 66 | EXPECT_EQ(0, TrackedMockResource::tracked_alive_count); |
[email protected] | 7f8b26b | 2011-08-18 15:41:01 | [diff] [blame] | 67 | EXPECT_EQ(0, resource_tracker().PluginResourceForHostResource(serialized)); |
[email protected] | f24448db | 2011-01-27 20:40:39 | [diff] [blame] | 68 | } |
| 69 | |
| 70 | } // namespace proxy |
[email protected] | 4d2efd2 | 2011-08-18 21:58:02 | [diff] [blame] | 71 | } // namespace ppapi |