src/chrome/browser/ui/search/instant_test_utils.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_UI_SEARCH_INSTANT_TEST_UTILS_H_
   6 #define CHROME_BROWSER_UI_SEARCH_INSTANT_TEST_UTILS_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/metrics/field_trial.h"
  13 #include "base/run_loop.h"
  14 #include "base/strings/string16.h"
  15 #include "chrome/browser/ui/browser.h"
  16 #include "chrome/browser/ui/browser_instant_controller.h"
  17 #include "chrome/browser/ui/browser_window.h"
  18 #include "chrome/browser/ui/location_bar/location_bar.h"
  19 #include "chrome/browser/ui/search/instant_controller.h"
  20 #include "chrome/common/search_types.h"
  21 #include "net/test/spawned_test_server/spawned_test_server.h"
  22 #include "url/gurl.h"
  23
  24 class BrowserInstantController;
  25 class InstantController;
  26 class InstantModel;
  27 class OmniboxView;
  28
  29 namespace content {
  30 class WebContents;
  31 };
  32
  33 // This utility class is meant to be used in a "mix-in" fashion, giving the
  34 // derived test class additional Instant-related functionality.
  35 class InstantTestBase {
  36  protected:
  37   InstantTestBase();
  38   virtual ~InstantTestBase();
  39
  40  protected:
  41   void SetupInstant(Browser* browser);
  42   void Init(const GURL& instant_url, const GURL& ntp_url,
  43             bool init_suggestions_url);
  44
  45   void SetInstantURL(const std::string& url);
  46
  47   void set_browser(Browser* browser) {
  48     browser_ = browser;
  49   }
  50
  51   BrowserInstantController* browser_instant() {
  52     return browser_->instant_controller();
  53   }
  54
  55   InstantController* instant() {
  56     return browser_->instant_controller()->instant();
  57   }
  58
  59   OmniboxView* omnibox() {
  60     return browser_->window()->GetLocationBar()->GetOmniboxView();
  61   }
  62
  63   const GURL& instant_url() const { return instant_url_; }
  64
  65   const GURL& ntp_url() const { return ntp_url_; }
  66
  67   net::SpawnedTestServer& https_test_server() { return https_test_server_; }
  68
  69   void KillInstantRenderView();
  70
  71   void FocusOmnibox();
  72
  73   void SetOmniboxText(const std::string& text);
  74
  75   void PressEnterAndWaitForNavigation();
  76   void PressEnterAndWaitForFrameLoad();
  77
  78   bool GetBoolFromJS(content::WebContents* contents,
  79                      const std::string& script,
  80                      bool* result) WARN_UNUSED_RESULT;
  81   bool GetIntFromJS(content::WebContents* contents,
  82                     const std::string& script,
  83                     int* result) WARN_UNUSED_RESULT;
  84   bool GetStringFromJS(content::WebContents* contents,
  85                        const std::string& script,
  86                        std::string* result) WARN_UNUSED_RESULT;
  87   bool CheckVisibilityIs(content::WebContents* contents,
  88                          bool expected) WARN_UNUSED_RESULT;
  89
  90   std::string GetOmniboxText();
  91
  92   // Loads a named image from url |image| from the given |rvh| host.  |loaded|
  93   // returns whether the image was able to load without error.
  94   // The method returns true if the JavaScript executed cleanly.
  95   bool LoadImage(content::RenderViewHost* rvh,
  96                  const std::string& image,
  97                  bool* loaded);
  98
  99   // Returns the omnibox's inline autocompletion (shown in blue highlight).
 100   base::string16 GetBlueText();
 101
 102  private:
 103   GURL instant_url_;
 104   GURL ntp_url_;
 105
 106   Browser* browser_;
 107
 108   // HTTPS Testing server, started on demand.
 109   net::SpawnedTestServer https_test_server_;
 110
 111   // Set to true to initialize suggestions URL in default search provider.
 112   bool init_suggestions_url_;
 113
 114   DISALLOW_COPY_AND_ASSIGN(InstantTestBase);
 115 };
 116
 117 #endif  // CHROME_BROWSER_UI_SEARCH_INSTANT_TEST_UTILS_H_