Index: ui/arc/notification/arc_notification_manager_unittest.cc |
diff --git a/ui/arc/notification/arc_notification_manager_unittest.cc b/ui/arc/notification/arc_notification_manager_unittest.cc |
index 866a6b54372de3920d294eb75728b6237c537e12..f7298c7a4e65ec403df6ef53cc911edfb963866f 100644 |
--- a/ui/arc/notification/arc_notification_manager_unittest.cc |
+++ b/ui/arc/notification/arc_notification_manager_unittest.cc |
@@ -114,7 +114,9 @@ |
std::unique_ptr<MockMessageCenter> message_center_; |
void SetUp() override { |
- arc_notifications_instance_.reset(new FakeNotificationsInstance()); |
+ mojom::NotificationsInstancePtr arc_notifications_instance; |
+ arc_notifications_instance_.reset( |
+ new FakeNotificationsInstance(GetProxy(&arc_notifications_instance))); |
service_.reset(new FakeArcBridgeService()); |
message_center_.reset(new MockMessageCenter()); |
@@ -123,7 +125,8 @@ |
NotificationsObserver observer; |
service_->notifications()->AddObserver(&observer); |
- service_->notifications()->SetInstance(arc_notifications_instance_.get()); |
+ service_->OnNotificationsInstanceReady( |
+ std::move(arc_notifications_instance)); |
while (!observer.IsReady()) |
loop_.RunUntilIdle(); |
@@ -164,6 +167,8 @@ |
// |notification| gets stale here. |
} |
+ arc_notifications_instance()->WaitForIncomingMethodCall(); |
+ |
ASSERT_EQ(1u, arc_notifications_instance()->events().size()); |
EXPECT_EQ(key, arc_notifications_instance()->events().at(0).first); |
EXPECT_EQ(mojom::ArcNotificationEvent::CLOSED, |