Home | History | Annotate | Download | only in browser 
      1 // Copyright (c) 2012 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 #include "base/command_line.h"
      6 #include "base/strings/string_util.h"
      7 #include "base/strings/utf_string_conversions.h"
      8 #include "chrome/browser/ui/browser.h"
      9 #include "chrome/browser/ui/tabs/tab_strip_model.h"
     10 #include "chrome/common/chrome_switches.h"
     11 #include "chrome/test/base/in_process_browser_test.h"
     12 #include "chrome/test/base/ui_test_utils.h"
     13 #include "content/public/browser/web_contents.h"
     14 #include "content/public/common/url_constants.h"
     15 #include "content/public/test/browser_test_utils.h"
     16 #include "net/test/spawned_test_server/spawned_test_server.h"
     17 
     18 namespace {
     19 
     20 struct IsSearchProviderTestData {
     21   IsSearchProviderTestData() : tab(NULL) {}
     22   IsSearchProviderTestData(content::WebContents* t, std::string h, GURL url)
     23       : tab(t), host(h), test_url(url) {
     24   }
     25 
     26   content::WebContents* tab;
     27   std::string host;
     28   GURL test_url;
     29 };
     30 
     31 }
     32 
     33 class SearchProviderTest : public InProcessBrowserTest {
     34  protected:
     35   SearchProviderTest() {}
     36 
     37   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     38     ASSERT_TRUE(test_server()->Start());
     39 
     40     // Map all hosts to our local server.
     41     std::string host_rule(
     42         "MAP * " + test_server()->host_port_pair().ToString());
     43     command_line->AppendSwitchASCII(switches::kHostRules, host_rule);
     44     // Use no proxy or otherwise this test will fail on a machine that has a
     45     // proxy configured.
     46     command_line->AppendSwitch(switches::kNoProxyServer);
     47 
     48     // Get the url for the test page.
     49     search_provider_test_url_ =
     50         test_server()->GetURL("files/is_search_provider_installed.html");
     51   }
     52 
     53   IsSearchProviderTestData StartIsSearchProviderInstalledTest(
     54       Browser* browser,
     55       const char* host,
     56       const char* expected_result) {
     57     GURL test_url(std::string("http://") + host +
     58         search_provider_test_url_.path() + "#" + expected_result);
     59     ui_test_utils::NavigateToURLWithDisposition(
     60         browser, test_url, NEW_FOREGROUND_TAB,
     61         ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
     62 
     63     // Bundle up information needed to verify the result.
     64     content::WebContents* tab =
     65         browser->tab_strip_model()->GetActiveWebContents();
     66     return IsSearchProviderTestData(tab, host, test_url);
     67   }
     68 
     69   void FinishIsSearchProviderInstalledTest(
     70       const IsSearchProviderTestData& data) {
     71     string16 title = data.tab->GetTitle();
     72     if (title.empty()) {
     73       content::TitleWatcher title_watcher(data.tab, ASCIIToUTF16("OK"));
     74       title_watcher.AlsoWaitForTitle(ASCIIToUTF16("FAIL"));
     75       title = title_watcher.WaitAndGetTitle();
     76     }
     77     EXPECT_EQ(ASCIIToUTF16("OK"), title);
     78   }
     79 
     80   GURL search_provider_test_url_;
     81 
     82  private:
     83   DISALLOW_COPY_AND_ASSIGN(SearchProviderTest);
     84 };
     85 
     86 #if defined(OS_WIN)
     87 // This is flaking on XP. See http://crbug.com/159530
     88 #define MAYBE_TestIsSearchProviderInstalled \
     89     DISABLED_TestIsSearchProviderInstalled
     90 #else
     91 #define MAYBE_TestIsSearchProviderInstalled TestIsSearchProviderInstalled
     92 #endif
     93 IN_PROC_BROWSER_TEST_F(SearchProviderTest,
     94                        MAYBE_TestIsSearchProviderInstalled) {
     95   // Use the default search provider, other installed search provider, and
     96   // one not installed as well. (Note that yahoo isn't tested because the
     97   // its host name varies a lot for different locales unlike Google and Bing,
     98   // which would make the test fail depending on the machine's locale.)
     99   const char* test_hosts[] = { "www.google.com",
    100                                "www.bing.com",
    101                                "localhost" };
    102   const char* expected_results[] = { "2",
    103                                      "1",
    104                                      "0" };
    105   COMPILE_ASSERT(arraysize(test_hosts) == arraysize(expected_results),
    106                  there_should_be_a_result_for_each_host);
    107   IsSearchProviderTestData test_data[2 * arraysize(test_hosts)];
    108 
    109   // Start results for the normal mode.
    110   for (size_t i = 0; i < arraysize(test_hosts); ++i) {
    111     test_data[i] = StartIsSearchProviderInstalledTest(
    112         browser(), test_hosts[i], expected_results[i]);
    113     FinishIsSearchProviderInstalledTest(test_data[i]);
    114   }
    115 
    116   // Start tests for incognito mode (and verify the result is 0).
    117   Browser* incognito_browser = CreateIncognitoBrowser();
    118   for (size_t i = 0; i < arraysize(test_hosts); ++i) {
    119     test_data[i + arraysize(test_hosts)] = StartIsSearchProviderInstalledTest(
    120         incognito_browser, test_hosts[i], "0");
    121     FinishIsSearchProviderInstalledTest(test_data[i + arraysize(test_hosts)]);
    122   }
    123 
    124   // The following should be re-enabled. At the moment, there are problems with
    125   // doing all of these queries in parallel -- see http://crbug.com/60043.
    126 #if 0
    127   // Remove the calls to FinishIsSearchProviderInstalledTest above when
    128   // re-enabling this code.
    129 
    130   // Do the verification.
    131   for (size_t i = 0; i < arraysize(test_data); ++i) {
    132     FinishIsSearchProviderInstalledTest(test_data[i]);
    133   }
    134 #endif
    135 }
    136 
    137 IN_PROC_BROWSER_TEST_F(SearchProviderTest,
    138                        TestIsSearchProviderInstalledWithException) {
    139   // Change the url for the test page to one that throws an exception when
    140   // toString is called on the argument given to isSearchProviderInstalled.
    141   search_provider_test_url_ = test_server()->GetURL(
    142       "files/is_search_provider_installed_with_exception.html");
    143 
    144   FinishIsSearchProviderInstalledTest(StartIsSearchProviderInstalledTest(
    145       browser(), "www.google.com", ""));
    146 }
    147