Update Context Client Visibility to use Scoped Pattern

cc/test/test_context_support.cc

 // Copyright 2013 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 "cc/test/test_context_support.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include "base/bind.h"
 #include "base/location.h"
+#include "base/memory/ptr_util.h"
 #include "base/single_thread_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 
 namespace cc {
+namespace {
 
-TestContextSupport::TestContextSupport()
-    : out_of_order_callbacks_(false), weak_ptr_factory_(this) {}
+// Class that DCHECKs if it is destructed without first having Release called.
+class ScopedVisibilityImpl : public gpu::ContextSupport::ScopedVisibility {
+ public:
+  explicit ScopedVisibilityImpl(gpu::ContextSupport* context_support)
+      : initial_context_support_(context_support) {}
+  ~ScopedVisibilityImpl() { DCHECK(!initial_context_support_); }
+
+  void Release(gpu::ContextSupport* context_support) {
+    DCHECK_EQ(initial_context_support_, context_support);
+    initial_context_support_ = nullptr;
+  }
+
+ private:
+  const gpu::ContextSupport* initial_context_support_;
+};
+
+}  // namespace
+
+TestContextSupport::TestContextSupport() : weak_ptr_factory_(this) {}
 
 TestContextSupport::~TestContextSupport() {}
 
 void TestContextSupport::SignalSyncToken(const gpu::SyncToken& sync_token,
                                          const base::Closure& callback) {
   sync_point_callbacks_.push_back(callback);
   base::ThreadTaskRunnerHandle::Get()->PostTask(
       FROM_HERE, base::Bind(&TestContextSupport::CallAllSyncPointCallbacks,
                             weak_ptr_factory_.GetWeakPtr()));
 }
 
 void TestContextSupport::SignalQuery(uint32_t query,
                                      const base::Closure& callback) {
   sync_point_callbacks_.push_back(callback);
   base::ThreadTaskRunnerHandle::Get()->PostTask(
       FROM_HERE, base::Bind(&TestContextSupport::CallAllSyncPointCallbacks,
                             weak_ptr_factory_.GetWeakPtr()));
 }
 
-void TestContextSupport::SetAggressivelyFreeResources(
-    bool aggressively_free_resources) {
-}
+void TestContextSupport::TrimResources() {}
 
 void TestContextSupport::CallAllSyncPointCallbacks() {
   size_t size = sync_point_callbacks_.size();
   if (out_of_order_callbacks_) {
     for (size_t i = size; i > 0; --i) {
       base::ThreadTaskRunnerHandle::Get()->PostTask(
           FROM_HERE, sync_point_callbacks_[i - 1]);
     }
   } else {
     for (size_t i = 0; i < size; ++i) {
@@ skipping 33 matching lines @@
 }
 
 uint64_t TestContextSupport::ShareGroupTracingGUID() const {
   NOTIMPLEMENTED();
   return 0;
 }
 
 void TestContextSupport::SetErrorMessageCallback(
     const base::Callback<void(const char*, int32_t)>& callback) {}
 
-void TestContextSupport::SetClientVisible(int client_id, bool is_visible) {
-  if (is_visible) {
-    visible_clients_.insert(client_id);
-  } else {
-    auto found = visible_clients_.find(client_id);
-    if (found != visible_clients_.end())
-      visible_clients_.erase(found);
-  }
+std::unique_ptr<gpu::ContextSupport::ScopedVisibility>
+TestContextSupport::ClientBecameVisible() {
+  ++num_visible_clients_;
+  return base::MakeUnique<ScopedVisibilityImpl>(this);
+}
+
+void TestContextSupport::ClientBecameNotVisible(
+    std::unique_ptr<ScopedVisibility> visibility) {
+  DCHECK(visibility);
+  DCHECK_GT(num_visible_clients_, 0u);
+  --num_visible_clients_;
+  static_cast<ScopedVisibilityImpl*>(visibility.get())->Release(this);
 }
 
 bool TestContextSupport::AnyClientsVisible() const {
-  return !visible_clients_.empty();
+  return num_visible_clients_ > 0;
 }
 
 }  // namespace cc