VR: Update UiSceneManager with screen capturing flag

chrome/browser/android/vr_shell/ui_scene_manager_unittest.cc

 // Copyright 2017 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 "chrome/browser/android/vr_shell/ui_scene_manager.h"
 
 #include "base/macros.h"
 #include "base/test/scoped_task_environment.h"
 #include "chrome/browser/android/vr_shell/ui_elements/ui_element.h"
 #include "chrome/browser/android/vr_shell/ui_elements/ui_element_debug_id.h"
@@ skipping 231 matching lines @@
     bool should_be_visible = visible_in_browsing.find(element->debug_id()) !=
                              visible_in_browsing.end();
     EXPECT_EQ(should_be_visible, element->visible());
   }
   {
     SCOPED_TRACE("Exited Fullsceen");
     EXPECT_EQ(initial_background, scene_->GetBackgroundColor());
   }
 }
 
+TEST_F(UiSceneManagerTest, UiUpdatesForWebVR) {
+  MakeManager(kNotInCct, kInWebVr);
+
+  manager_->SetWebVrSecureOrigin(true);
+  manager_->SetAudioCapturingIndicator(true);
+  manager_->SetVideoCapturingIndicator(true);
+  manager_->SetScreenCapturingIndicator(true);
+
+  // All elements should be hidden.
+  for (const auto& element : scene_->GetUiElements()) {
+    SCOPED_TRACE(element->debug_id());
+    EXPECT_FALSE(element->visible());
+  }
+}
+
+TEST_F(UiSceneManagerTest, UiUpdateTransitionToWebVR) {
+  MakeManager(kNotInCct, kNotInWebVr);
+  manager_->SetAudioCapturingIndicator(true);
+  manager_->SetVideoCapturingIndicator(true);
+  manager_->SetScreenCapturingIndicator(true);
+
+  // Transition to WebVR mode
+  manager_->SetWebVrMode(true);
+  manager_->SetWebVrSecureOrigin(true);
+
+  // All elements should be hidden.
+  for (const auto& element : scene_->GetUiElements()) {
+    SCOPED_TRACE(element->debug_id());
+    EXPECT_FALSE(element->visible());
+  }
+}
+
+TEST_F(UiSceneManagerTest, CaptureIndicatorsVisibility) {
+  MakeManager(kNotInCct, kNotInWebVr);
+  EXPECT_FALSE(IsVisible(kAudioCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kVideoCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kScreenCaptureIndicator));
+
+  manager_->SetAudioCapturingIndicator(true);
+  manager_->SetVideoCapturingIndicator(true);
+  manager_->SetScreenCapturingIndicator(true);
+
+  EXPECT_TRUE(IsVisible(kAudioCaptureIndicator));
+  EXPECT_TRUE(IsVisible(kVideoCaptureIndicator));
+  EXPECT_TRUE(IsVisible(kScreenCaptureIndicator));
+
+  manager_->SetWebVrMode(true);
+  EXPECT_FALSE(IsVisible(kAudioCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kVideoCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kScreenCaptureIndicator));
+
+  manager_->SetWebVrMode(false);
+  EXPECT_TRUE(IsVisible(kAudioCaptureIndicator));
+  EXPECT_TRUE(IsVisible(kVideoCaptureIndicator));
+  EXPECT_TRUE(IsVisible(kScreenCaptureIndicator));
+
+  manager_->SetAudioCapturingIndicator(false);
+  manager_->SetVideoCapturingIndicator(false);
+  manager_->SetScreenCapturingIndicator(false);
+
+  EXPECT_FALSE(IsVisible(kAudioCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kVideoCaptureIndicator));
+  EXPECT_FALSE(IsVisible(kScreenCaptureIndicator));
+}
 }  // namespace vr_shell