Index: chrome/browser/notifications/notification_interactive_uitest_support.cc |
diff --git a/chrome/browser/notifications/notification_interactive_uitest_support.cc b/chrome/browser/notifications/notification_interactive_uitest_support.cc |
index 21467c6b3d2550fa858fd4977901bd93fdad9beb..ef3047e5e4537a525b6d62d0e59856494e1ee5f7 100644 |
--- a/chrome/browser/notifications/notification_interactive_uitest_support.cc |
+++ b/chrome/browser/notifications/notification_interactive_uitest_support.cc |
@@ -240,6 +240,14 @@ bool NotificationsTest::RequestPermissionAndWait(Browser* browser) { |
return observer.request_shown(); |
} |
+std::string NotificationsTest::QueryPermissionStatus(Browser* browser) { |
+ std::string result; |
+ content::WebContents* web_contents = GetActiveWebContents(browser); |
+ EXPECT_TRUE(content::ExecuteScriptAndExtractString( |
+ web_contents, "queryPermissionStatus();", &result)); |
+ return result; |
+} |
+ |
bool NotificationsTest::CancelNotification(const char* notification_id, |
Browser* browser) { |
std::string script = |
@@ -293,6 +301,12 @@ content::WebContents* NotificationsTest::GetActiveWebContents( |
return browser->tab_strip_model()->GetActiveWebContents(); |
} |
+void NotificationsTest::EnablePermissionsEmbargo() { |
+ feature_list_.InitWithFeatures({features::kBlockPromptsIfDismissedOften, |
+ features::kBlockPromptsIfIgnoredOften}, |
+ {}); |
+} |
+ |
void NotificationsTest::EnableFullscreenNotifications() { |
feature_list_.InitWithFeatures( |
{features::kPreferHtmlOverPlugins, |