Web shell test to go back and forward.

ios/web/shell/test/web_shell_navigation_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 <UIKit/UIKit.h>
+#import <WebKit/WebKit.h>
 #import <XCTest/XCTest.h>
 
 #import <EarlGrey/EarlGrey.h>
 
+#include "base/strings/sys_string_conversions.h"
+#import "ios/web/public/test/http_server.h"
+#include "ios/web/public/test/http_server_util.h"
+#include "ios/web/shell/test/utils/navigation_egutil.h"
+#import "ios/web/shell/test/utils/web_view_egutil.h"
 #import "ios/web/shell/view_controller.h"
 
 @interface CRWWebShellNavigationTest : XCTestCase
 
 @end
 
 @implementation CRWWebShellNavigationTest
 
-// Sample test to load a live URL, go back and then forward.
-- (void)testExternalURLBackAndForward {
+// Set up called once for the class.
++ (void)setUp {
+  [super setUp];
+  [[EarlGrey selectElementWithMatcher:shell_webViewContainingText(@"Chromium")]
+      assertWithMatcher:grey_notNil()];
+  web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance();
+  server.StartOrDie();
+  DCHECK(server.IsRunning());
+}
+
+// Tear down called once for the class.
++ (void)tearDown {
+  [super tearDown];
+  web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance();
+  server.Stop();
+  DCHECK(!server.IsRunning());
+}
+
+// Tear down called after each test.
+- (void)tearDown {
+  [super tearDown];
+  web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance();
+  server.RemoveAllResponseProviders();
+}
+
+// Tests the back and forward button after entering two URLs.
+- (void)testWebScenarioBrowsingBackAndForward {
+  // Create map of canned responses and set up the test HTML server.
+  std::map<GURL, std::string> responses;
+  const GURL firstURL = web::test::HttpServer::MakeUrl("http://firstURL");
+  NSString* firstResponse = @"Test Page 1";
+  responses[firstURL] = base::SysNSStringToUTF8(firstResponse);
+
+  const GURL secondURL = web::test::HttpServer::MakeUrl("http://secondURL");
+  NSString* secondResponse = @"Test Page 2";
+  responses[secondURL] = base::SysNSStringToUTF8(secondResponse);
+
+  web::test::SetUpSimpleHttpServer(responses);
+
+  navigation_egutil::LoadURL(firstURL);
+
   [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellAddressFieldAccessibilityLabel)]
-      performAction:grey_tap()];
-  [[EarlGrey selectElementWithMatcher:grey_accessibilityLabel(@"Clear text")]
-      performAction:grey_tap()];
+      selectElementWithMatcher:shell_webViewContainingText(firstResponse)]
+      assertWithMatcher:grey_notNil()];
+
+  navigation_egutil::LoadURL(secondURL);
+
   [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellAddressFieldAccessibilityLabel)]
-      performAction:grey_typeText(@"http://browsingtest.appspot.com\n")];
+      selectElementWithMatcher:shell_webViewContainingText(secondResponse)]
+      assertWithMatcher:grey_notNil()];
+
   [[EarlGrey
       selectElementWithMatcher:grey_accessibilityLabel(
                                    kWebShellBackButtonAccessibilityLabel)]
       performAction:grey_tap()];
   [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellForwardButtonAccessibilityLabel)]
-      performAction:grey_tap()];
-}
-
-// Sample test to load a live URL, go back, forward, and then back again.
-- (void)testExternalURLBackAndForwardAndBack {
-  [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellAddressFieldAccessibilityLabel)]
-      performAction:grey_tap()];
-  [[EarlGrey selectElementWithMatcher:grey_accessibilityLabel(@"Clear text")]
-      performAction:grey_tap()];
-  [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellAddressFieldAccessibilityLabel)]
-      performAction:grey_typeText(@"http://browsingtest.appspot.com\n")];
-  [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellBackButtonAccessibilityLabel)]
-      performAction:grey_tap()];
+      selectElementWithMatcher:shell_webViewContainingText(firstResponse)]
+      assertWithMatcher:grey_notNil()];
   [[EarlGrey
       selectElementWithMatcher:grey_accessibilityLabel(
                                    kWebShellForwardButtonAccessibilityLabel)]
       performAction:grey_tap()];
+
   [[EarlGrey
-      selectElementWithMatcher:grey_accessibilityLabel(
-                                   kWebShellBackButtonAccessibilityLabel)]
-      performAction:grey_tap()];
+      selectElementWithMatcher:shell_webViewContainingText(secondResponse)]
+      assertWithMatcher:grey_notNil()];
 }
+
 @end