Make it possible to write simple unit tests for the pepper implementation. Th...

webkit/glue/plugins/pepper_resource_tracker.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "webkit/glue/plugins/pepper_resource_tracker.h"
 
 #include <limits>
 #include <set>
 
 #include "base/logging.h"
 #include "base/rand_util.h"
 #include "ppapi/c/pp_resource.h"
 #include "webkit/glue/plugins/pepper_resource.h"
 
 namespace pepper {
 
 scoped_refptr<Resource> ResourceTracker::GetResource(PP_Resource res) const {
   ResourceMap::const_iterator result = live_resources_.find(res);
   if (result == live_resources_.end()) {
     return scoped_refptr<Resource>();
   }
   return result->second.first;
 }
 
+// static
+ResourceTracker* ResourceTracker::singleton_override_ = NULL;
+
 ResourceTracker::ResourceTracker()
     : last_id_(0) {
 }
 
 ResourceTracker::~ResourceTracker() {
 }
 
 // static
 ResourceTracker* ResourceTracker::Get() {
+  if (singleton_override_)
+    return singleton_override_;
   return Singleton<ResourceTracker>::get();
 }
 
 PP_Resource ResourceTracker::AddResource(Resource* resource) {
   // If the plugin manages to create 4B resources...
   if (last_id_ == std::numeric_limits<PP_Resource>::max()) {
     return 0;
   }
   // Add the resource with plugin use-count 1.
   ++last_id_;
@@ skipping 22 matching lines @@
       live_resources_.erase(i);
     }
     return true;
   } else {
     return false;
   }
 }
 
 void ResourceTracker::ForceDeletePluginResourceRefs(PP_Resource res) {
   ResourceMap::iterator i = live_resources_.find(res);
-  if (i != live_resources_.end())
+  if (i == live_resources_.end())
     return;  // Nothing to do.
 
   i->second.second = 0;
   i->second.first->StoppedTracking();
   live_resources_.erase(i);
 }
 
 uint32 ResourceTracker::GetLiveObjectsForModule(PluginModule* module) const {
   // Since this is for testing only, we'll just go through all of them and
   // count.
@@ skipping 75 matching lines @@
   module_map_.erase(found);
 }
 
 PluginModule* ResourceTracker::GetModule(PP_Module module) {
   ModuleMap::iterator found = module_map_.find(module);
   if (found == module_map_.end())
     return NULL;
   return found->second;
 }
 
+// static
+void ResourceTracker::SetSingletonOverride(ResourceTracker* tracker) {
+  DCHECK(!singleton_override_);
+  singleton_override_ = tracker;
+}
+
+// static
+void ResourceTracker::ClearSingletonOverride() {
+  DCHECK(singleton_override_);
+  singleton_override_ = NULL;
+}
+
 }  // namespace pepper