Add chrome-search: access from Instant overlay

chrome/browser/instant/instant_test_utils.cc

 // Copyright 2013 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 "chrome/browser/instant/instant_test_utils.h"
 
 #include "base/command_line.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
+#include "chrome/browser/ui/omnibox/omnibox_view.h"
+#include "chrome/common/chrome_notification_types.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/interactive_test_utils.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/result_codes.h"
 #include "content/public/test/browser_test_utils.h"
 
 namespace {
 
 std::string WrapScript(const std::string& script) {
   return "domAutomationController.send(" + script + ")";
 }
@@ skipping 18 matching lines @@
   run_loop_.Run();
 }
 
 void InstantTestModelObserver::PreviewStateChanged(const InstantModel& model) {
   if (model.mode().mode == desired_mode_type_)
     run_loop_.Quit();
 }
 
 // InstantTestBase -----------------------------------------------------------
 
-void InstantTestBase::SetupInstant() {
+void InstantTestBase::SetupInstant(Browser* browser) {
+  browser_ = browser;
   TemplateURLService* service =
-      TemplateURLServiceFactory::GetForProfile(browser()->profile());
+      TemplateURLServiceFactory::GetForProfile(browser_->profile());
   ui_test_utils::WaitForTemplateURLServiceToLoad(service);
 
   TemplateURLData data;
   // Necessary to use exact URL for both the main URL and the alternate URL for
   // search term extraction to work in InstantExtended.
   data.SetURL(instant_url_.spec() + "q={searchTerms}");
   data.instant_url = instant_url_.spec();
   data.alternate_urls.push_back(instant_url_.spec() + "#q={searchTerms}");
   data.search_terms_replacement_key = "strk";
 
-  TemplateURL* template_url = new TemplateURL(browser()->profile(), data);
+  TemplateURL* template_url = new TemplateURL(browser_->profile(), data);
   service->Add(template_url);  // Takes ownership of |template_url|.
   service->SetDefaultSearchProvider(template_url);
 
-  browser()->profile()->GetPrefs()->SetBoolean(prefs::kInstantEnabled, true);
+  browser_->profile()->GetPrefs()->SetBoolean(prefs::kInstantEnabled, true);
 
   // TODO(shishir): Fix this ugly hack.
   instant()->SetInstantEnabled(false, true);
   instant()->SetInstantEnabled(true, false);
 }
 
+void InstantTestBase::Init(const GURL& instant_url) {
+  instant_url_ = instant_url;
+}
+
 void InstantTestBase::KillInstantRenderView() {
   base::KillProcess(
       instant()->GetPreviewContents()->GetRenderProcessHost()->GetHandle(),
       content::RESULT_CODE_KILLED,
       false);
 }
 
 void InstantTestBase::FocusOmnibox() {
   // If the omnibox already has focus, just notify Instant.
   if (omnibox()->model()->has_focus()) {
     instant()->OmniboxFocusChanged(OMNIBOX_FOCUS_VISIBLE,
                                    OMNIBOX_FOCUS_CHANGE_EXPLICIT, NULL);
   } else {
-    browser()->window()->GetLocationBar()->FocusLocation(false);
+    browser_->window()->GetLocationBar()->FocusLocation(false);
   }
 }
 
+void InstantTestBase::FocusOmniboxAndWaitForInstantSupport() {
+  content::WindowedNotificationObserver observer(
+      chrome::NOTIFICATION_INSTANT_OVERLAY_SUPPORT_DETERMINED,
+      content::NotificationService::AllSources());
+  FocusOmnibox();
+  observer.Wait();
+}
+
+void InstantTestBase::FocusOmniboxAndWaitForInstantExtendedSupport() {
+  content::WindowedNotificationObserver ntp_observer(
+      chrome::NOTIFICATION_INSTANT_NTP_SUPPORT_DETERMINED,
+      content::NotificationService::AllSources());
+  content::WindowedNotificationObserver overlay_observer(
+      chrome::NOTIFICATION_INSTANT_OVERLAY_SUPPORT_DETERMINED,
+      content::NotificationService::AllSources());
+  FocusOmnibox();
+  ntp_observer.Wait();
+  overlay_observer.Wait();
+}
+
 void InstantTestBase::SetOmniboxText(const std::string& text) {
   FocusOmnibox();
   omnibox()->SetUserText(UTF8ToUTF16(text));
 }
 
 void InstantTestBase::SetOmniboxTextAndWaitForInstantToShow(
     const std::string& text) {
   InstantTestModelObserver observer(
       instant()->model(), chrome::search::Mode::MODE_SEARCH_SUGGESTIONS);
   SetOmniboxText(text);
@@ skipping 34 matching lines @@
       actual == expected;
 }
 
 bool InstantTestBase::HasUserInputInProgress() {
   return omnibox()->model()->user_input_in_progress_;
 }
 
 bool InstantTestBase::HasTemporaryText() {
   return omnibox()->model()->has_temporary_text_;
 }
+
+bool InstantTestBase::LoadImage(content::RenderViewHost* rvh,
+                                const std::string& image,
+                                bool* loaded) {
+  std::string js_chrome =
+      "var img = document.createElement('img');"
+      "img.onerror = function() { domAutomationController.send(false); };"
+      "img.onload  = function() { domAutomationController.send(true); };"
+      "img.src = '" + image + "';";
+  return content::ExecuteScriptAndExtractBool(rvh, js_chrome, loaded);
+}