Set the original url correctly if the frame is loaded via loadData

content/public/test/content_browser_test_utils.cc

 // Copyright (c) 2012 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.
 
 #include "content/public/test/content_browser_test_utils.h"
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
@@ skipping 25 matching lines @@
                                                 const GURL& url,
                                                 int number_of_navigations) {
   WaitForLoadStop(window->web_contents());
   TestNavigationObserver same_tab_observer(window->web_contents(),
                                            number_of_navigations);
 
   window->LoadURL(url);
   same_tab_observer.Wait();
 }
 
+void LoadDataWithBaseURL(Shell* window, const GURL& url,
+    const std::string data, const GURL& base_url) {
+  WaitForLoadStop(window->web_contents());
+  TestNavigationObserver same_tab_observer(window->web_contents(), 1);
+
+  window->LoadDataWithBaseURL(url, data, base_url);
+  same_tab_observer.Wait();
+}
+
 void NavigateToURL(Shell* window, const GURL& url) {
   NavigateToURLBlockUntilNavigationsComplete(window, url, 1);
 }
 
 void WaitForAppModalDialog(Shell* window) {
   ShellJavaScriptDialogManager* dialog_manager=
       static_cast<ShellJavaScriptDialogManager*>(
           window->GetJavaScriptDialogManager());
 
   scoped_refptr<MessageLoopRunner> runner = new MessageLoopRunner();
@@ skipping 20 matching lines @@
 }
 
 void ShellAddedObserver::ShellCreated(Shell* shell) {
   DCHECK(!shell_);
   shell_ = shell;
   if (runner_.get())
     runner_->QuitClosure().Run();
 }
 
 }  // namespace content