1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
#!/usr/bin/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 logging
import os
import re
import pyauto_functional # Must be imported before pyauto
import pyauto
import test_utils
class InfobarTest(pyauto.PyUITest):
"""TestCase for Infobars."""
def Debug(self):
"""Test method for experimentation.
This method will not run automatically.
To run:
python chrome/test/functional/infobars.py infobars.InfobarTest.Debug
"""
import pprint
pp = pprint.PrettyPrinter(indent=2)
while True:
raw_input('Hit <enter> to dump info.. ')
info = self.GetBrowserInfo()
for window in info['windows']:
for tab in window['tabs']:
print 'Window', window['index'], 'tab', tab['index']
pp.pprint(tab['infobars'])
def setUp(self):
pyauto.PyUITest.setUp(self)
self._flash_plugin_type = 'Plug-in'
if (self.IsChromeOS() and
self.GetBrowserInfo()['properties']['branding'] == 'Google Chrome'):
self._flash_plugin_type = 'Pepper Plugin'
def _GetTabInfo(self, windex=0, tab_index=0):
"""Helper to return info for the given tab in the given window.
Defaults to first tab in first window.
"""
return self.GetBrowserInfo()['windows'][windex]['tabs'][tab_index]
def testPluginCrashInfobar(self):
"""Verify the "plugin crashed" infobar."""
flash_url = self.GetFileURLForDataPath('plugin', 'flash.swf')
# Trigger flash plugin
self.NavigateToURL(flash_url)
child_processes = self.GetBrowserInfo()['child_processes']
flash = [x for x in child_processes if
x['type'] == self._flash_plugin_type and
x['name'] == 'Shockwave Flash'][0]
self.assertTrue(flash)
logging.info('Killing flash plugin. pid %d' % flash['pid'])
self.Kill(flash['pid'])
self.assertTrue(self.WaitForInfobarCount(1))
crash_infobar = self._GetTabInfo()['infobars']
self.assertTrue(crash_infobar)
self.assertEqual(1, len(crash_infobar))
self.assertTrue(re.match('The following plug-in has crashed:',
crash_infobar[0]['text']))
self.assertEqual('confirm_infobar', crash_infobar[0]['type'])
# Dismiss the infobar
self.PerformActionOnInfobar('dismiss', infobar_index=0)
self.assertFalse(self._GetTabInfo()['infobars'])
def _VerifyGeolocationInfobar(self, match_text, windex, tab_index):
"""Verify geolocation infobar and match given text.
Assumes that geolocation infobar is showing up in the given tab in the
given window.
"""
tab_info = self._GetTabInfo(windex, tab_index)
geolocation_infobar = tab_info['infobars']
self.assertTrue(geolocation_infobar)
self.assertEqual(1, len(geolocation_infobar))
self.assertEqual(match_text, geolocation_infobar[0]['text'])
self.assertEqual('Learn more', geolocation_infobar[0]['link_text'])
self.assertEqual(2, len(geolocation_infobar[0]['buttons']))
self.assertEqual('Allow', geolocation_infobar[0]['buttons'][0])
self.assertEqual('Deny', geolocation_infobar[0]['buttons'][1])
def testGeolocationInfobar(self):
"""Verify geoLocation infobar."""
url = self.GetFileURLForDataPath( # triggers geolocation
'geolocation', 'geolocation_on_load.html')
match_text='file:/// wants to track your physical location'
self.NavigateToURL(url)
self.assertTrue(self.WaitForInfobarCount(1))
self._VerifyGeolocationInfobar(windex=0, tab_index=0, match_text=match_text)
# Accept, and verify that the infobar went away
self.PerformActionOnInfobar('accept', infobar_index=0)
self.assertFalse(self._GetTabInfo()['infobars'])
def testGeolocationInfobarInMultipleTabsAndWindows(self):
"""Verify GeoLocation inforbar in multiple tabs."""
url = self.GetFileURLForDataPath( # triggers geolocation
'geolocation', 'geolocation_on_load.html')
match_text='file:/// wants to track your physical location'
for tab_index in range(1, 2):
self.AppendTab(pyauto.GURL(url))
self.assertTrue(
self.WaitForInfobarCount(1, windex=0, tab_index=tab_index))
self._VerifyGeolocationInfobar(windex=0, tab_index=tab_index,
match_text=match_text)
# Try in a new window
self.OpenNewBrowserWindow(True)
self.NavigateToURL(url, 1, 0)
self.assertTrue(self.WaitForInfobarCount(1, windex=1, tab_index=0))
self._VerifyGeolocationInfobar(windex=1, tab_index=0, match_text=match_text)
# Incognito window
self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
self.NavigateToURL(url, 2, 0)
self.assertTrue(self.WaitForInfobarCount(1, windex=2, tab_index=0))
self._VerifyGeolocationInfobar(windex=2, tab_index=0, match_text=match_text)
def testMultipleDownloadsInfobar(self):
"""Verify the mutiple downloads infobar."""
zip_file = 'a_zip_file.zip'
html_file = 'download-a_zip_file.html'
assert pyauto.PyUITest.IsEnUS()
file_url = self.GetFileURLForDataPath('downloads', html_file)
match_text = 'This site is attempting to download multiple files. ' \
'Do you want to allow this?'
self.NavigateToURL('chrome://downloads') # trigger download manager
test_utils.RemoveDownloadedTestFile(self, zip_file)
self.DownloadAndWaitForStart(file_url)
# trigger page reload, which triggers the download infobar
self.GetBrowserWindow(0).GetTab(0).Reload()
self.assertTrue(self.WaitForInfobarCount(1))
tab_info = self._GetTabInfo(0, 0)
infobars = tab_info['infobars']
self.assertTrue(infobars, 'Expected the multiple downloads infobar')
self.assertEqual(1, len(infobars))
self.assertEqual(match_text, infobars[0]['text'])
self.assertEqual(2, len(infobars[0]['buttons']))
self.assertEqual('Allow', infobars[0]['buttons'][0])
self.assertEqual('Deny', infobars[0]['buttons'][1])
self.WaitForAllDownloadsToComplete()
test_utils.RemoveDownloadedTestFile(self, zip_file)
def testPluginCrashForMultiTabs(self):
"""Verify plugin crash infobar shows up only on the tabs using plugin."""
if self.IsMac():
# On Mac 10.5, flash files loaded too quickly after firing browser ends
# up getting downloaded, which seems to indicate that the plugin hasn't
# been registered yet.
# Hack to register Flash plugin on Mac 10.5. crbug.com/94123
self.GetPluginsInfo()
non_flash_url = self.GetFileURLForDataPath('english_page.html')
flash_url = self.GetFileURLForDataPath('plugin', 'FlashSpin.swf')
# False = Non flash url, True = Flash url
# We have set of these values to compare a flash page and a non-flash page
urls_type = [False, True, False, True, False]
for count in range(2):
self.AppendTab(pyauto.GURL(flash_url))
self.AppendTab(pyauto.GURL(non_flash_url))
# Killing flash process
child_processes = self.GetBrowserInfo()['child_processes']
flash = [x for x in child_processes if
x['type'] == self._flash_plugin_type and
x['name'] == 'Shockwave Flash'][0]
self.assertTrue(flash)
self.Kill(flash['pid'])
# Crash plugin infobar should show up in the second tab of this window
# so passing window and tab argument in the wait for an infobar.
self.assertTrue(self.WaitForInfobarCount(1, windex=0, tab_index=1))
for i in range(len(urls_type)):
# Verify that if page doesn't have flash plugin,
# it should not have infobar popped-up
self.ActivateTab(i)
if not urls_type[i]:
info = self.GetBrowserInfo()
self.assertFalse(
info['windows'][0]['tabs'][i]['infobars'],
msg='Did not expect crash infobar in tab at index %d' % i)
elif urls_type[i]:
self.assertTrue(
self.WaitForInfobarCount(1, windex=0, tab_index=i),
msg='Expected crash infobar in tab at index %d' % i)
infobar = self.GetBrowserInfo()['windows'][0]['tabs'][i]['infobars']
self.assertEqual(infobar[0]['type'], 'confirm_infobar')
self.assertEqual(len(infobar), 1)
if __name__ == '__main__':
pyauto_functional.Main()
|
