1 #!/usr/bin/env python 2 # Copyright 2013 The Chromium Authors. All rights reserved. 3 # Use of this source code is governed by a BSD-style license that can be 4 # found in the LICENSE file. 5 6 import logging 7 import os 8 import re 9 10 import pyauto_functional # Must be imported before pyauto 11 import pyauto 12 import test_utils 13 14 15 class InfobarTest(pyauto.PyUITest): 16 """TestCase for Infobars.""" 17 18 def Debug(self): 19 """Test method for experimentation. 20 21 This method will not run automatically. 22 To run: 23 python chrome/test/functional/infobars.py infobars.InfobarTest.Debug 24 """ 25 while True: 26 raw_input('Hit <enter> to dump info.. ') 27 info = self.GetBrowserInfo() 28 for window in info['windows']: 29 for tab in window['tabs']: 30 print 'Window', window['index'], 'tab', tab['index'] 31 self.pprint(tab['infobars']) 32 33 def setUp(self): 34 pyauto.PyUITest.setUp(self) 35 self._flash_plugin_type = 'Plug-in' 36 if self.GetBrowserInfo()['properties']['branding'] == 'Google Chrome': 37 self._flash_plugin_type = 'Pepper Plugin' 38 # Forcibly trigger all plugins to get registered. crbug.com/94123 39 # Sometimes flash files loaded too quickly after firing browser 40 # ends up getting downloaded, which seems to indicate that the plugin 41 # hasn't been registered yet. 42 self.GetPluginsInfo() 43 44 def _GetTabInfo(self, windex=0, tab_index=0): 45 """Helper to return info for the given tab in the given window. 46 47 Defaults to first tab in first window. 48 """ 49 return self.GetBrowserInfo()['windows'][windex]['tabs'][tab_index] 50 51 def testPluginCrashInfobar(self): 52 """Verify the "plugin crashed" infobar.""" 53 flash_url = self.GetFileURLForContentDataPath('plugin', 'flash.swf') 54 # Trigger flash plugin 55 self.NavigateToURL(flash_url) 56 child_processes = self.GetBrowserInfo()['child_processes'] 57 flash = [x for x in child_processes if 58 x['type'] == self._flash_plugin_type and 59 x['name'] == 'Shockwave Flash'][0] 60 self.assertTrue(flash) 61 logging.info('Killing flash plugin. pid %d' % flash['pid']) 62 self.Kill(flash['pid']) 63 self.assertTrue(self.WaitForInfobarCount(1)) 64 crash_infobar = self._GetTabInfo()['infobars'] 65 self.assertTrue(crash_infobar) 66 self.assertEqual(1, len(crash_infobar)) 67 self.assertTrue('crashed' in crash_infobar[0]['text']) 68 self.assertEqual('confirm_infobar', crash_infobar[0]['type']) 69 # Dismiss the infobar 70 self.PerformActionOnInfobar('dismiss', infobar_index=0) 71 self.assertFalse(self._GetTabInfo()['infobars']) 72 73 def _VerifyGeolocationInfobar(self, windex, tab_index): 74 """Verify geolocation infobar properties. 75 76 Assumes that geolocation infobar is showing up in the given tab in the 77 given window. 78 """ 79 # TODO(dyu): Remove this helper function when a function to identify 80 # infobar_type and index of the type is implemented. 81 tab_info = self._GetTabInfo(windex, tab_index) 82 geolocation_infobar = tab_info['infobars'] 83 self.assertTrue(geolocation_infobar) 84 self.assertEqual(1, len(geolocation_infobar)) 85 self.assertEqual('Learn more', geolocation_infobar[0]['link_text']) 86 self.assertEqual(2, len(geolocation_infobar[0]['buttons'])) 87 self.assertEqual('Allow', geolocation_infobar[0]['buttons'][0]) 88 self.assertEqual('Deny', geolocation_infobar[0]['buttons'][1]) 89 90 def testGeolocationInfobar(self): 91 """Verify geoLocation infobar.""" 92 url = self.GetHttpURLForDataPath('geolocation', 'geolocation_on_load.html') 93 self.NavigateToURL(url) 94 self.assertTrue(self.WaitForInfobarCount(1)) 95 self._VerifyGeolocationInfobar(windex=0, tab_index=0) 96 # Accept, and verify that the infobar went away 97 self.PerformActionOnInfobar('accept', infobar_index=0) 98 self.assertFalse(self._GetTabInfo()['infobars']) 99 100 def testGeolocationInfobarInMultipleTabsAndWindows(self): 101 """Verify GeoLocation inforbar in multiple tabs.""" 102 url = self.GetFileURLForDataPath( # triggers geolocation 103 'geolocation', 'geolocation_on_load.html') 104 for tab_index in range(1, 2): 105 self.AppendTab(pyauto.GURL(url)) 106 self.assertTrue( 107 self.WaitForInfobarCount(1, windex=0, tab_index=tab_index)) 108 self._VerifyGeolocationInfobar(windex=0, tab_index=tab_index) 109 # Try in a new window 110 self.OpenNewBrowserWindow(True) 111 self.NavigateToURL(url, 1, 0) 112 self.assertTrue(self.WaitForInfobarCount(1, windex=1, tab_index=0)) 113 self._VerifyGeolocationInfobar(windex=1, tab_index=0) 114 # Incognito window 115 self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) 116 self.NavigateToURL(url, 2, 0) 117 self.assertTrue(self.WaitForInfobarCount(1, windex=2, tab_index=0)) 118 self._VerifyGeolocationInfobar(windex=2, tab_index=0) 119 120 def _GetFlashCrashInfobarCount(self, windex=0, tab_index=0): 121 """Returns the count of 'Shockwave Flash has crashed' infobars.""" 122 browser_window = self.GetBrowserInfo()['windows'][windex] 123 infobars = browser_window['tabs'][tab_index]['infobars'] 124 flash_crash_infobar_count = 0 125 for infobar in infobars: 126 if (('text' in infobar) and 127 infobar['text'].startswith('Shockwave Flash has crashed')): 128 flash_crash_infobar_count += 1 129 return flash_crash_infobar_count 130 131 def testPluginCrashForMultiTabs(self): 132 """Verify plugin crash infobar shows up only on the tabs using plugin.""" 133 non_flash_url = self.GetFileURLForDataPath('english_page.html') 134 flash_url = self.GetFileURLForContentDataPath('plugin', 'FlashSpin.swf') 135 # False = Non flash url, True = Flash url 136 # We have set of these values to compare a flash page and a non-flash page 137 urls_type = [False, True, False, True, False] 138 for _ in range(2): 139 self.AppendTab(pyauto.GURL(flash_url)) 140 self.AppendTab(pyauto.GURL(non_flash_url)) 141 # Killing flash process 142 child_processes = self.GetBrowserInfo()['child_processes'] 143 flash = [x for x in child_processes if 144 x['type'] == self._flash_plugin_type and 145 x['name'] == 'Shockwave Flash'][0] 146 self.assertTrue(flash) 147 self.Kill(flash['pid']) 148 # Crash plugin infobar should show up in the second tab of this window 149 # so passing window and tab argument in the wait for an infobar. 150 self.assertTrue(self.WaitForInfobarCount(1, windex=0, tab_index=1)) 151 for i in range(len(urls_type)): 152 # Verify that if page doesn't have flash plugin, 153 # it should not have infobar popped-up 154 self.ActivateTab(i) 155 if not urls_type[i]: 156 self.assertEqual( 157 self._GetFlashCrashInfobarCount(0, i), 0, 158 msg='Did not expect crash infobar in tab at index %d' % i) 159 elif urls_type[i]: 160 self.assertEqual( 161 self._GetFlashCrashInfobarCount(0, i), 1, 162 msg='Expected crash infobar in tab at index %d' % i) 163 infobar = self.GetBrowserInfo()['windows'][0]['tabs'][i]['infobars'] 164 self.assertEqual(infobar[0]['type'], 'confirm_infobar') 165 self.assertEqual(len(infobar), 1) 166 167 168 class OneClickInfobarTest(pyauto.PyUITest): 169 """Tests for one-click sign in infobar.""" 170 171 BLOCK_COOKIE_PATTERN = {'https://accounts.google.com/': {'cookies': 2}} 172 OC_INFOBAR_TYPE = 'oneclicklogin_infobar' 173 PW_INFOBAR_TYPE = 'password_infobar' 174 URL = 'https://www.google.com/accounts/ServiceLogin' 175 URL_LOGIN = 'https://www.google.com/accounts/Login' 176 URL_LOGOUT = 'https://www.google.com/accounts/Logout' 177 178 def setUp(self): 179 pyauto.PyUITest.setUp(self) 180 self._driver = self.NewWebDriver() 181 182 def _LogIntoGoogleAccount(self, tab_index=0, windex=0): 183 """Log into Google account. 184 185 Args: 186 tab_index: The tab index, default is 0. 187 windex: The window index, default is 0. 188 """ 189 creds = self.GetPrivateInfo()['test_google_account'] 190 username = creds['username'] 191 password = creds['password'] 192 test_utils.GoogleAccountsLogin(self, username, password, tab_index, windex) 193 # TODO(dyu): Use WaitUntilNavigationCompletes after investigating 194 # crbug.com/124877 195 self.WaitUntil( 196 lambda: self.GetDOMValue('document.readyState'), 197 expect_retval='complete') 198 199 def _PerformActionOnInfobar(self, action): 200 """Perform an action on the infobar: accept, cancel, or dismiss. 201 202 The one-click sign in infobar must show in the first tab of the first 203 window. If action is 'accept' then the account is synced. If the action is 204 'cancel' then the infobar should be dismissed and never shown again. The 205 account will not be synced. If the action is 'dismiss' then the infobar will 206 shown again after the next login. 207 208 Args: 209 action: The action to perform on the infobar. 210 """ 211 infobar_index = test_utils.WaitForInfobarTypeAndGetIndex( 212 self, self.OC_INFOBAR_TYPE) 213 self.PerformActionOnInfobar(action, infobar_index) 214 215 def _DisplayOneClickInfobar(self, tab_index=0, windex=0): 216 """One-click sign in infobar appears after logging into google account. 217 218 Args: 219 tab_index: The tab index, default is 0. 220 windex: The window index, default is 0. 221 """ 222 self._LogIntoGoogleAccount(tab_index=tab_index, windex=windex) 223 self.assertTrue(self.WaitUntil( 224 lambda: test_utils.GetInfobarIndexByType( 225 self, self.OC_INFOBAR_TYPE, 226 tab_index=tab_index, windex=windex) is not None), 227 msg='The one-click login infobar did not appear.') 228 229 def testDisplayOneClickInfobar(self): 230 """Verify one-click infobar appears after login into google account. 231 232 One-click infobar should appear after signing into a google account 233 for the first time using a clean profile. 234 """ 235 self._DisplayOneClickInfobar() 236 237 def testNoOneClickInfobarAfterCancel(self): 238 """Verify one-click infobar does not appear again after clicking cancel. 239 240 The one-click infobar should not display again after logging into an 241 account and selecting to reject sync the first time. The test covers 242 restarting the browser with the same profile and verifying the one-click 243 infobar does not show after login. 244 245 This test also verifies that the password infobar displays. 246 """ 247 self._DisplayOneClickInfobar() 248 self._PerformActionOnInfobar(action='cancel') # Click 'No thanks' button. 249 self.NavigateToURL(self.URL_LOGOUT) 250 self._LogIntoGoogleAccount() 251 test_utils.WaitForInfobarTypeAndGetIndex(self, self.PW_INFOBAR_TYPE) 252 test_utils.AssertInfobarTypeDoesNotAppear(self, self.OC_INFOBAR_TYPE) 253 # Restart browser with the same profile. 254 self.RestartBrowser(clear_profile=False) 255 self.NavigateToURL(self.URL_LOGOUT) 256 self._LogIntoGoogleAccount() 257 test_utils.AssertInfobarTypeDoesNotAppear(self, self.OC_INFOBAR_TYPE) 258 259 def testDisplayOneClickInfobarAfterDismiss(self): 260 """Verify one-click infobar appears again after clicking dismiss button. 261 262 The one-click infobar should display again after logging into an 263 account and clicking to dismiss the infobar the first time. 264 265 This test also verifies that the password infobar does not display. 266 The one-click infobar should supersede the password infobar. 267 """ 268 self._DisplayOneClickInfobar() 269 self._PerformActionOnInfobar(action='dismiss') # Click 'x' button. 270 self.NavigateToURL(self.URL_LOGOUT) 271 self._LogIntoGoogleAccount() 272 test_utils.WaitForInfobarTypeAndGetIndex(self, self.OC_INFOBAR_TYPE) 273 test_utils.AssertInfobarTypeDoesNotAppear(self, self.PW_INFOBAR_TYPE) 274 275 def _OpenSecondProfile(self): 276 """Create a second profile.""" 277 self.OpenNewBrowserWindowWithNewProfile() 278 self.assertEqual(2, len(self.GetMultiProfileInfo()['profiles']), 279 msg='The second profile was not created.') 280 281 def testDisplayOneClickInfobarPerProfile(self): 282 """Verify one-click infobar appears for each profile after sign-in.""" 283 # Default profile. 284 self._DisplayOneClickInfobar() 285 self._OpenSecondProfile() 286 self._DisplayOneClickInfobar(windex=1) 287 288 def testNoOneClickInfobarWhenCookiesBlocked(self): 289 """Verify one-click infobar does not show when cookies are blocked. 290 291 One-click sign in should not be enabled if cookies are blocked for Google 292 accounts domain. 293 294 This test verifies the following bug: crbug.com/117841 295 """ 296 # Block cookies for Google accounts domain. 297 self.SetPrefs(pyauto.kContentSettingsPatternPairs, 298 self.BLOCK_COOKIE_PATTERN) 299 self._LogIntoGoogleAccount() 300 test_utils.AssertInfobarTypeDoesNotAppear(self, self.OC_INFOBAR_TYPE) 301 302 def testOneClickInfobarShownWhenWinLoseFocus(self): 303 """Verify one-click infobar still shows when window loses focus. 304 305 This test verifies the following bug: crbug.com/121739 306 """ 307 self._LogIntoGoogleAccount() 308 test_utils.WaitForInfobarTypeAndGetIndex(self, self.OC_INFOBAR_TYPE) 309 # Open new window to shift focus away. 310 self.OpenNewBrowserWindow(True) 311 test_utils.GetInfobarIndexByType(self, self.OC_INFOBAR_TYPE) 312 313 def testNoOneClickInfobarInIncognito(self): 314 """Verify that one-click infobar does not show up in incognito mode.""" 315 self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) 316 self._LogIntoGoogleAccount(windex=1) 317 test_utils.AssertInfobarTypeDoesNotAppear( 318 self, self.OC_INFOBAR_TYPE, windex=1) 319 320 321 if __name__ == '__main__': 322 pyauto_functional.Main() 323
