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 eadbb842b9a0596211c253d2f72f9dd35e6b71f6..f7298c7a4e65ec403df6ef53cc911edfb963866f 100644 |
--- a/ui/arc/notification/arc_notification_manager_unittest.cc |
+++ b/ui/arc/notification/arc_notification_manager_unittest.cc |
@@ -2,8 +2,13 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
+#include <memory> |
+#include <string> |
+#include <vector> |
+ |
#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
+#include "components/arc/instance_holder.h" |
#include "components/arc/test/fake_arc_bridge_instance.h" |
#include "components/arc/test/fake_arc_bridge_service.h" |
#include "components/arc/test/fake_notifications_instance.h" |
@@ -52,17 +57,18 @@ class MockMessageCenter : public message_center::FakeMessageCenter { |
DISALLOW_COPY_AND_ASSIGN(MockMessageCenter); |
}; |
-class ArcBridgeServiceObserver : public ArcBridgeService::Observer { |
+class NotificationsObserver |
+ : public InstanceHolder<mojom::NotificationsInstance>::Observer { |
public: |
- ArcBridgeServiceObserver() = default; |
- void OnNotificationsInstanceReady() override { ready_ = true; } |
+ NotificationsObserver() = default; |
+ void OnInstanceReady() override { ready_ = true; } |
bool IsReady() { return ready_; } |
private: |
bool ready_ = false; |
- DISALLOW_COPY_AND_ASSIGN(ArcBridgeServiceObserver); |
+ DISALLOW_COPY_AND_ASSIGN(NotificationsObserver); |
}; |
} // anonymous namespace |
@@ -117,15 +123,15 @@ class ArcNotificationManagerTest : public testing::Test { |
arc_notification_manager_.reset(new ArcNotificationManager( |
service(), EmptyAccountId(), message_center_.get())); |
- ArcBridgeServiceObserver observer; |
- service_->AddObserver(&observer); |
+ NotificationsObserver observer; |
+ service_->notifications()->AddObserver(&observer); |
service_->OnNotificationsInstanceReady( |
std::move(arc_notifications_instance)); |
while (!observer.IsReady()) |
loop_.RunUntilIdle(); |
- service_->RemoveObserver(&observer); |
+ service_->notifications()->RemoveObserver(&observer); |
} |
void TearDown() override { |
@@ -177,7 +183,7 @@ TEST_F(ArcNotificationManagerTest, NotificationRemovedByConnectionClose) { |
CreateNotificationWithKey("notification3"); |
EXPECT_EQ(3u, message_center()->GetVisibleNotifications().size()); |
- arc_notification_manager()->OnNotificationsInstanceClosed(); |
+ arc_notification_manager()->OnInstanceClosed(); |
EXPECT_EQ(0u, message_center()->GetVisibleNotifications().size()); |
} |