mus: Implement Window Server Capture

components/mus/public/cpp/tests/window_tree_client_impl_unittest.cc

 // Copyright 2015 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 "components/mus/public/cpp/lib/window_tree_client_impl.h"
 
 #include <stdint.h>
 
 #include "base/logging.h"
 #include "base/macros.h"
@@ skipping 645 matching lines @@
   ASSERT_TRUE(root2);
   root1->AddTransientWindow(root2);
 
   WindowTracker tracker;
   tracker.Add(root1);
   tracker.Add(root2);
   setup.reset();
   EXPECT_TRUE(tracker.windows().empty());
 }
 
+// Tests both SetCapture and ReleaseCapture, to ensure that Window is properly
+// updated on failures.
+TEST_F(WindowTreeClientImplTest, ExplicitCapture) {
+  WindowTreeSetup setup;
+  Window* root = setup.GetFirstRoot();
+  ASSERT_TRUE(root);
+
+  root->SetCapture();
+  EXPECT_TRUE(root->has_capture());
+  uint32_t change_id1;
+  ASSERT_TRUE(setup.window_tree()->GetAndClearChangeId(&change_id1));
+  setup.window_tree_client()->OnChangeCompleted(change_id1, false);
+  EXPECT_FALSE(root->has_capture());
+
+  root->SetCapture();
+  EXPECT_TRUE(root->has_capture());
+  uint32_t change_id2;
+  ASSERT_TRUE(setup.window_tree()->GetAndClearChangeId(&change_id2));
+  setup.window_tree_client()->OnChangeCompleted(change_id2, true);
+  EXPECT_TRUE(root->has_capture());
+
+  root->ReleaseCapture();
+  EXPECT_FALSE(root->has_capture());
+  uint32_t change_id3;
+  ASSERT_TRUE(setup.window_tree()->GetAndClearChangeId(&change_id3));
+  setup.window_tree_client()->OnChangeCompleted(change_id3, false);
+  EXPECT_TRUE(root->has_capture());
+
+  root->ReleaseCapture();
+  uint32_t change_id4;
+  ASSERT_TRUE(setup.window_tree()->GetAndClearChangeId(&change_id4));
+  setup.window_tree_client()->OnChangeCompleted(change_id4, true);
+  EXPECT_FALSE(root->has_capture());
+}
+
 }  // namespace mus