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 {
+
+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:
+  // |initial_context_support_| is just stored/used for sanity-checking Release.

[danakj, 2016/08/19 18:29:55]

Comment not quite true now

[ericrk, 2016/08/19 18:49:57]

Done.

+  const gpu::ContextSupport* initial_context_support_;
+};
+
+}  // namespace
 
 TestContextSupport::TestContextSupport()
-    : out_of_order_callbacks_(false), weak_ptr_factory_(this) {}
+    : out_of_order_callbacks_(false),
+      visible_clients_(0),

[danakj, 2016/08/19 18:29:55]

can you do this in the header? (mayeb move out_of_order_callbacks there too?)

[ericrk, 2016/08/19 18:49:57]

Done.

+      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()));
 }
@@ skipping 55 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() {
+  ++visible_clients_;
+  return base::MakeUnique<ScopedVisibilityImpl>(this);
+}
+
+void TestContextSupport::ClientBecameNotVisible(
+    std::unique_ptr<ScopedVisibility> visibility) {
+  DCHECK_GT(visible_clients_, 0u);
+  --visible_clients_;
+  static_cast<ScopedVisibilityImpl*>(visibility.get())->Release(this);
 }
 
 bool TestContextSupport::AnyClientsVisible() const {
-  return !visible_clients_.empty();
+  return visible_clients_ > 0;
 }
 
 }  // namespace cc