Hide notification tray when the user flow doesn't allow

chrome/browser/chromeos/login/supervised/supervised_user_creation_browsertest.cc

 // Copyright 2014 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 <string>
 
+#include "ash/common/system/status_area_widget.h"
+#include "ash/common/system/web_notification/web_notification_tray.h"
+#include "ash/test/status_area_widget_test_helper.h"
 #include "base/compiler_specific.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/login/login_manager_test.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/supervised/supervised_user_authentication.h"
 #include "chrome/browser/chromeos/login/supervised/supervised_user_test_base.h"
@@ skipping 17 matching lines @@
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 
 using testing::_;
 using chromeos::SupervisedUserTestBase;
 using chromeos::kTestSupervisedUserDisplayName;
 using chromeos::kTestManager;
 
 namespace chromeos {
 
+namespace {
+
+bool GetWebNotificationTrayVisibility() {
+  return ash::StatusAreaWidgetTestHelper::GetStatusAreaWidget()
+      ->web_notification_tray()
+      ->visible();
+}
+
+}  // anonymous namespace
+
 class SupervisedUserCreationTest : public SupervisedUserTestBase {
  public:
   SupervisedUserCreationTest() : SupervisedUserTestBase() {}
 
  private:
   DISALLOW_COPY_AND_ASSIGN(SupervisedUserCreationTest);
 };
 
 class SupervisedUserTransactionCleanupTest : public SupervisedUserTestBase {
  public:
@@ skipping 123 matching lines @@
 
   std::string user_id = registration_utility_stub_->supervised_user_id();
 
   // Make sure user is already in list.
   ASSERT_EQ(3UL, user_manager::UserManager::Get()->GetUsers().size());
 
   // We wait for token now. Press cancel button at this point.
   JSEvalOrExitBrowser("$('supervised-user-creation').cancel()");
 }
 
+IN_PROC_BROWSER_TEST_F(SupervisedUserCreationTest,
+                       PRE_PRE_PRE_CheckNoNotificationTray) {
+  PrepareUsers();
+}
+
+IN_PROC_BROWSER_TEST_F(SupervisedUserCreationTest,
+                       PRE_PRE_CheckNoNotificationTray) {
+  // Before sign-in, the tray should not be visible.
+  EXPECT_FALSE(GetWebNotificationTrayVisibility());
+
+  StartFlowLoginAsManager();
+
+  // On supervised user creation flow, the tray should not be visible.
+  EXPECT_FALSE(GetWebNotificationTrayVisibility());
+
+  FillNewUserData(kTestSupervisedUserDisplayName);
+  StartUserCreation("supervised-user-creation-next-button",
+                    kTestSupervisedUserDisplayName);
+}
+
+IN_PROC_BROWSER_TEST_F(SupervisedUserCreationTest,
+                       PRE_CheckNoNotificationTray) {
+  SigninAsSupervisedUser(true, 0, kTestSupervisedUserDisplayName);
+
+  // After sign-in, the tray should be visible.
+  EXPECT_TRUE(GetWebNotificationTrayVisibility());
+}
+
+IN_PROC_BROWSER_TEST_F(SupervisedUserCreationTest, CheckNoNotificationTray) {
+  RemoveSupervisedUser(3, 0, kTestSupervisedUserDisplayName);
+}
+
 IN_PROC_BROWSER_TEST_(
     SupervisedUserTransactionCleanupTest,
     CreateAndCancelSupervisedUser,
     SupervisedUserTransactionCleanupTest2,
     testing::internal::GetTypeId<SupervisedUserTransactionCleanupTest>()) {
   // Make sure there is no supervised user in list.
   ASSERT_EQ(2UL, user_manager::UserManager::Get()->GetUsers().size());
 }
 
 }  // namespace chromeos