Upstream Chrome on iOS source code [9/11].

ios/chrome/browser/ui/toolbar/toolbar_egtest.mm

Index: ios/chrome/browser/ui/toolbar/toolbar_egtest.mm
diff --git a/ios/chrome/browser/ui/toolbar/toolbar_egtest.mm b/ios/chrome/browser/ui/toolbar/toolbar_egtest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..6a426fdf867b830f632492358679171fccb307bc
--- /dev/null
+++ b/ios/chrome/browser/ui/toolbar/toolbar_egtest.mm
@@ -0,0 +1,451 @@
+// 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 <XCTest/XCTest.h>
+
+#include "base/ios/ios_util.h"
+#include "components/strings/grit/components_strings.h"
+#import "ios/chrome/browser/ui/commands/generic_chrome_command.h"
+#include "ios/chrome/browser/ui/commands/ios_command_ids.h"
+#import "ios/chrome/browser/ui/ntp/new_tab_page_controller.h"
+#import "ios/chrome/browser/ui/omnibox/omnibox_popup_material_row.h"
+#import "ios/chrome/browser/ui/toolbar/toolbar_controller.h"
+#import "ios/chrome/browser/ui/tools_menu/tools_menu_view_controller.h"
+#include "ios/chrome/browser/ui/ui_util.h"
+#include "ios/chrome/grit/ios_strings.h"
+#import "ios/chrome/test/app/chrome_test_util.h"
+#import "ios/chrome/test/app/tab_test_util.h"
+#import "ios/chrome/test/earl_grey/chrome_assertions.h"
+#import "ios/chrome/test/earl_grey/chrome_earl_grey.h"
+#import "ios/chrome/test/earl_grey/chrome_earl_grey_ui.h"
+#import "ios/chrome/test/earl_grey/chrome_matchers.h"
+#import "ios/chrome/test/earl_grey/chrome_test_case.h"
+#import "ios/testing/earl_grey/disabled_test_macros.h"
+#import "ios/web/public/test/http_server.h"
+#include "ios/web/public/test/http_server_util.h"
+#include "ui/base/l10n/l10n_util_mac.h"
+
+using chrome_test_util::buttonWithAccessibilityLabelId;
+using chrome_test_util::omniboxText;
+using chrome_test_util::webViewContainingText;
+
+// Toolbar integration tests for Chrome.
+@interface ToolbarTestCase : ChromeTestCase
+@end
+
+namespace {
+
+// Displays the |panel_type| new tab page.  On a phone this will send a command
+// to display a dialog, on tablet this calls -selectPanel to slide the NTP.
+void SelectNewTabPagePanel(NewTabPage::PanelIdentifier panel_type) {
+  NewTabPageController* ntp_controller =
+      chrome_test_util::GetCurrentNewTabPageController();
+  if (IsIPadIdiom()) {
+    [ntp_controller selectPanel:panel_type];
+  } else {
+    NSUInteger tag = 0;
+    if (panel_type == NewTabPage::PanelIdentifier::kBookmarksPanel) {
+      tag = IDC_SHOW_BOOKMARK_MANAGER;
+    } else if (panel_type == NewTabPage::PanelIdentifier::kOpenTabsPanel) {
+      tag = IDC_SHOW_OTHER_DEVICES;
+    }
+    if (tag) {
+      base::scoped_nsobject<GenericChromeCommand> command(
+          [[GenericChromeCommand alloc] initWithTag:tag]);
+      chrome_test_util::RunCommandWithActiveViewController(command);
+    }
+  }
+  [[GREYUIThreadExecutor sharedInstance] drainUntilIdle];
+}
+
+}  // namespace
+
+@implementation ToolbarTestCase
+
+#pragma mark Tests
+
+// Verifies that entering a URL in the omnibox navigates to the correct URL and
+// displays content.
+- (void)testEnterURL {
+  web::test::SetUpFileBasedHttpServer();
+  const GURL URL = web::test::HttpServer::MakeUrl(
+      "http://ios/testing/data/http_server_files/destination.html");
+  [ChromeEarlGrey loadURL:URL];
+  [[EarlGrey selectElementWithMatcher:omniboxText(URL.GetContent())]
+      assertWithMatcher:grey_notNil()];
+  [[EarlGrey selectElementWithMatcher:webViewContainingText("You've arrived")]
+      assertWithMatcher:grey_notNil()];
+}
+
+// Verifies opening a new tab from the tools menu.
+- (void)testNewTabFromMenu {
+  chrome_test_util::AssertMainTabCount(1);
+
+  // Open tab via the UI.
+  [ChromeEarlGreyUI openToolsMenu];
+  id<GREYMatcher> newTabButtonMatcher =
+      grey_accessibilityID(kToolsMenuNewTabId);
+  [[EarlGrey selectElementWithMatcher:newTabButtonMatcher]
+      performAction:grey_tap()];
+
+  chrome_test_util::AssertMainTabCount(2);
+}
+
+// Verifies opening a new incognito tab from the tools menu.
+// TODO(crbug.com/631078): Enable this test.
+- (void)DISABLED_testNewIncognitoTabFromMenu {
+  chrome_test_util::AssertIncognitoTabCount(0);
+
+  // Open incognito tab.
+  [ChromeEarlGreyUI openToolsMenu];
+  id<GREYMatcher> newIncognitoTabButtonMatcher =
+      grey_accessibilityID(kToolsMenuNewIncognitoTabId);
+  [[EarlGrey selectElementWithMatcher:newIncognitoTabButtonMatcher]
+      performAction:grey_tap()];
+
+  chrome_test_util::AssertIncognitoTabCount(1);
+}
+
+// Tests whether input mode in an omnibox can be canceled via "Cancel" button
+// and asserts it doesn't commit the omnibox contents if the input is canceled.
+- (void)testToolbarOmniboxCancel {
+  // Handset only (tablet does not have cancel button).
+  if (IsIPadIdiom()) {
+    EARL_GREY_TEST_SKIPPED(@"Test not support on iPad");
+  }
+
+  const GURL URL = web::test::HttpServer::MakeUrl("http://origin");
+
+  [ChromeEarlGrey loadURL:URL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"foo")];
+
+  id<GREYMatcher> CancelButton =
+      grey_allOf(chrome_test_util::buttonWithAccessibilityLabelId(IDS_CANCEL),
+                 grey_not(grey_accessibilityID(@"Typing Shield")), nil);
+  [[EarlGrey selectElementWithMatcher:CancelButton] performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+}
+
+// Tests whether input mode in an omnibox can be canceled via "hide keyboard"
+// button and asserts it doesn't commit the omnibox contents if the input is
+// canceled.
+- (void)testToolbarOmniboxHideKeyboard {
+  // TODO(crbug.com/642559): Enable the test for iPad when typing bug is fixed.
+  if (IsIPadIdiom()) {
+    EARL_GREY_TEST_DISABLED(@"Disabled for iPad due to a simulator bug.");
+  }
+
+  // Tablet only (handset keyboard does not have "hide keyboard" button).
+  if (!IsIPadIdiom()) {
+    EARL_GREY_TEST_SKIPPED(@"Test not support on iPhone");
+  }
+
+  const GURL URL = web::test::HttpServer::MakeUrl("http://origin");
+
+  [ChromeEarlGrey loadURL:URL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"foo")];
+
+  id<GREYMatcher> hideKeyboard = grey_accessibilityLabel(@"Hide keyboard");
+  [[EarlGrey selectElementWithMatcher:hideKeyboard] performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+}
+
+// Tests whether input mode in an omnibox can be canceled via tapping the typing
+// shield and asserts it doesn't commit the omnibox contents if the input is
+// canceled.
+- (void)testToolbarOmniboxTypingShield {
+  // Tablet only (handset keyboard does not have "hide keyboard" button).
+  if (!IsIPadIdiom()) {
+    EARL_GREY_TEST_SKIPPED(@"Test not support on iPhone");
+  }
+
+  const GURL URL = web::test::HttpServer::MakeUrl("http://origin");
+
+  [ChromeEarlGrey loadURL:URL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"foo")];
+
+  id<GREYMatcher> typingShield = grey_accessibilityID(@"Typing Shield");
+  [[EarlGrey selectElementWithMatcher:typingShield] performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+}
+
+// Verifies the existence and state of toolbar UI elements.
+- (void)testToolbarUI {
+  id<GREYMatcher> backButton =
+      chrome_test_util::buttonWithAccessibilityLabelId(IDS_ACCNAME_BACK);
+  id<GREYMatcher> forwardButton =
+      chrome_test_util::buttonWithAccessibilityLabelId(IDS_ACCNAME_FORWARD);
+  id<GREYMatcher> reloadButton =
+      chrome_test_util::buttonWithAccessibilityLabelId(IDS_IOS_ACCNAME_RELOAD);
+  id<GREYMatcher> bookmarkButton =
+      chrome_test_util::buttonWithAccessibilityLabelId(IDS_TOOLTIP_STAR);
+  id<GREYMatcher> voiceSearchButton =
+      grey_allOf(chrome_test_util::buttonWithAccessibilityLabelId(
+                     IDS_IOS_ACCNAME_VOICE_SEARCH),
+                 grey_ancestor(grey_kindOfClass([ToolbarView class])), nil);
+  NSString* ntpOmniboxLabel = l10n_util::GetNSString(IDS_OMNIBOX_EMPTY_HINT);
+  NSString* focusedOmniboxLabel = l10n_util::GetNSString(IDS_ACCNAME_LOCATION);
+  NSString* omniboxLabel =
+      IsIPadIdiom() ? focusedOmniboxLabel : ntpOmniboxLabel;
+  id<GREYMatcher> locationbarButton =
+      grey_allOf(grey_accessibilityLabel(omniboxLabel),
+                 grey_minimumVisiblePercent(0.2), nil);
+
+  [[EarlGrey selectElementWithMatcher:locationbarButton]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  if (IsIPadIdiom()) {
+    [[EarlGrey selectElementWithMatcher:backButton]
+        assertWithMatcher:grey_sufficientlyVisible()];
+    [[EarlGrey selectElementWithMatcher:forwardButton]
+        assertWithMatcher:grey_sufficientlyVisible()];
+    [[EarlGrey selectElementWithMatcher:reloadButton]
+        assertWithMatcher:grey_sufficientlyVisible()];
+    [[EarlGrey selectElementWithMatcher:bookmarkButton]
+        assertWithMatcher:grey_sufficientlyVisible()];
+    [[EarlGrey selectElementWithMatcher:voiceSearchButton]
+        assertWithMatcher:grey_sufficientlyVisible()];
+  } else {
+    [[EarlGrey selectElementWithMatcher:chrome_test_util::toolsMenuButton()]
+        assertWithMatcher:grey_sufficientlyVisible()];
+    [[EarlGrey selectElementWithMatcher:chrome_test_util::showTabsButton()]
+        assertWithMatcher:grey_sufficientlyVisible()];
+  }
+
+  // Navigate to a page and verify the back button is enabled.
+  [ChromeEarlGrey loadURL:GURL("chrome://version")];
+  [[EarlGrey selectElementWithMatcher:backButton]
+      assertWithMatcher:grey_interactable()];
+}
+
+// Verifies that the keyboard is properly dismissed when a toolbar button
+// is pressed (iPad specific).
+- (void)testIPadKeyboardDismissOnButtonPress {
+  // Tablet only (handset keyboard does not have "hide keyboard" button).
+  if (!IsIPadIdiom()) {
+    EARL_GREY_TEST_SKIPPED(@"Test not supported on iPhone");
+  }
+
+  // Load some page so that the "Back" button is tappable.
+  [ChromeEarlGrey loadURL:GURL("chrome://version")];
+
+  // First test: check that the keyboard is opened when tapping the omnibox,
+  // and that it is dismissed when the "Back" button is tapped.
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_tap()];
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityID(@"Typing Shield")]
+      assertWithMatcher:grey_notNil()];
+
+  id<GREYMatcher> backButton =
+      chrome_test_util::buttonWithAccessibilityLabelId(IDS_ACCNAME_BACK);
+  [[EarlGrey selectElementWithMatcher:backButton] performAction:grey_tap()];
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityID(@"Typing Shield")]
+      assertWithMatcher:grey_notVisible()];
+
+  // Second test: check that the keyboard is opened when tapping the omnibox,
+  // and that it is dismissed when the tools menu button is tapped.
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_tap()];
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityID(@"Typing Shield")]
+      assertWithMatcher:grey_notNil()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::toolsMenuButton()]
+      performAction:grey_tap()];
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityID(@"Typing Shield")]
+      assertWithMatcher:grey_notVisible()];
+}
+
+// Verifies that copying and pasting a URL includes the hidden protocol prefix.
+- (void)testCopyPasteURL {
+  std::map<GURL, std::string> responses;
+  const GURL URL = web::test::HttpServer::MakeUrl("http://testPage");
+  const GURL secondURL = web::test::HttpServer::MakeUrl("http://pastePage");
+
+  [ChromeEarlGrey loadURL:URL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.GetContent())];
+
+  NSString* shareButtonLabel = l10n_util::GetNSString(IDS_IOS_TOOLS_MENU_SHARE);
+  if (!IsIPadIdiom()) {
+    [ChromeEarlGreyUI openToolsMenu];
+  }
+  id<GREYMatcher> share_button =
+      chrome_test_util::buttonWithAccessibilityLabel(shareButtonLabel);
+  [[EarlGrey selectElementWithMatcher:share_button] performAction:grey_tap()];
+
+  [[EarlGrey
+      selectElementWithMatcher:chrome_test_util::buttonWithAccessibilityLabel(
+                                   @"Copy")] performAction:grey_tap()];
+
+  [ChromeEarlGrey loadURL:secondURL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_longPress()];
+
+  id<GREYMatcher> pasteBarButton = grey_allOf(
+      grey_accessibilityLabel(@"Paste"),
+      grey_not(grey_accessibilityTrait(UIAccessibilityTraitButton)),
+      grey_not(grey_accessibilityTrait(UIAccessibilityTraitStaticText)), nil);
+  [[EarlGrey selectElementWithMatcher:pasteBarButton] performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText(URL.spec().c_str())];
+}
+
+// Verifies that the clear text button clears any text in the omnibox.
+- (void)testOmniboxClearTextButton {
+  const GURL URL = web::test::HttpServer::MakeUrl("http://origin");
+
+  [ChromeEarlGrey loadURL:URL];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"foo")];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText("foo")];
+
+  id<GREYMatcher> CancelButton = grey_accessibilityLabel(@"Clear Text");
+  [[EarlGrey selectElementWithMatcher:CancelButton] performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      assertWithMatcher:chrome_test_util::omniboxText("")];
+}
+
+// Types JavaScript into Omnibox and verify that an alert is displayed.
+- (void)testTypeJavaScriptIntoOmnibox {
+  // TODO(crbug.com/642544): Enable the test for iPad when typing bug is fixed.
+  if (IsIPadIdiom() && base::ios::IsRunningOnIOS10OrLater()) {
+    EARL_GREY_TEST_DISABLED(@"Disabled for iOS10 iPad due to a typing bug.");
+  }
+  [ChromeEarlGrey loadURL:GURL("chrome://version")];
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"javascript:alert('Hello');")];
+
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityID(@"Go")]
+      performAction:grey_tap()];
+
+  [[EarlGrey selectElementWithMatcher:grey_accessibilityLabel(@"Hello")]
+      assertWithMatcher:grey_notNil()];
+}
+
+// Tests typing in the omnibox.
+- (void)testToolbarOmniboxTyping {
+  SelectNewTabPagePanel(NewTabPage::kMostVisitedPanel);
+
+  id<GREYMatcher> locationbarButton = grey_allOf(
+      grey_accessibilityLabel(l10n_util::GetNSString(IDS_OMNIBOX_EMPTY_HINT)),
+      grey_minimumVisiblePercent(0.2), nil);
+  [[EarlGrey selectElementWithMatcher:locationbarButton]
+      assertWithMatcher:grey_text(@"Search or type URL")];
+
+  [[EarlGrey selectElementWithMatcher:locationbarButton]
+      performAction:grey_typeText(@"a")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"a"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"b")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"ab"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"C")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"1")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC1"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"2")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC12"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"@")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC12@"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"{")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC12@{"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+      performAction:grey_typeText(@"#")];
+  [[EarlGrey
+      selectElementWithMatcher:grey_allOf(grey_accessibilityLabel(@"abC12@{#"),
+                                          grey_kindOfClass(
+                                              [OmniboxPopupMaterialRow class]),
+                                          nil)]
+      assertWithMatcher:grey_sufficientlyVisible()];
+
+  // TODO(crbug.com/642559): Enable these steps for iPad when typing bug
+  // is fixed.
+  if (IsIPadIdiom()) {
+    EARL_GREY_TEST_DISABLED(@"Disabled for iPad due to a simulator bug.");
+  } else {
+    NSString* cancelButton = l10n_util::GetNSString(IDS_CANCEL);
+    NSString* typingShield = @"Hide keyboard";
+    NSString* clearText = IsIPadIdiom() ? typingShield : cancelButton;
+
+    [[EarlGrey selectElementWithMatcher:grey_accessibilityLabel(clearText)]
+        performAction:grey_tap()];
+    [[EarlGrey selectElementWithMatcher:chrome_test_util::omnibox()]
+        assertWithMatcher:chrome_test_util::omniboxText("")];
+
+    SelectNewTabPagePanel(NewTabPage::kMostVisitedPanel);
+  }
+}
+@end