Index: chrome/browser/notifications/platform_notification_service_unittest.cc |
diff --git a/chrome/browser/notifications/platform_notification_service_unittest.cc b/chrome/browser/notifications/platform_notification_service_unittest.cc |
index 840d32c9f7e2e0bdb1745317341a3f6e9c058bac..a7ed01980cb9ce9884ff49c81cfaac0125f46931 100644 |
--- a/chrome/browser/notifications/platform_notification_service_unittest.cc |
+++ b/chrome/browser/notifications/platform_notification_service_unittest.cc |
@@ -7,6 +7,7 @@ |
#include <utility> |
#include "base/bind.h" |
+#include "base/feature_list.h" |
#include "base/macros.h" |
#include "base/memory/ptr_util.h" |
#include "base/run_loop.h" |
@@ -21,6 +22,7 @@ |
#include "chrome/browser/notifications/notification_test_util.h" |
#include "chrome/browser/notifications/platform_notification_service_impl.h" |
#include "chrome/browser/notifications/stub_notification_platform_bridge.h" |
+#include "chrome/common/chrome_features.h" |
#include "chrome/test/base/testing_browser_process.h" |
#include "chrome/test/base/testing_profile.h" |
#include "chrome/test/base/testing_profile_manager.h" |
@@ -167,6 +169,16 @@ class PlatformNotificationServiceTest : public testing::Test { |
return static_cast<StubNotificationPlatformBridge*>( |
g_browser_process->notification_platform_bridge()) |
->GetNotificationAt(profile_->GetPath().BaseName().value(), 0); |
+#elif BUILDFLAG(ENABLE_NATIVE_NOTIFICATIONS) |
+ if (base::FeatureList::IsEnabled(features::kNativeNotifications)) { |
+ return static_cast<StubNotificationPlatformBridge*>( |
+ g_browser_process->notification_platform_bridge()) |
+ ->GetNotificationAt(profile_->GetPath().BaseName().value(), 0); |
+ } else { |
+ return static_cast<StubNotificationUIManager*>( |
+ g_browser_process->notification_ui_manager()) |
+ ->GetNotificationAt(0); |
+ } |
#else |
return static_cast<StubNotificationUIManager*>( |
g_browser_process->notification_ui_manager()) |