InstantExtended: tests!

chrome/browser/instant/instant_controller.cc

 // Copyright 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 "chrome/browser/instant/instant_controller.h"
 
 #include "base/command_line.h"
 #include "base/metrics/histogram.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
@@ skipping 140 matching lines @@
 const char* InstantController::kLocalOmniboxPopupURL =
     "chrome://local-omnibox-popup/local-omnibox-popup.html";
 
 InstantController::InstantController(chrome::BrowserInstantController* browser,
                                      bool extended_enabled,
                                      bool use_local_preview_only)
     : browser_(browser),
       extended_enabled_(extended_enabled),
       instant_enabled_(false),
       use_local_preview_only_(use_local_preview_only),
+      in_testing_mode_(false),
       model_(ALLOW_THIS_IN_INITIALIZER_LIST(this)),
       last_omnibox_text_has_inline_autocompletion_(false),
       last_verbatim_(false),
       last_transition_type_(content::PAGE_TRANSITION_LINK),
       last_match_was_search_(false),
       omnibox_focus_state_(OMNIBOX_FOCUS_NONE),
       start_margin_(0),
       end_margin_(0),
       allow_preview_to_show_search_suggestions_(false) {
 }
@@ skipping 743 matching lines @@
 
 void InstantController::NavigateToURL(const GURL& url,
                                       content::PageTransition transition) {
   if (!extended_enabled_)
     return;
   if (loader_)
     HideLoader();
   browser_->OpenURLInCurrentTab(url, transition);
 }
 
+void InstantController::SetTestingMode() {
+  in_testing_mode_ = true;
+}
+
 bool InstantController::ResetLoader(const TemplateURL* template_url,
                                     const content::WebContents* active_tab,
                                     bool fallback_to_local) {
   std::string instant_url;
   if (!GetInstantURL(template_url, &instant_url)) {
     if (!fallback_to_local || !extended_enabled_)
       return false;
 
     // If we are in extended mode, fallback to the local popup.
     instant_url = kLocalOmniboxPopupURL;
@@ skipping 231 matching lines @@
     return false;
 
   // Even if the URL template doesn't have search terms, it may have other
   // components (such as {google:baseURL}) that need to be replaced.
   *instant_url = instant_url_ref.ReplaceSearchTerms(
       TemplateURLRef::SearchTermsArgs(string16()));
 
   // Extended mode should always use HTTPS. TODO(sreeram): This section can be
   // removed if TemplateURLs supported "https://{google:host}/..." instead of
   // only supporting "{google:baseURL}...".
-  if (extended_enabled_) {
+  if (extended_enabled_ && !in_testing_mode_) {
     GURL url_obj(*instant_url);
     if (!url_obj.is_valid())
       return false;
 
     if (!url_obj.SchemeIsSecure()) {
       std::string new_scheme = "https";
       std::string new_port = "443";
       GURL::Replacements secure;
       secure.SetSchemeStr(new_scheme);
       secure.SetPortStr(new_port);
       url_obj = url_obj.ReplaceComponents(secure);
 
       if (!url_obj.is_valid())
         return false;
 
       *instant_url = url_obj.spec();
     }
   }
 
   std::map<std::string, int>::const_iterator iter =
       blacklisted_urls_.find(*instant_url);
   if (iter != blacklisted_urls_.end() &&
       iter->second > kMaxInstantSupportFailures)
     return false;
 
   return true;
 }