Index: ash/system/chromeos/network/network_state_notifier_unittest.cc |
diff --git a/ash/system/chromeos/network/network_state_notifier_unittest.cc b/ash/system/chromeos/network/network_state_notifier_unittest.cc |
index d8d036ad6531c94c98c715a832fba31af8db4738..4743f0a9c8d86be14bad629609aa130e8b9850b7 100644 |
--- a/ash/system/chromeos/network/network_state_notifier_unittest.cc |
+++ b/ash/system/chromeos/network/network_state_notifier_unittest.cc |
@@ -15,6 +15,7 @@ |
#include "chromeos/dbus/shill_device_client.h" |
#include "chromeos/dbus/shill_service_client.h" |
#include "third_party/cros_system_api/dbus/service_constants.h" |
+#include "ui/message_center/message_center.h" |
namespace { |
@@ -90,7 +91,10 @@ TEST_F(NetworkStateNotifierTest, ConnectionFailure) { |
ash::network_connect::ConnectToNetwork("wifi1", NULL /* owning_window */); |
RunAllPendingInMessageLoop(); |
// Failure should spawn a notification. |
- EXPECT_TRUE(GetSystemTray()->CloseNotificationBubbleForTest()); |
+ message_center::MessageCenter* message_center = |
+ message_center::MessageCenter::Get(); |
+ EXPECT_TRUE(message_center->HasNotification( |
+ network_connect::kNetworkConnectNotificationId)); |
} |
} // namespace test |