CrOS: Do not allow notifications to be added during shutdown.

chrome/browser/notifications/message_center_notifications_unittest.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 <memory>
 
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_timeouts.h"
 #include "base/values.h"
 #include "build/build_config.h"
@@ skipping 33 matching lines @@
  protected:
   void SetUp() override {
     BrowserWithTestWindowTest::SetUp();
 #if !defined(OS_CHROMEOS)
     // BrowserWithTestWindowTest owns an AshTestHelper on OS_CHROMEOS, which
     // in turn initializes the message center.  On other platforms, we need to
     // initialize it here.
     MessageCenter::Initialize();
 #endif
 
-
     TestingBrowserProcess* browser_process = TestingBrowserProcess::GetGlobal();
     profile_manager_.reset(new TestingProfileManager(browser_process));
     ASSERT_TRUE(profile_manager_->SetUp());
 
     message_center_ = MessageCenter::Get();
     delegate_ = new FakeMessageCenterTrayDelegate(message_center_);
     notification_manager()->SetMessageCenterTrayDelegateForTest(delegate_);
   }
 
   void TearDown() override {
@@ skipping 32 matching lines @@
   std::unique_ptr<TestingProfileManager> profile_manager_;
   MessageCenter* message_center_;
   FakeMessageCenterTrayDelegate* delegate_;
 };
 
 TEST_F(MessageCenterNotificationManagerTest, SetupNotificationManager) {
   TestingProfile profile;
   notification_manager()->Add(GetANotification("test"), &profile);
 }
 
+TEST_F(MessageCenterNotificationManagerTest, AddNotificationOnShutdown) {
+  TestingProfile profile;
+  EXPECT_TRUE(message_center()->NotificationCount() == 0);
+  notification_manager()->Add(GetANotification("test"), &profile);
+  EXPECT_TRUE(message_center()->NotificationCount() == 1);
+
+  // Verify the number of notifications does not increase when trying to add a
+  // notifcation on shutdown.
+  notification_manager()->StartShutdown();
+  EXPECT_TRUE(message_center()->NotificationCount() == 0);
+  notification_manager()->Add(GetANotification("test2"), &profile);
+  EXPECT_TRUE(message_center()->NotificationCount() == 0);
+}
+
 TEST_F(MessageCenterNotificationManagerTest, UpdateNotification) {
   TestingProfile profile;
   EXPECT_TRUE(message_center()->NotificationCount() == 0);
   notification_manager()->Add(GetANotification("test"), &profile);
   EXPECT_TRUE(message_center()->NotificationCount() == 1);
   ASSERT_TRUE(
       notification_manager()->Update(GetANotification("test"), &profile));
   EXPECT_TRUE(message_center()->NotificationCount() == 1);
 }
 
@@ skipping 14 matching lines @@
   EXPECT_EQ(0u, message_center()->NotificationCount());
   notification_manager()->Add(GetANotification("test"), &profile);
   EXPECT_EQ(1u, message_center()->NotificationCount());
   notification_manager()->Update(GetANotification("test"), &profile);
   EXPECT_EQ(1u, message_center()->NotificationCount());
   chrome::MultiUserWindowManager::DeleteInstance();
 }
 #endif
 
 }  // namespace message_center