Check page URL origin to determine JavaScript alert titles.

ios/chrome/browser/ui/dialogs/javascript_dialog_egtest.mm

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import <EarlGrey/EarlGrey.h>
 #import <UIKit/UIKit.h>
 #import <XCTest/XCTest.h>
 
 #import "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 182 matching lines @@
     [[EarlGrey selectElementWithMatcher:titleLabel]
         assertWithMatcher:grey_notNil()
                     error:&error];
     return !error;
   };
   GREYAssert(testing::WaitUntilConditionOrTimeout(
                  testing::kWaitForUIElementTimeout, condition),
              @"Alert with title was not present: %@", alert_label);
 }
 
-void WaitForJavaScripDialogToBeShown() {
-  NSString* alert_label = [DialogPresenter
-      localizedTitleForJavaScriptAlertFromPage:HttpServer::MakeUrl(
-                                                   kJavaScriptTestURL)];
-  WaitForAlertToBeShown(alert_label);
+void WaitForJavaScriptDialogToBeShown() {
+  GURL javaScriptURL = HttpServer::MakeUrl(kJavaScriptTestURL);
+
+  NSString* hostname = base::SysUTF8ToNSString(javaScriptURL.host());
+  NSString* expectedTitle = l10n_util::GetNSStringF(
+      IDS_JAVASCRIPT_MESSAGEBOX_TITLE, base::SysNSStringToUTF16(hostname));
+
+  WaitForAlertToBeShown(expectedTitle);
 }
 
 // Injects JavaScript to show a dialog with |type|, verifying that it was
 // properly displayed.
 void ShowJavaScriptDialog(JavaScriptAlertType type) {
   DisplayJavaScriptAlert(type);
 
-  WaitForJavaScripDialogToBeShown();
+  WaitForJavaScriptDialogToBeShown();
 
   // Check the message of the alert.
   id<GREYMatcher> messageLabel =
       chrome_test_util::StaticTextWithAccessibilityLabel(
           GetMessageForAlertWithType(type));
   [[EarlGrey selectElementWithMatcher:messageLabel]
       assertWithMatcher:grey_notNil()];
 }
 
 // Assert no javascript alert is visible.
 void AssertJavaScriptAlertNotPresent() {
   ConditionBlock condition = ^{
     NSError* error = nil;
-    NSString* alertLabel = [DialogPresenter
-        localizedTitleForJavaScriptAlertFromPage:HttpServer::MakeUrl(
-                                                     kJavaScriptTestURL)];
+    GURL javaScriptURL = HttpServer::MakeUrl(kJavaScriptTestURL);
+    NSString* hostname = base::SysUTF8ToNSString(javaScriptURL.host());
+    NSString* expectedTitle = l10n_util::GetNSStringF(
+        IDS_JAVASCRIPT_MESSAGEBOX_TITLE, base::SysNSStringToUTF16(hostname));
+
     id<GREYMatcher> titleLabel =
-        chrome_test_util::StaticTextWithAccessibilityLabel(alertLabel);
+        chrome_test_util::StaticTextWithAccessibilityLabel(expectedTitle);
     [[EarlGrey selectElementWithMatcher:titleLabel] assertWithMatcher:grey_nil()
                                                                 error:&error];
     return !error;
   };
 
   GREYAssert(testing::WaitUntilConditionOrTimeout(
                  testing::kWaitForJSCompletionTimeout, condition),
              @"Javascript alert title was still present");
 }
 
@@ skipping 192 matching lines @@
   [ChromeEarlGrey waitForWebViewContainingText:kPromptResultBodyCancelled];
 }
 
 // Tests that JavaScript alerts that are shown in a loop can be suppressed.
 - (void)testShowJavaScriptAlertLoop {
   // Load the blank test page and show alerts in a loop.
   [self loadBlankTestPage];
   web::WebState* webState = chrome_test_util::GetCurrentWebState();
   NSString* script = GetJavaScriptAlertLoopScript();
   webState->ExecuteJavaScript(base::SysNSStringToUTF16(script));
-  WaitForJavaScripDialogToBeShown();
+  WaitForJavaScriptDialogToBeShown();
 
   [[EarlGrey selectElementWithMatcher:OKButton()] performAction:grey_tap()];
-  WaitForJavaScripDialogToBeShown();
+  WaitForJavaScriptDialogToBeShown();
 
   // Tap the suppress dialogs button.
   TapSuppressDialogsButton();
 
   // Wait for confirmation action sheet to be shown.
   NSString* alertLabel =
       l10n_util::GetNSString(IDS_JAVASCRIPT_MESSAGEBOX_SUPPRESS_OPTION);
   WaitForAlertToBeShown(alertLabel);
 
   // Tap the suppress dialogs confirmation button.
@@ skipping 28 matching lines @@
   DisplayJavaScriptAlert(JavaScriptAlertType::ALERT);
 
   // Make sure the alert is not present.
   AssertJavaScriptAlertNotPresent();
 
   // Close the settings.
   [[EarlGrey selectElementWithMatcher:NavigationBarDoneButton()]
       performAction:grey_tap()];
 
   // Make sure the alert is present.
-  WaitForJavaScripDialogToBeShown();
+  WaitForJavaScriptDialogToBeShown();
 
   [[EarlGrey selectElementWithMatcher:OKButton()] performAction:grey_tap()];
 
   // Wait for the html body to be reset to the correct value.
   [ChromeEarlGrey waitForWebViewContainingText:kAlertResultBody];
 }
 
 // Tests that an alert is presented after displaying the share menu.
 - (void)testShowJavaScriptAfterShareMenu {
 // TODO(crbug.com/663026): Reenable the test for devices.
@@ skipping 58 matching lines @@
   // continues to animate until the dialog is closed.  Disabling EarlGrey
   // synchronization code for iPad allows the test to detect and dismiss the
   // dialog while this animation is occurring.
   if (IsIPadIdiom()) {
     [[GREYConfiguration sharedInstance]
             setValue:@(NO)
         forConfigKey:kGREYConfigKeySynchronizationEnabled];
   }
 
   // Wait for the alert to be shown.
-  NSString* alertLabel = [DialogPresenter
-      localizedTitleForJavaScriptAlertFromPage:HttpServer::MakeUrl(
-                                                   kJavaScriptTestURL)];
-  WaitForAlertToBeShown(alertLabel);
+  GURL javaScriptURL = HttpServer::MakeUrl(kJavaScriptTestURL);
+  NSString* hostname = base::SysUTF8ToNSString(javaScriptURL.host());
+  NSString* expectedTitle = l10n_util::GetNSStringF(
+      IDS_JAVASCRIPT_MESSAGEBOX_TITLE, base::SysNSStringToUTF16(hostname));
+
+  WaitForAlertToBeShown(expectedTitle);
 
   // Verify that the omnibox shows the correct URL when the dialog is visible.
   GURL onloadURL = HttpServer::MakeUrl(kOnLoadAlertURL);
   std::string title = base::UTF16ToUTF8(web::GetDisplayTitleForUrl(onloadURL));
   [[EarlGrey selectElementWithMatcher:chrome_test_util::OmniboxText(title)]
       assertWithMatcher:grey_notNil()];
 
   [[EarlGrey selectElementWithMatcher:OKButton()] performAction:grey_tap()];
 
   // Reenable synchronization on iPads now that the dialog has been dismissed.
   if (IsIPadIdiom()) {
     [[GREYConfiguration sharedInstance]
             setValue:@(YES)
         forConfigKey:kGREYConfigKeySynchronizationEnabled];
   }
 }
 
 @end