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'])
