diff options
Diffstat (limited to 'chrome/test/functional/omnibox.py')
| -rwxr-xr-x | chrome/test/functional/omnibox.py | 369 |
1 files changed, 0 insertions, 369 deletions
diff --git a/chrome/test/functional/omnibox.py b/chrome/test/functional/omnibox.py deleted file mode 100755 index 75f6aac..0000000 --- a/chrome/test/functional/omnibox.py +++ /dev/null @@ -1,369 +0,0 @@ -#!/usr/bin/env python -# Copyright (c) 2011 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. - -import glob -import os -import re -import shutil -import tempfile -import urlparse - -import pyauto_functional # Must be imported before pyauto -import pyauto -import test_utils - - -class OmniboxTest(pyauto.PyUITest): - """Test cases for the omnibox.""" - - def Debug(self): - """Test method for experimentation. - - This method will not run automatically. - """ - import time - while True: - self.pprint(self.GetOmniboxInfo().omniboxdict) - time.sleep(1) - - def testFocusOnStartup(self): - """Verify that the omnibox has focus on startup.""" - self.WaitUntilOmniboxReadyHack() - self.assertTrue(self.GetOmniboxInfo().Properties('has_focus')) - - def testHistoryResult(self): - """Verify that the omnibox can fetch items from the history.""" - url = self.GetFileURLForDataPath('title2.html') - title = 'Title Of Awesomeness' - self.AppendTab(pyauto.GURL(url)) - def _VerifyHistoryResult(query_list, description, windex=0): - """Verify result matching given description for given list of queries.""" - for query_text in query_list: - matches = test_utils.GetOmniboxMatchesFor( - self, query_text, windex=windex, - attr_dict={'description': description}) - self.assertTrue(matches) - self.assertEqual(1, len(matches)) - item = matches[0] - self.assertEqual(url, item['destination_url']) - # Query using URL & title. - _VerifyHistoryResult([url, title], title) - # Verify results in another tab. - self.AppendTab(pyauto.GURL()) - _VerifyHistoryResult([url, title], title) - # Verify results in another window. - self.OpenNewBrowserWindow(True) - self.WaitUntilOmniboxReadyHack(windex=1) - _VerifyHistoryResult([url, title], title, windex=1) - # Verify results in an incognito window. - self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) - self.WaitUntilOmniboxReadyHack(windex=2) - _VerifyHistoryResult([url, title], title, windex=2) - - def _VerifyOmniboxURLMatches(self, url, description, windex=0): - """Verify URL match results from the omnibox. - - Args: - url: The URL to use. - description: The string description within the history page and Google - search to match against. - windex: The window index to work on. Defaults to 0 (first window). - """ - matches_description = test_utils.GetOmniboxMatchesFor( - self, url, windex=windex, attr_dict={'description': description}) - self.assertEqual(1, len(matches_description)) - if description == 'Google Search': - self.assertTrue(re.match('http://www.google.com/search.+', - matches_description[0]['destination_url'])) - else: - self.assertEqual(url, matches_description[0]['destination_url']) - - def testFetchHistoryResultItems(self): - """Verify omnibox fetches history items in 2nd tab, window and incognito.""" - url = self.GetFileURLForDataPath('title2.html') - title = 'Title Of Awesomeness' - desc = 'Google Search' - # Fetch history page item in the second tab. - self.AppendTab(pyauto.GURL(url)) - self._VerifyOmniboxURLMatches(url, title) - # Fetch history page items in the second window. - self.OpenNewBrowserWindow(True) - self.NavigateToURL(url, 1, 0) - self._VerifyOmniboxURLMatches(url, title, windex=1) - # Fetch google search items in incognito window. - self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) - self.NavigateToURL(url, 2, 0) - self._VerifyOmniboxURLMatches(url, desc, windex=2) - - def testSelect(self): - """Verify omnibox popup selection.""" - url1 = self.GetFileURLForDataPath('title2.html') - url2 = self.GetFileURLForDataPath('title1.html') - title1 = 'Title Of Awesomeness' - self.NavigateToURL(url1) - self.NavigateToURL(url2) - matches = test_utils.GetOmniboxMatchesFor(self, 'file://') - self.assertTrue(matches) - # Find the index of match for |url1|. - index = None - for i, match in enumerate(matches): - if match['description'] == title1: - index = i - self.assertTrue(index is not None) - self.OmniboxMovePopupSelection(index) # Select |url1| line in popup. - self.assertEqual(url1, self.GetOmniboxInfo().Text()) - self.OmniboxAcceptInput() - self.assertEqual(title1, self.GetActiveTabTitle()) - - def testInlineAutoComplete(self): - """Verify inline autocomplete for a pre-visited URL.""" - self.NavigateToURL('http://www.google.com') - matches = test_utils.GetOmniboxMatchesFor(self, 'goog') - self.assertTrue(matches) - # Omnibox should suggest auto completed URL as the first item. - matches_description = matches[0] - self.assertTrue('www.google.com' in matches_description['contents']) - self.assertEqual('history-url', matches_description['type']) - # The URL should be inline-autocompleted in the omnibox. - self.assertTrue('google.com' in self.GetOmniboxInfo().Text()) - - def testCrazyFilenames(self): - """Test omnibox query with filenames containing special chars. - - The files are created on the fly and cleaned after use. - """ - filename = os.path.join(self.DataDir(), 'downloads', 'crazy_filenames.txt') - zip_names = self.EvalDataFrom(filename) - # We got .zip filenames. Change them to .html. - crazy_filenames = [x.replace('.zip', '.html') for x in zip_names] - title = 'given title' - - def _CreateFile(name): - """Create the given html file.""" - fp = open(name, 'w') # |name| could be unicode. - print >>fp, '<html><title>%s</title><body>' % title - print >>fp, 'This is a junk file named <h2>%s</h2>' % repr(name) - print >>fp, '</body></html>' - fp.close() - - crazy_fileurls = [] - # Temp dir for hosting crazy filenames. - temp_dir = tempfile.mkdtemp(prefix='omnibox') - # Windows has a dual nature dealing with unicode filenames. - # While the files are internally saved as unicode, there's a non-unicode - # aware API that returns a locale-dependent coding on the true unicode - # filenames. This messes up things. - # Filesystem-interfacing functions like os.listdir() need to - # be given unicode strings to "do the right thing" on win. - # Ref: http://boodebr.org/main/python/all-about-python-and-unicode - try: - for filename in crazy_filenames: # |filename| is unicode. - file_path = os.path.join(temp_dir, filename.encode('utf-8')) - _CreateFile(os.path.join(temp_dir, filename)) - file_url = self.GetFileURLForPath(file_path) - crazy_fileurls.append(file_url) - self.NavigateToURL(file_url) - - # Verify omnibox queries. - for file_url in crazy_fileurls: - matches = test_utils.GetOmniboxMatchesFor(self, - file_url, attr_dict={'type': 'url-what-you-typed', - 'description': title}) - self.assertTrue(matches) - self.assertEqual(1, len(matches)) - self.assertTrue(os.path.basename(file_url) in - matches[0]['destination_url']) - finally: - shutil.rmtree(unicode(temp_dir)) # Unicode so that Win treats nicely. - - def testSuggest(self): - """Verify suggested results in omnibox.""" - matches = test_utils.GetOmniboxMatchesFor(self, 'apple') - self.assertTrue(matches) - self.assertTrue([x for x in matches if x['type'] == 'search-suggest']) - - def testDifferentTypesOfResults(self): - """Verify different types of results from omnibox. - - This includes history result, bookmark result, suggest results. - """ - url = 'http://www.google.com/' - title = 'Google' - search_string = 'google' - self.AddBookmarkURL( # Add a bookmark. - self.GetBookmarkModel().BookmarkBar()['id'], 0, title, url) - self.NavigateToURL(url) # Build up history. - matches = test_utils.GetOmniboxMatchesFor(self, search_string) - self.assertTrue(matches) - # Verify starred result (indicating bookmarked url). - self.assertTrue([x for x in matches if x['starred'] == True]) - for item_type in ('history-url', 'search-what-you-typed', - 'search-suggest',): - self.assertTrue([x for x in matches if x['type'] == item_type]) - - def testSuggestPref(self): - """Verify no suggests for omnibox when suggested-services disabled.""" - search_string = 'apple' - self.assertTrue(self.GetPrefsInfo().Prefs(pyauto.kSearchSuggestEnabled)) - matches = test_utils.GetOmniboxMatchesFor(self, search_string) - self.assertTrue(matches) - self.assertTrue([x for x in matches if x['type'] == 'search-suggest']) - # Disable suggest-service. - self.SetPrefs(pyauto.kSearchSuggestEnabled, False) - self.assertFalse(self.GetPrefsInfo().Prefs(pyauto.kSearchSuggestEnabled)) - matches = test_utils.GetOmniboxMatchesFor(self, search_string) - self.assertTrue(matches) - # Verify there are no suggest results. - self.assertFalse([x for x in matches if x['type'] == 'search-suggest']) - - def testAutoCompleteForSearch(self): - """Verify omnibox autocomplete for search.""" - search_string = 'youtu' - verify_string = 'youtube' - matches = test_utils.GetOmniboxMatchesFor(self, search_string) - # Retrieve last contents element. - matches_description = matches[-1]['contents'].split() - self.assertEqual(verify_string, matches_description[0]) - - def _GotContentHistory(self, search_text, url): - """Check if omnibox returns a previously-visited page for given search text. - - Args: - search_text: The string search text. - url: The string URL to look for in the omnibox matches. - - Returns: - True, if the omnibox returns the previously-visited page for the given - search text, or False otherwise. - """ - # Omnibox doesn't change results if searching the same text repeatedly. - # So setting '' in omnibox before the next repeated search. - self.SetOmniboxText('') - matches = test_utils.GetOmniboxMatchesFor(self, search_text) - matches_description = [x for x in matches if x['destination_url'] == url] - return 1 == len(matches_description) - - def testContentHistory(self): - """Verify omnibox results when entering page content. - - Test verifies that visited page shows up in omnibox on entering page - content. - """ - url = self.GetFileURLForPath( - os.path.join(self.DataDir(), 'find_in_page', 'largepage.html')) - self.NavigateToURL(url) - self.assertTrue(self.WaitUntil( - lambda: self._GotContentHistory('British throne', url))) - - def testOmniboxSearchHistory(self): - """Verify page navigation/search from omnibox are added to the history.""" - url = self.GetFileURLForDataPath('title2.html') - self.NavigateToURL(url) - self.AppendTab(pyauto.GURL('about:blank')) - self.SetOmniboxText('java') - self.WaitUntilOmniboxQueryDone() - self.OmniboxAcceptInput() - history = self.GetHistoryInfo().History() - self.assertEqual(2, len(history)) - self.assertEqual(url, history[1]['url']) - self.assertEqual('java - Google Search', history[0]['title']) - - def _VerifyHasBookmarkResult(self, matches): - """Verify that we have a bookmark result. - - Args: - matches: A list of match items, as returned by - test_utils.GetOmniboxMatchesFor(). - """ - matches_starred = [result for result in matches if result['starred']] - self.assertTrue(matches_starred) - self.assertEqual(1, len(matches_starred)) - - def _CheckBookmarkResultForVariousInputs(self, url, title, windex=0): - """Check if we get the bookmark for complete and partial inputs. - - Args: - url: A string URL. - title: A string title for the given URL. - windex: The window index to use. Defaults to 0 (first window). - """ - # Check if the complete URL would get the bookmark. - url_matches = test_utils.GetOmniboxMatchesFor(self, url, windex=windex) - self._VerifyHasBookmarkResult(url_matches) - # Check if the complete title would get the bookmark. - title_matches = test_utils.GetOmniboxMatchesFor(self, title, windex=windex) - self._VerifyHasBookmarkResult(title_matches) - # Check if the partial URL would get the bookmark. - split_url = urlparse.urlsplit(url) - partial_url = test_utils.GetOmniboxMatchesFor( - self, split_url.scheme, windex=windex) - self._VerifyHasBookmarkResult(partial_url) - # Check if the partial title would get the bookmark. - split_title = title.split() - search_term = split_title[len(split_title) - 1] - partial_title = test_utils.GetOmniboxMatchesFor( - self, search_term, windex=windex) - self._VerifyHasBookmarkResult(partial_title) - - def testBookmarkResultInNewTabAndWindow(self): - """Verify omnibox finds bookmarks in search options of new tabs/windows.""" - url = self.GetFileURLForDataPath('title2.html') - self.NavigateToURL(url) - title = 'This is Awesomeness' - bookmarks = self.GetBookmarkModel() - bar_id = bookmarks.BookmarkBar()['id'] - self.AddBookmarkURL(bar_id, 0, title, url) - bookmarks = self.GetBookmarkModel() - nodes = bookmarks.FindByTitle(title) - self.AppendTab(pyauto.GURL(url)) - self._CheckBookmarkResultForVariousInputs(url, title) - self.OpenNewBrowserWindow(True) - self.assertEqual(2, self.GetBrowserWindowCount()) - self.NavigateToURL(url, 1, 0) - self._CheckBookmarkResultForVariousInputs(url, title, windex=1) - self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) - self.assertEqual(3, self.GetBrowserWindowCount()) - self.NavigateToURL(url, 2, 0) - self._CheckBookmarkResultForVariousInputs(url, title, windex=2) - - def testAutoCompleteForNonAsciiSearch(self): - """Verify can search/autocomplete with non-ASCII incomplete keywords.""" - search_string = u'\u767e' - verify_string = u'\u767e\u5ea6\u4e00\u4e0b' - matches = test_utils.GetOmniboxMatchesFor(self, search_string) - self.assertTrue(verify_string in matches[-1]['contents']) - - -class OmniboxLiveTest(pyauto.PyUITest): - """Test cases for the omnibox that hit live servers (such as Google).""" - - def ExtraChromeFlags(self): - """Override default list of extra flags used in pyauto tests.""" - # Force the suggest field trial group. This doesn't guarantee that there - # will be no experimental behaviour, but there's no other way to disable - # all suggest field trials at the moment. TODO(mpearson): Consider allowing - # the suggest_url to be overridden using a flag (so that we can omit the - # "sugexp=chrome,mod=<n>" CGI param), or provide some other way to turn off - # all suggest field trials. - return ['--force-fieldtrials=OmniboxSearchSuggest/10/'] - - def testGoogleSearch(self): - """Verify Google search item in omnibox results.""" - search_text = 'hello world' - verify_str = 'Google Search' - url_re = 'http://www.google.com/search\?.*q=hello\+world.*' - matches_description = test_utils.GetOmniboxMatchesFor( - self, search_text, attr_dict={'description': verify_str}) - self.assertTrue(matches_description) - # There should be a least one entry with the description Google. Suggest - # results may end up having 'Google Search' in them, so use >=. - self.assertTrue(len(matches_description) >= 1) - item = matches_description[0] - self.assertTrue(re.search(url_re, item['destination_url'])) - self.assertEqual('search-what-you-typed', item['type']) - - -if __name__ == '__main__': - pyauto_functional.Main() |