ppapi/proxy/plugin_resource_tracker_unittest.cc

   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
   5 #include "base/memory/scoped_ptr.h"
   6 #include "base/process/process.h"
   7 #include "ppapi/proxy/mock_resource.h"
   8 #include "ppapi/proxy/plugin_dispatcher.h"
   9 #include "ppapi/proxy/plugin_resource_tracker.h"
  10 #include "ppapi/proxy/ppapi_messages.h"
  11 #include "ppapi/proxy/ppapi_proxy_test.h"
  12 #include "ppapi/shared_impl/proxy_lock.h"
  13
  14 namespace ppapi {
  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() {}
  41 };
  42
  43 TEST_F(PluginResourceTrackerTest, PluginResourceForHostResource) {
  44   ProxyAutoLock lock;
  45
  46   PP_Resource host_resource = 0x5678;
  47
  48   HostResource serialized;
  49   serialized.SetHostResource(pp_instance(), host_resource);
  50
  51   // When we haven't added an object, the return value should be 0.
  52   EXPECT_EQ(0, resource_tracker().PluginResourceForHostResource(serialized));
  53
  54   EXPECT_EQ(0, TrackedMockResource::tracked_alive_count);
  55   TrackedMockResource* object = new TrackedMockResource(serialized);
  56   EXPECT_EQ(1, TrackedMockResource::tracked_alive_count);
  57   PP_Resource plugin_resource = object->GetReference();
  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,
  62             resource_tracker().PluginResourceForHostResource(serialized));
  63
  64   // Releasing the resource should have freed it.
  65   resource_tracker().ReleaseResource(plugin_resource);
  66   EXPECT_EQ(0, TrackedMockResource::tracked_alive_count);
  67   EXPECT_EQ(0, resource_tracker().PluginResourceForHostResource(serialized));
  68 }
  69
  70 }  // namespace proxy
  71 }  // namespace ppapi