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 OminiboxMatchFound(): if is_incognito: matches = test_utils.GetOmniboxMatchesFor(self, search_str, windex=2) else: matches = test_utils.GetOmniboxMatchesFor(self, search_str) for x in matches: if x['destination_url'] == app_url: return True return False
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 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 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 _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'])
def testUnblockedPopupAddedToOmnibox(self): """Verify that an unblocked popup shows up in omnibox suggestions.""" file_url = self.GetFileURLForDataPath('popup_blocker', 'popup-window-open.html') self.NavigateToURL(file_url) self.assertEqual(1, len(self.GetBlockedPopupsInfo())) self.UnblockAndLaunchBlockedPopup(0) search_string = 'data:text/html,<title>Popup Success!</title> \ you should not see this message if popup blocker is enabled' matches = test_utils.GetOmniboxMatchesFor(self, search_string) self.assertEqual(search_string, matches[0]['destination_url']) self.assertEqual(search_string, matches[0]['contents'])
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'])
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 _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 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 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 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'])
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'])