| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ | 5 #ifndef CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ |
| 6 #define CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ | 6 #define CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/basictypes.h" |
| 11 #include "base/compiler_specific.h" |
| 10 #include "base/run_loop.h" | 12 #include "base/run_loop.h" |
| 11 #include "chrome/browser/instant/instant_controller.h" | |
| 12 #include "chrome/browser/instant/instant_model_observer.h" | 13 #include "chrome/browser/instant/instant_model_observer.h" |
| 13 #include "chrome/browser/ui/browser.h" | 14 #include "googleurl/src/gurl.h" |
| 14 #include "chrome/browser/ui/browser_instant_controller.h" | 15 #include "net/test/test_server.h" |
| 15 #include "chrome/browser/ui/browser_window.h" | 16 |
| 16 #include "chrome/browser/ui/omnibox/location_bar.h" | 17 class Browser; |
| 17 #include "chrome/browser/ui/omnibox/omnibox_view.h" | 18 class InstantController; |
| 18 #include "chrome/common/search_types.h" | 19 class InstantModel; |
| 19 #include "chrome/test/base/in_process_browser_test.h" | 20 class InstantService; |
| 21 class OmniboxView; |
| 22 struct TemplateURLData; |
| 23 |
| 24 namespace content { |
| 25 class WebContents; |
| 26 } |
| 20 | 27 |
| 21 class InstantTestModelObserver : public InstantModelObserver { | 28 class InstantTestModelObserver : public InstantModelObserver { |
| 22 public: | 29 public: |
| 23 InstantTestModelObserver(InstantModel* model, | 30 InstantTestModelObserver(InstantController* controller, |
| 24 chrome::search::Mode::Type desired_mode_type); | 31 const InstantModel* desired_model); |
| 25 ~InstantTestModelObserver(); | 32 virtual ~InstantTestModelObserver(); |
| 26 | 33 |
| 27 void WaitUntilDesiredPreviewState(); | 34 void WaitForDesiredOverlayState(); |
| 28 | |
| 29 // Overridden from InstantModelObserver: | |
| 30 virtual void PreviewStateChanged(const InstantModel& model) OVERRIDE; | |
| 31 | 35 |
| 32 private: | 36 private: |
| 33 InstantModel* const model_; | 37 // Overridden from InstantModelObserver: |
| 34 const chrome::search::Mode::Type desired_mode_type_; | 38 virtual void OverlayStateChanged(const InstantModel& model) OVERRIDE; |
| 39 |
| 40 InstantController* const controller_; |
| 41 const InstantModel* const desired_model_; |
| 35 base::RunLoop run_loop_; | 42 base::RunLoop run_loop_; |
| 36 | 43 |
| 37 DISALLOW_COPY_AND_ASSIGN(InstantTestModelObserver); | 44 DISALLOW_COPY_AND_ASSIGN(InstantTestModelObserver); |
| 38 }; | 45 }; |
| 39 | 46 |
| 40 class InstantTestBase : public InProcessBrowserTest { | 47 // This utility class is meant to be used in a "mix-in" fashion, giving the |
| 48 // derived test class additional Instant-related functionality. |
| 49 class InstantTestBase { |
| 41 public: | 50 public: |
| 42 InstantTestBase() | 51 InstantTestBase(); |
| 43 : https_test_server_( | 52 virtual ~InstantTestBase(); |
| 44 net::TestServer::TYPE_HTTPS, | |
| 45 net::BaseTestServer::SSLOptions(), | |
| 46 base::FilePath(FILE_PATH_LITERAL("chrome/test/data"))) { | |
| 47 } | |
| 48 | 53 |
| 49 protected: | 54 // Accessors for common objects. These objects are meant for use by derived |
| 50 void SetupInstant(); | 55 // classes. InstantTestBase doesn't itself use them. |
| 56 net::TestServer* https_test_server() { return &https_test_server_; } |
| 57 const GURL& instant_url() const { return instant_url_; } |
| 58 void set_instant_url(const GURL& url) { instant_url_ = url; } |
| 59 void set_browser(Browser* browser) { browser_ = browser; } |
| 51 | 60 |
| 52 InstantController* instant() { | 61 // Initializes the user profile dir so that Instant is disabled. Returns true |
| 53 return browser()->instant_controller()->instant(); | 62 // on success. Call this from InProcessBrowserTest::SetUpUserDataDirectory() |
| 54 } | 63 // to ensure that Instant doesn't try to activate on browser startup. To |
| 64 // re-enable Instant later, call SetupDefaultSearchProvider(). |
| 65 bool DisableInstantOnStartup(); |
| 55 | 66 |
| 56 OmniboxView* omnibox() { | 67 // Sets up the default search engine. NOTE: |template_data| must have a valid |
| 57 return browser()->window()->GetLocationBar()->GetLocationEntry(); | 68 // Instant URL for this to work. In particular, InstantTestBase doesn't fill |
| 58 } | 69 // it in using |instant_url_|. |
| 70 void SetupDefaultSearchProvider(const TemplateURLData& template_data); |
| 59 | 71 |
| 60 void KillInstantRenderView(); | 72 InstantService* service() const; |
| 73 InstantController* instant() const; |
| 74 OmniboxView* omnibox() const; |
| 61 | 75 |
| 62 void FocusOmnibox(); | 76 void FocusOmnibox(); |
| 63 void FocusOmniboxAndWaitForInstantSupport(); | 77 void FocusOmniboxAndWaitForInstantSupport(); |
| 64 | 78 |
| 65 void SetOmniboxText(const std::string& text); | 79 void SetOmniboxText(const std::string& text); |
| 66 void SetOmniboxTextAndWaitForInstantToShow(const std::string& text); | 80 void SetOmniboxTextAndWaitForOverlayToShow(const std::string& text); |
| 67 | 81 |
| 68 bool GetBoolFromJS(content::WebContents* contents, | 82 bool GetBoolFromJS(content::WebContents* contents, |
| 69 const std::string& script, | 83 const std::string& script, |
| 70 bool* result) WARN_UNUSED_RESULT; | 84 bool* result) WARN_UNUSED_RESULT; |
| 71 bool GetIntFromJS(content::WebContents* contents, | 85 bool GetIntFromJS(content::WebContents* contents, |
| 72 const std::string& script, | 86 const std::string& script, |
| 73 int* result) WARN_UNUSED_RESULT; | 87 int* result) WARN_UNUSED_RESULT; |
| 74 bool GetStringFromJS(content::WebContents* contents, | 88 bool GetStringFromJS(content::WebContents* contents, |
| 75 const std::string& script, | 89 const std::string& script, |
| 76 std::string* result) WARN_UNUSED_RESULT; | 90 std::string* result) WARN_UNUSED_RESULT; |
| 77 bool ExecuteScript(const std::string& script) WARN_UNUSED_RESULT; | |
| 78 bool CheckVisibilityIs(content::WebContents* contents, | 91 bool CheckVisibilityIs(content::WebContents* contents, |
| 79 bool expected) WARN_UNUSED_RESULT; | 92 bool expected) WARN_UNUSED_RESULT; |
| 80 bool HasUserInputInProgress(); | |
| 81 bool HasTemporaryText(); | |
| 82 | 93 |
| 94 void KillOverlayRenderView(); |
| 95 |
| 96 private: |
| 97 net::TestServer https_test_server_; |
| 83 GURL instant_url_; | 98 GURL instant_url_; |
| 99 Browser* browser_; |
| 84 | 100 |
| 85 // HTTPS Testing server, started on demand. | 101 DISALLOW_COPY_AND_ASSIGN(InstantTestBase); |
| 86 net::TestServer https_test_server_; | |
| 87 }; | 102 }; |
| 88 | 103 |
| 89 #endif // CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ | 104 #endif // CHROME_BROWSER_INSTANT_INSTANT_TEST_UTILS_H_ |
| OLD | NEW |