[ios] Implemented LoadHtml w/o using data:// URLs.

ios/web/public/test/web_test_with_web_state.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 "ios/web/public/test/web_test_with_web_state.h"
 
 #import <WebKit/WebKit.h>
 
 #include "base/base64.h"
 #include "base/strings/sys_string_conversions.h"
@@ skipping 55 matching lines @@
 void WebTestWithWebState::WillProcessTask(
     const base::PendingTask& pending_task) {
   // Nothing to do.
 }
 
 void WebTestWithWebState::DidProcessTask(
     const base::PendingTask& pending_task) {
   processed_a_task_ = true;
 }
 
+void WebTestWithWebState::LoadHtml(NSString* html, const GURL& url) {
+  ASSERT_FALSE(web_state()->IsLoading());
+
+  CRWWebController* web_controller = GetWebController(web_state());
+  [web_controller loadHTML:html forURL:url];
+
+  // Wait until the navigation is committed to update MIME type.
+  ASSERT_EQ(web::LOAD_REQUESTED, web_controller.loadPhase);

[Jackie Quinn, 2016/09/01 22:27:03]

Are we sure to catch it in LOAD_REQUESTED here? Any chance we would hit
PAGE_LOADING here?

[Eugene But (OOO till 7-30), 2016/09/01 23:02:14]

This worked with all tests we have, so I don't think PAGE_LOADING could happen
here. So far I was running this only locally but I want to give it a try on the
bots :)

+  base::TimeDelta spin_delay = base::TimeDelta::FromMilliseconds(10);
+  while (web_controller.loadPhase != PAGE_LOADING)

[Jackie Quinn, 2016/09/01 22:27:03]

web::PAGE_LOADING?

[Eugene But (OOO till 7-30), 2016/09/01 23:02:14]

This code is already in web namespace, so I removed web:: from other places
instead.

+    base::test::ios::SpinRunLoopWithMaxDelay(spin_delay);
+
+  // loadHTML:forURL: does not notify web view delegate about received response,
+  // so web controller does not get a chance to properly update MIME type and it
+  // should be set manually after navigation is committed but before WebState
+  // signal load completion and clients will start checking if MIME type is in
+  // fact HTML.
+  [web_controller webStateImpl]->SetContentsMimeType("text/html");
+
+  // Wait until the page is loaded.
+  ASSERT_EQ(web::PAGE_LOADING, web_controller.loadPhase);
+  while (web_controller.loadPhase != PAGE_LOADED)

[Jackie Quinn, 2016/09/01 22:27:03]

web::PAGE_LOADED? or maybe load requested above doesn't need web namespace.

[Eugene But (OOO till 7-30), 2016/09/01 23:02:14]

Yeah, I removed namespace.

+    base::test::ios::SpinRunLoopWithMaxDelay(spin_delay);
+
+  // Wait until scripts execution becomes possible.
+  base::test::ios::WaitUntilCondition(^bool {
+    return [ExecuteJavaScript(@"0;") isEqual:@0];
+  });
+}
+
 void WebTestWithWebState::LoadHtml(NSString* html) {
-  LoadHtml([html UTF8String]);
+  GURL url("https://chromium.test/");
+  LoadHtml(html, url);
 }
 
 void WebTestWithWebState::LoadHtml(const std::string& html) {
-  NSString* load_check = CreateLoadCheck();
-  std::string marked_html = html + [load_check UTF8String];
-  std::string encoded_html;
-  base::Base64Encode(marked_html, &encoded_html);
-  GURL url("data:text/html;charset=utf8;base64," + encoded_html);
-  LoadURL(url);
-
-  if (ResetPageIfNavigationStalled(load_check)) {
-    LoadHtml(html);
-  }
-}
-
-void WebTestWithWebState::LoadURL(const GURL& url) {
-  // First step is to ensure that the web controller has finished any previous
-  // page loads so the new load is not confused.
-  while ([GetWebController(web_state()) loadPhase] != PAGE_LOADED)
-    WaitForBackgroundTasks();
-  id originalMockDelegate =
-      [OCMockObject niceMockForProtocol:@protocol(CRWWebDelegate)];
-  id mockDelegate =
-      [[WebDelegateMock alloc] initWithRepresentedObject:originalMockDelegate];
-  id existingDelegate = GetWebController(web_state()).delegate;
-  GetWebController(web_state()).delegate = mockDelegate;
-
-  web::NavigationManagerImpl& navManager =
-      [GetWebController(web_state()) webStateImpl]->GetNavigationManagerImpl();
-  navManager.InitializeSession(@"name", nil, NO, 0);
-  [navManager.GetSessionController() addPendingEntry:url
-                                            referrer:web::Referrer()
-                                          transition:ui::PAGE_TRANSITION_TYPED
-                                   rendererInitiated:NO];
-
-  [GetWebController(web_state()) loadCurrentURL];
-  while ([GetWebController(web_state()) loadPhase] != PAGE_LOADED)
-    WaitForBackgroundTasks();
-  GetWebController(web_state()).delegate = existingDelegate;
-  [web_state()->GetView() layoutIfNeeded];
+  LoadHtml(base::SysUTF8ToNSString(html));
 }
 
 void WebTestWithWebState::WaitForBackgroundTasks() {
   // Because tasks can add new tasks to either queue, the loop continues until
   // the first pass where no activity is seen from either queue.
   bool activitySeen = false;
   base::MessageLoop* messageLoop = base::MessageLoop::current();
   messageLoop->AddTaskObserver(this);
   do {
     activitySeen = false;
@@ skipping 42 matching lines @@
 }
 
 web::WebState* WebTestWithWebState::web_state() {
   return web_state_.get();
 }
 
 const web::WebState* WebTestWithWebState::web_state() const {
   return web_state_.get();
 }
 
-bool WebTestWithWebState::ResetPageIfNavigationStalled(NSString* load_check) {
-  id inner_html = ExecuteJavaScript(
-      @"(document && document.body && document.body.innerHTML) || 'undefined'");
-  if (![inner_html rangeOfString:load_check].length) {
-    web_state_->SetWebUsageEnabled(false);
-    web_state_->SetWebUsageEnabled(true);
-    [GetWebController(web_state()) triggerPendingLoad];
-    return true;
-  }
-  return false;
-}
-
 NSString* WebTestWithWebState::CreateLoadCheck() {
   return [NSString stringWithFormat:@"<p style=\"display: none;\">%d</p>",
                                     s_html_load_count++];
 }
 
 }  // namespace web